From 557420060748dcd4e2356f2723477a6906a3cdb7 Mon Sep 17 00:00:00 2001 From: Athanasios Voutsadakis Date: Tue, 14 Nov 2017 14:51:17 +0000 Subject: [PATCH] Copy set_config_values from api --- tests/conftest.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/tests/conftest.py b/tests/conftest.py index 2a1958781..860633847 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -2457,6 +2457,19 @@ def set_config(app, name, value): app.config[name] = old_val +@contextmanager +def set_config_values(app, dict): + old_values = {} + + for key in dict: + old_values[key] = app.config.get(key) + app.config[key] = dict[key] + + yield + + for key in dict: + app.config[key] = old_values[key] + @pytest.fixture(scope='function') def valid_token(app_, fake_uuid): return generate_token(