blob: afd9f6265e61fc9c8ae0c82be28322703939bb96 [file] [log] [blame]
# Pylint doesn't play well with fixtures and dependency injection from pytest
# pylint: disable=redefined-outer-name
import os
import shutil
import pytest
from buildstream.testing import cli # pylint: disable=unused-import
from buildstream.exceptions import ErrorDomain
from tests.testutils import create_artifact_share
DATA_DIR = os.path.join(os.path.dirname(os.path.realpath(__file__)), "project")
# Test modes of `bst artifact pull` when given an artifact
# When pulling an artifact with --deps none, we expect that artifact to be pulled
("none", ["target.bst"]),
# When pulling an artifact with --deps build, we expect the build deps to be pulled
("build", ["import-bin.bst", "compose-all.bst"]),
# Pulling an artifact with --deps run is not supported without a local project
("run", []),
# Pulling an artifact with --deps all is not supported without a local project
("all", []),
ids=["none", "build", "run", "all"],
@pytest.mark.parametrize("with_project", [True, False], ids=["with-project", "without-project"])
def test_pull(cli, tmpdir, datafiles, deps, expect_cached, with_project):
project = str(datafiles)
with create_artifact_share(os.path.join(str(tmpdir), "artifactshare")) as share:
# Build the element to push it to cache, and explicitly configure local cache so we can check it
local_cache = os.path.join(str(tmpdir), "cache")
cli.configure({"cachedir": local_cache, "artifacts": {"servers": [{"url": share.repo, "push": True}]}})
# Build it
result =, args=["build", "target.bst"])
# Assert it is cached locally and remotely
assert cli.get_element_state(project, "target.bst") == "cached"
assert share.get_artifact(cli.get_artifact_name(project, "test", "target.bst"))
# Obtain the artifact name for pulling purposes
artifact_name = cli.get_artifact_name(project, "test", "target.bst")
# Translate the expected element names into artifact names
expect_cached_artifacts = [
cli.get_artifact_name(project, "test", element_name) for element_name in expect_cached
# Discard the local cache
shutil.rmtree(str(os.path.join(str(tmpdir), "cache", "cas")))
shutil.rmtree(str(os.path.join(str(tmpdir), "cache", "artifacts")))
assert cli.get_element_state(project, "target.bst") != "cached"
# Delete the project.conf if we're going to try this without a project
if not with_project:
os.remove(os.path.join(project, "project.conf"))
# Now run our pull test
result =, args=["artifact", "pull", "--deps", deps, artifact_name])
if deps in ["all", "run"]:
result.assert_main_error(ErrorDomain.STREAM, "deps-not-supported")
# After pulling, assert that we have the expected elements cached again.
# Note that we do not use cli.get_element_states() here because the project.conf
# might not be present, so we poke at the cache directly for this assertion.
for expect in expect_cached_artifacts:
assert os.path.exists(os.path.join(local_cache, "artifacts", "refs", expect))