1 files changed, 139 insertions, 0 deletions
diff --git a/tests/frontend/progress.py b/tests/frontend/progress.py
new file mode 100644
index 000000000..e3b127f3b
--- /dev/null
+++ b/tests/frontend/progress.py
@@ -0,0 +1,139 @@
+# Pylint doesn't play well with fixtures and dependency injection from pytest
+# pylint: disable=redefined-outer-name
+
+import os
+import pytest
+
+from buildstream.testing import cli  # pylint: disable=unused-import
+from buildstream import _yaml
+from buildstream._exceptions import ErrorDomain, LoadErrorReason
+
+from tests.testutils import generate_junction
+
+# Project directory
+DATA_DIR = os.path.join(os.path.dirname(os.path.realpath(__file__)), )
+
+
+@pytest.mark.datafiles(os.path.join(DATA_DIR, 'project'))
+def test_show_progress_tally(cli, datafiles):
+    # Check that the progress reporting messages give correct tallies
+    project = str(datafiles)
+    result = cli.run(project=project, args=['show', 'compose-all.bst'])
+    result.assert_success()
+    assert "  3 subtasks processed" in result.stderr
+    assert "3 of 3 subtasks processed" in result.stderr
+
+
+@pytest.mark.datafiles(os.path.join(DATA_DIR, 'project'))
+def test_junction_tally(cli, tmpdir, datafiles):
+    # Check that the progress reporting messages count elements in junctions
+    project = str(datafiles)
+    subproject_path = os.path.join(project, 'files', 'sub-project')
+    junction_path = os.path.join(project, 'elements', 'junction.bst')
+    element_path = os.path.join(project, 'elements', 'junction-dep.bst')
+
+    # Create a repo to hold the subproject and generate a junction element for it
+    generate_junction(tmpdir, subproject_path, junction_path, store_ref=True)
+
+    # Create a stack element to depend on a cross junction element
+    #
+    element = {
+        'kind': 'stack',
+        'depends': [{
+            'junction': 'junction.bst',
+            'filename': 'import-etc.bst'
+        }]
+    }
+    _yaml.roundtrip_dump(element, element_path)
+
+    result = cli.run(project=project,
+                     silent=True,
+                     args=['source', 'fetch', 'junction.bst'])
+    result.assert_success()
+
+    # Assert the correct progress tallies are in the logging
+    result = cli.run(project=project, args=['show', 'junction-dep.bst'])
+    assert "  2 subtasks processed" in result.stderr
+    assert "2 of 2 subtasks processed" in result.stderr
+
+
+@pytest.mark.datafiles(os.path.join(DATA_DIR, 'project'))
+def test_nested_junction_tally(cli, tmpdir, datafiles):
+    # Check that the progress reporting messages count elements in
+    # junctions of junctions
+    project = str(datafiles)
+    sub1_path = os.path.join(project, 'files', 'sub-project')
+    sub2_path = os.path.join(project, 'files', 'sub2-project')
+    # A junction element which pulls sub1 into sub2
+    sub1_element = os.path.join(project, 'files', 'sub2-project', 'elements', 'sub-junction.bst')
+    # A junction element which pulls sub2 into the main project
+    sub2_element = os.path.join(project, 'elements', 'junction.bst')
+    element_path = os.path.join(project, 'elements', 'junction-dep.bst')
+
+    generate_junction(tmpdir / "sub-project", sub1_path, sub1_element, store_ref=True)
+    generate_junction(tmpdir / "sub2-project", sub2_path, sub2_element, store_ref=True)
+
+    # Create a stack element to depend on a cross junction element
+    #
+    element = {
+        'kind': 'stack',
+        'depends': [{
+            'junction': 'junction.bst',
+            'filename': 'import-sub.bst'
+        }]
+    }
+    _yaml.roundtrip_dump(element, element_path)
+
+    result = cli.run(project=project,
+                     silent=True,
+                     args=['source', 'fetch', 'junction.bst'])
+    result.assert_success()
+
+    # Assert the correct progress tallies are in the logging
+    result = cli.run(project=project, args=['show', 'junction-dep.bst'])
+    assert "  3 subtasks processed" in result.stderr
+    assert "3 of 3 subtasks processed" in result.stderr
+
+
+@pytest.mark.datafiles(os.path.join(DATA_DIR, 'project'))
+def test_junction_dep_tally(cli, tmpdir, datafiles):
+    # Check that the progress reporting messages count elements in junctions
+    project = str(datafiles)
+    subproject_path = os.path.join(project, 'files', 'sub-project')
+    junction_path = os.path.join(project, 'elements', 'junction.bst')
+    element_path = os.path.join(project, 'elements', 'junction-dep.bst')
+
+    # Create a repo to hold the subproject and generate a junction element for it
+    generate_junction(tmpdir, subproject_path, junction_path, store_ref=True)
+
+    # Add dependencies to the junction (not allowed, but let's do it
+    # anyway)
+    with open(junction_path, 'a') as f:
+        deps = {
+            'depends': [
+                'manual.bst'
+            ]
+        }
+        _yaml.roundtrip_dump(deps, f)
+
+    # Create a stack element to depend on a cross junction element
+    #
+    element = {
+        'kind': 'stack',
+        'depends': [{
+            'junction': 'junction.bst',
+            'filename': 'import-etc.bst'
+        }]
+    }
+    _yaml.roundtrip_dump(element, element_path)
+
+    result = cli.run(project=project,
+                     silent=True,
+                     args=['source', 'fetch', 'junction-dep.bst'])
+
+    # Since we aren't allowed to specify any dependencies on a
+    # junction, we should fail
+    result.assert_main_error(ErrorDomain.LOAD, LoadErrorReason.INVALID_JUNCTION)
+
+    # We don't get a final tally in this case
+    assert "subtasks processed" not in result.stderr