Handle cross junction elements in workspaces.

Workspaces are now index by colon separated junction path. This
now allows to create workspaces for elements in external projects.

Workspaces are owned by context instead of root project. However
it is initialized once top-level project is registered as we need
to resolve paths relatively to this top-level project.

Part of #359.

1 files changed, 117 insertions, 0 deletions

diff --git a/tests/frontend/cross_junction_workspace.py b/tests/frontend/cross_junction_workspace.py
new file mode 100644
index 000000000..eb2bc2eb8
--- /dev/null
+++ b/tests/frontend/cross_junction_workspace.py
@@ -0,0 +1,117 @@
+import os
+from tests.testutils import cli, create_repo
+from buildstream import _yaml
+
+
+def prepare_junction_project(cli, tmpdir):
+    main_project = tmpdir.join("main")
+    sub_project = tmpdir.join("sub")
+    os.makedirs(str(main_project))
+    os.makedirs(str(sub_project))
+
+    _yaml.dump({'name': 'main'}, str(main_project.join("project.conf")))
+    _yaml.dump({'name': 'sub'}, str(sub_project.join("project.conf")))
+
+    import_dir = tmpdir.join("import")
+    os.makedirs(str(import_dir))
+    with open(str(import_dir.join("hello.txt")), "w") as f:
+        f.write("hello!")
+
+    import_repo_dir = tmpdir.join("import_repo")
+    os.makedirs(str(import_repo_dir))
+    import_repo = create_repo("git", str(import_repo_dir))
+    import_ref = import_repo.create(str(import_dir))
+
+    _yaml.dump({'kind': 'import',
+                'sources': [import_repo.source_config(ref=import_ref)]},
+               str(sub_project.join("data.bst")))
+
+    sub_repo_dir = tmpdir.join("sub_repo")
+    os.makedirs(str(sub_repo_dir))
+    sub_repo = create_repo("git", str(sub_repo_dir))
+    sub_ref = sub_repo.create(str(sub_project))
+
+    _yaml.dump({'kind': 'junction',
+                'sources': [sub_repo.source_config(ref=sub_ref)]},
+               str(main_project.join("sub.bst")))
+
+    args = ['fetch', 'sub.bst']
+    result = cli.run(project=str(main_project), args=args)
+    result.assert_success()
+
+    return str(main_project)
+
+
+def open_cross_junction(cli, tmpdir):
+    project = prepare_junction_project(cli, tmpdir)
+    workspace = tmpdir.join("workspace")
+
+    element = 'sub.bst:data.bst'
+    args = ['workspace', 'open', element, str(workspace)]
+    result = cli.run(project=project, args=args)
+    result.assert_success()
+
+    assert cli.get_element_state(project, element) == 'buildable'
+    assert os.path.exists(str(workspace.join('hello.txt')))
+
+    return project, workspace
+
+
+def test_open_cross_junction(cli, tmpdir):
+    open_cross_junction(cli, tmpdir)
+
+
+def test_list_cross_junction(cli, tmpdir):
+    project, workspace = open_cross_junction(cli, tmpdir)
+
+    element = 'sub.bst:data.bst'
+
+    args = ['workspace', 'list']
+    result = cli.run(project=project, args=args)
+    result.assert_success()
+
+    loaded = _yaml.load_data(result.output)
+    assert isinstance(loaded.get('workspaces'), list)
+    workspaces = loaded['workspaces']
+    assert len(workspaces) == 1
+    assert 'element' in workspaces[0]
+    assert workspaces[0]['element'] == element
+
+
+def test_close_cross_junction(cli, tmpdir):
+    project, workspace = open_cross_junction(cli, tmpdir)
+
+    element = 'sub.bst:data.bst'
+    args = ['workspace', 'close', '--remove-dir', element]
+    result = cli.run(project=project, args=args)
+    result.assert_success()
+
+    assert not os.path.exists(str(workspace))
+
+    args = ['workspace', 'list']
+    result = cli.run(project=project, args=args)
+    result.assert_success()
+
+    loaded = _yaml.load_data(result.output)
+    assert isinstance(loaded.get('workspaces'), list)
+    workspaces = loaded['workspaces']
+    assert len(workspaces) == 0
+
+
+def test_close_all_cross_junction(cli, tmpdir):
+    project, workspace = open_cross_junction(cli, tmpdir)
+
+    args = ['workspace', 'close', '--remove-dir', '--all']
+    result = cli.run(project=project, args=args)
+    result.assert_success()
+
+    assert not os.path.exists(str(workspace))
+
+    args = ['workspace', 'list']
+    result = cli.run(project=project, args=args)
+    result.assert_success()
+
+    loaded = _yaml.load_data(result.output)
+    assert isinstance(loaded.get('workspaces'), list)
+    workspaces = loaded['workspaces']
+    assert len(workspaces) == 0