tests/frontend/version.py: Use the fixture to run buildstream

1 files changed, 3 insertions, 11 deletions

diff --git a/tests/frontend/version.py b/tests/frontend/version.py
index 217ee418d..cd6da8bd7 100644
--- a/tests/frontend/version.py
+++ b/tests/frontend/version.py
@@ -1,12 +1,9 @@
-from click.testing import CliRunner
 import pytest
+from tests.testutils.runcli import cli
 
 # For utils.get_bst_version()
 from buildstream import utils
 
-# Import the main cli entrypoint
-from buildstream._frontend.main import cli
-
 
 def assert_version(cli_version_output):
     major, minor = utils.get_bst_version()
@@ -18,12 +15,7 @@ def assert_version(cli_version_output):
                              .format(cli_version_output))
 
 
-@pytest.fixture(scope="module")
-def runner():
-    return CliRunner()
-
-
-def test_version(runner):
-    result = runner.invoke(cli, ['--version'])
+def test_version(cli):
+    result = cli.run(args=['--version'])
     assert result.exit_code == 0
     assert_version(result.output)