Move test-only config to corresponding fixtures

freedomofpress · Nov 13, 2021 · 8f3c8e5 · 8f3c8e5
1 parent 194797f
commit 8f3c8e5
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/securedrop/tests/conftest.py b/securedrop/tests/conftest.py
@@ -196,6 +196,10 @@ def alembic_config(config: SDConfig) -> str:
 
 @pytest.fixture(scope='function')
 def source_app(config: SDConfig) -> Generator[Flask, None, None]:
+    config.SOURCE_APP_FLASK_CONFIG_CLS.TESTING = True
+    # Disable CSRF checks to make writing tests easier
+    config.SOURCE_APP_FLASK_CONFIG_CLS.WTF_CSRF_ENABLED = False    
+
     app = create_source_app(config)
     app.config['SERVER_NAME'] = 'localhost.localdomain'
     with app.app_context():
@@ -209,6 +213,10 @@ def source_app(config: SDConfig) -> Generator[Flask, None, None]:
 
 @pytest.fixture(scope='function')
 def journalist_app(config: SDConfig) -> Generator[Flask, None, None]:
+    config.JOURNALIST_APP_FLASK_CONFIG_CLS.TESTING = True
+    # Disable CSRF checks to make writing tests easier
+    config.JOURNALIST_APP_FLASK_CONFIG_CLS.WTF_CSRF_ENABLED = False    
+
     app = create_journalist_app(config)
     app.config['SERVER_NAME'] = 'localhost.localdomain'
     with app.app_context():