test: move ConfigForTesting

2017-02-03 16:04:31 -05:00 · 2017-02-03 16:04:31 -05:00 · dd033e4feb
commit dd033e4feb
parent 1d59095460
2 changed files with 12 additions and 12 deletions
--- a/test/test_api_usage.py
+++ b/test/test_api_usage.py
@ -117,6 +117,16 @@ FAKE_APPLICATION_CLIENT_ID = 'deadbeef'
 CSRF_TOKEN_KEY = '_csrf_token'
 CSRF_TOKEN = '123csrfforme'

+
+class ConfigForTesting(object):
+  def __enter__(self):
+    config_provider.reset_for_test()
+    return config_provider
+
+  def __exit__(self, type, value, traceback):
+    config_provider.reset_for_test()
+
+
 class ApiTestCase(unittest.TestCase):
  maxDiff = None