tests/frontend/version.py: Added first frontend test

1 files changed, 29 insertions, 0 deletions

diff --git a/tests/frontend/version.py b/tests/frontend/version.py
new file mode 100644
index 000000000..217ee418d
--- /dev/null
+++ b/tests/frontend/version.py
@@ -0,0 +1,29 @@
+from click.testing import CliRunner
+import pytest
+
+# 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()
+    expected_start = "cli, version {}.{}".format(major, minor)
+    if not cli_version_output.startswith(expected_start):
+        raise AssertionError("Version output expected to begin with '{}',"
+                             .format(expected_start) +
+                             " output was: {}"
+                             .format(cli_version_output))
+
+
+@pytest.fixture(scope="module")
+def runner():
+    return CliRunner()
+
+
+def test_version(runner):
+    result = runner.invoke(cli, ['--version'])
+    assert result.exit_code == 0
+    assert_version(result.output)