1 files changed, 83 insertions, 0 deletions
diff --git a/buildstream/testing/_sourcetests/build_checkout.py b/buildstream/testing/_sourcetests/build_checkout.py
new file mode 100644
index 000000000..3619d2b7e
--- /dev/null
+++ b/buildstream/testing/_sourcetests/build_checkout.py
@@ -0,0 +1,83 @@
+#
+#  Copyright (C) 2018 Codethink Limited
+#  Copyright (C) 2019 Bloomberg Finance LP
+#
+#  This program is free software; you can redistribute it and/or
+#  modify it under the terms of the GNU Lesser General Public
+#  License as published by the Free Software Foundation; either
+#  version 2 of the License, or (at your option) any later version.
+#
+#  This library is distributed in the hope that it will be useful,
+#  but WITHOUT ANY WARRANTY; without even the implied warranty of
+#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+#  Lesser General Public License for more details.
+#
+#  You should have received a copy of the GNU Lesser General Public
+#  License along with this library. If not, see <http://www.gnu.org/licenses/>.
+#
+
+# 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 create_repo, ALL_REPO_KINDS
+from buildstream.testing import cli  # pylint: disable=unused-import
+from buildstream import _yaml
+
+# Project directory
+TOP_DIR = os.path.dirname(os.path.realpath(__file__))
+DATA_DIR = os.path.join(TOP_DIR, 'project')
+
+fetch_build_checkout_combos = \
+    [("strict", kind) for kind in ALL_REPO_KINDS] + \
+    [("non-strict", kind) for kind in ALL_REPO_KINDS]
+
+
+def strict_args(args, strict):
+    if strict != "strict":
+        return ['--no-strict', *args]
+    return args
+
+
+@pytest.mark.datafiles(DATA_DIR)
+@pytest.mark.parametrize("strict,kind", fetch_build_checkout_combos)
+def test_fetch_build_checkout(cli, tmpdir, datafiles, strict, kind):
+    checkout = os.path.join(cli.directory, 'checkout')
+    project = str(datafiles)
+    dev_files_path = os.path.join(project, 'files', 'dev-files')
+    element_path = os.path.join(project, 'elements')
+    element_name = 'build-test-{}.bst'.format(kind)
+
+    # Create our repo object of the given source type with
+    # the dev files, and then collect the initial ref.
+    #
+    repo = create_repo(kind, str(tmpdir))
+    ref = repo.create(dev_files_path)
+
+    # Write out our test target
+    element = {
+        'kind': 'import',
+        'sources': [
+            repo.source_config(ref=ref)
+        ]
+    }
+    _yaml.dump(element,
+               os.path.join(element_path,
+                            element_name))
+
+    assert cli.get_element_state(project, element_name) == 'fetch needed'
+    result = cli.run(project=project, args=strict_args(['build', element_name], strict))
+    result.assert_success()
+    assert cli.get_element_state(project, element_name) == 'cached'
+
+    # Now check it out
+    result = cli.run(project=project, args=strict_args([
+        'artifact', 'checkout', element_name, '--directory', checkout
+    ], strict))
+    result.assert_success()
+
+    # Check that the pony.h include from files/dev-files exists
+    filename = os.path.join(checkout, 'usr', 'include', 'pony.h')
+    assert os.path.exists(filename)