tests/frontend/push.py: Adding test for `bst push`

1 files changed, 98 insertions, 0 deletions

diff --git a/tests/frontend/push.py b/tests/frontend/push.py
new file mode 100644
index 000000000..c7de02f54
--- /dev/null
+++ b/tests/frontend/push.py
@@ -0,0 +1,98 @@
+import os
+import pytest
+from tests.testutils import cli, create_artifact_share
+
+from buildstream import _yaml
+
+# Project directory
+DATA_DIR = os.path.join(
+    os.path.dirname(os.path.realpath(__file__)),
+    "project",
+)
+
+
+# Assert that a given artifact is in the share
+#
+def assert_shared(cli, share, project, element_name):
+    # NOTE: 'test' here is the name of the project
+    # specified in the project.conf we are testing with.
+    #
+    cache_key = cli.get_element_key(project, element_name)
+    if not share.has_artifact('test', element_name, cache_key):
+        raise AssertionError("Artifact share at {} does not contain the expected element {}"
+                             .format(share.repo, element_name))
+
+
+@pytest.mark.datafiles(DATA_DIR)
+def test_push(cli, tmpdir, datafiles):
+    project = os.path.join(datafiles.dirname, datafiles.basename)
+    share = create_artifact_share(os.path.join(str(tmpdir), 'artifactshare'))
+
+    # First build it without the artifact cache configured
+    result = cli.run(project=project, args=['build', 'target.bst'])
+    assert result.exit_code == 0
+
+    # Assert that we are now cached locally
+    state = cli.get_element_state(project, 'target.bst')
+    assert state == 'cached'
+
+    # Configure artifact share
+    cli.configure({
+        'artifacts': {
+            'pull-url': share.repo,
+            'push-url': share.repo,
+        }
+    })
+
+    # Now try bst push
+    result = cli.run(project=project, args=['push', 'target.bst'])
+    assert result.exit_code == 0
+
+    # And finally assert that the artifact is in the share
+    assert_shared(cli, share, project, 'target.bst')
+
+
+@pytest.mark.datafiles(DATA_DIR)
+def test_push_all(cli, tmpdir, datafiles):
+    project = os.path.join(datafiles.dirname, datafiles.basename)
+    share = create_artifact_share(os.path.join(str(tmpdir), 'artifactshare'))
+
+    # First build it without the artifact cache configured
+    result = cli.run(project=project, args=['build', 'target.bst'])
+    assert result.exit_code == 0
+
+    # Assert that we are now cached locally
+    state = cli.get_element_state(project, 'target.bst')
+    assert state == 'cached'
+
+    # Configure artifact share
+    cli.configure({
+        #
+        # FIXME: This test hangs "sometimes" if we allow
+        #        concurrent push.
+        #
+        #        It's not too bad to ignore since we're
+        #        using the local artifact cache functionality
+        #        only, but it should probably be fixed.
+        #
+        'scheduler': {
+            'pushers': 1
+        },
+        'artifacts': {
+            'pull-url': share.repo,
+            'push-url': share.repo,
+        }
+    })
+
+    # Now try bst push all the deps
+    result = cli.run(project=project, args=[
+        'push', 'target.bst',
+        '--deps', 'all'
+    ])
+    assert result.exit_code == 0
+
+    # And finally assert that all the artifacts are in the share
+    assert_shared(cli, share, project, 'target.bst')
+    assert_shared(cli, share, project, 'import-bin.bst')
+    assert_shared(cli, share, project, 'import-dev.bst')
+    assert_shared(cli, share, project, 'compose-all.bst')