diff options
author | Jürg Billeter <j@bitron.ch> | 2019-07-09 17:22:17 +0200 |
---|---|---|
committer | Jürg Billeter <j@bitron.ch> | 2019-07-16 15:36:10 +0200 |
commit | fdba3062724675a4d55f93898d1604100d6e12b5 (patch) | |
tree | 648b2770bb8dc52c85c93ba2571c8e81ff5aafc3 /tests | |
parent | 36737b18cedc754d7b8b3c2f513bb44f5c30ee43 (diff) | |
download | buildstream-fdba3062724675a4d55f93898d1604100d6e12b5.tar.gz |
tests/sourcecache/push.py: Use dummy_context()
Diffstat (limited to 'tests')
-rw-r--r-- | tests/sourcecache/push.py | 55 |
1 files changed, 26 insertions, 29 deletions
diff --git a/tests/sourcecache/push.py b/tests/sourcecache/push.py index e9c72d47c..3fe6d78f7 100644 --- a/tests/sourcecache/push.py +++ b/tests/sourcecache/push.py @@ -23,14 +23,13 @@ import os import shutil import pytest -from buildstream._context import Context from buildstream._exceptions import ErrorDomain from buildstream._project import Project from buildstream import _yaml from buildstream.testing import cli # pylint: disable=unused-import from buildstream.testing import create_repo -from tests.testutils import create_artifact_share +from tests.testutils import create_artifact_share, dummy_context DATA_DIR = os.path.join(os.path.dirname(os.path.realpath(__file__)), "project") @@ -70,33 +69,31 @@ def test_source_push(cli, tmpdir, datafiles): _yaml.roundtrip_dump(element, os.path.join(element_path, element_name)) # get the source object - context = Context() - context.load(config=user_config_file) - context.messenger.set_message_handler(message_handler) - project = Project(project_dir, context) - project.ensure_fully_loaded() - - element = project.load_elements(['push.bst'])[0] - assert not element._source_cached() - source = list(element.sources())[0] - - # check we don't have it in the current cache - cas = context.get_cascache() - assert not cas.contains(source._get_source_name()) - - # build the element, this should fetch and then push the source to the - # remote - res = cli.run(project=project_dir, args=['build', 'push.bst']) - res.assert_success() - assert "Pushed source" in res.stderr - - # check that we've got the remote locally now - sourcecache = context.sourcecache - assert sourcecache.contains(source) - - # check that's the remote CAS now has it - digest = sourcecache.export(source)._get_digest() - assert share.has_object(digest) + with dummy_context(config=user_config_file) as context: + project = Project(project_dir, context) + project.ensure_fully_loaded() + + element = project.load_elements(['push.bst'])[0] + assert not element._source_cached() + source = list(element.sources())[0] + + # check we don't have it in the current cache + cas = context.get_cascache() + assert not cas.contains(source._get_source_name()) + + # build the element, this should fetch and then push the source to the + # remote + res = cli.run(project=project_dir, args=['build', 'push.bst']) + res.assert_success() + assert "Pushed source" in res.stderr + + # check that we've got the remote locally now + sourcecache = context.sourcecache + assert sourcecache.contains(source) + + # check that's the remote CAS now has it + digest = sourcecache.export(source)._get_digest() + assert share.has_object(digest) @pytest.mark.datafiles(DATA_DIR) |