diff options
Diffstat (limited to 'tests/testutils/context.py')
-rw-r--r-- | tests/testutils/context.py | 46 |
1 files changed, 46 insertions, 0 deletions
diff --git a/tests/testutils/context.py b/tests/testutils/context.py new file mode 100644 index 000000000..899bad247 --- /dev/null +++ b/tests/testutils/context.py @@ -0,0 +1,46 @@ +# +# 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/>. + +import os + +from contextlib import contextmanager + +from buildstream._context import Context + + +# Handle messages from the pipeline +def _dummy_message_handler(message, is_silenced): + pass + + +# dummy_context() +# +# Context manager to create minimal context for tests. +# +# Args: +# config (filename): The configuration file, if any +# +@contextmanager +def dummy_context(*, config=None): + with Context() as context: + if not config: + config = os.devnull + + context.load(config=config) + + context.messenger.set_message_handler(_dummy_message_handler) + + yield context |