code refactor to fix duplication in pytest fixtures

config.py

@@ -6,7 +6,6 @@ class Config:
 
     # general config
     SECRET_KEY = os.environ.get("FLASK_SECRET_KEY")
-    PROPAGATE_EXCEPTIONS = True
 
     # database
     SQLALCHEMY_DATABASE_URI = os.environ.get("DATABASE_URL")

tests/conftest_wrapper.py

@@ -0,0 +1,14 @@
+from config import Config as config_prod
+from config_dev import Config as config_dev
+from application import create_app
+
+
+def wrapper_client(branch):
+    if branch == "dev":
+        app_config = config_dev
+    else:
+        app_config = config_prod
+
+    flask_app = create_app(app_config)
+    testing_client = flask_app.test_client()
+    return testing_client

tests/test_dev/conftest.py

@@ -1,7 +1,5 @@
 import pytest
-from dotenv import load_dotenv
-from application import create_app
-from config_dev import Config
+from tests.conftest_wrapper import wrapper_client
 
 # holds all the fixtures
 # fixtures are ran once during request, this client will be passed in
@@ -10,6 +8,4 @@
 
 @pytest.fixture(scope="module")
 def client(request):
-    flask_app = create_app(Config)
-    testing_client = flask_app.test_client()
-    return testing_client
+    return wrapper_client("dev")

tests/test_prod/conftest.py

@@ -1,7 +1,5 @@
 import pytest
-from dotenv import load_dotenv
-from application import create_app
-from config import Config
+from tests.conftest_wrapper import wrapper_client
 
 # holds all the fixtures
 # fixtures are ran once during request, this client will be passed in
@@ -10,6 +8,4 @@
 
 @pytest.fixture(scope="module")
 def client(request):
-    flask_app = create_app(Config)
-    testing_client = flask_app.test_client()
-    return testing_client
+    return wrapper_client("prod")