summaryrefslogtreecommitdiff
path: root/tests/testutils
diff options
context:
space:
mode:
Diffstat (limited to 'tests/testutils')
-rw-r--r--tests/testutils/__init__.py1
-rw-r--r--tests/testutils/junction.py36
2 files changed, 37 insertions, 0 deletions
diff --git a/tests/testutils/__init__.py b/tests/testutils/__init__.py
index 93143b505..e9db94989 100644
--- a/tests/testutils/__init__.py
+++ b/tests/testutils/__init__.py
@@ -2,3 +2,4 @@ from .runcli import cli, cli_integration
from .repo import create_repo, ALL_REPO_KINDS
from .artifactshare import create_artifact_share
from .element_generators import create_element_size
+from .junction import generate_junction
diff --git a/tests/testutils/junction.py b/tests/testutils/junction.py
new file mode 100644
index 000000000..efc429ef6
--- /dev/null
+++ b/tests/testutils/junction.py
@@ -0,0 +1,36 @@
+import os
+from tests.testutils import create_repo
+from buildstream import _yaml
+
+
+# generate_junction()
+#
+# Generates a junction element with a git repository
+#
+# Args:
+# tmpdir: The tmpdir fixture, for storing the generated git repo
+# subproject_path: The path for the subproject, to add to the git repo
+# junction_path: The location to store the generated junction element
+# store_ref: Whether to store the ref in the junction.bst file
+#
+# Returns:
+# (str): The ref
+#
+def generate_junction(tmpdir, subproject_path, junction_path, *, store_ref=True):
+ # Create a repo to hold the subproject and generate
+ # a junction element for it
+ #
+ repo = create_repo('git', str(tmpdir))
+ source_ref = ref = repo.create(subproject_path)
+ if not store_ref:
+ source_ref = None
+
+ element = {
+ 'kind': 'junction',
+ 'sources': [
+ repo.source_config(ref=source_ref)
+ ]
+ }
+ _yaml.dump(element, junction_path)
+
+ return ref