Fix tests

.deepsource.toml

@@ -1,7 +1,9 @@
 version = 1
 
+exclude_patterns = ["tests/**", "db_init.py"]
+
 [[analyzers]]
 name = "python"
 
   [analyzers.meta]
-  runtime_version = "3.x.x"
+  runtime_version = "3.x.x"

.github/workflows/pytest.yml

@@ -50,12 +50,12 @@ jobs:
 
       - name: Set up the Database
         env:
-          DATABASE_URL: postgresql+psycopg2://postgres:postgres@localhost:5432/labconnect
+          DB: postgresql+psycopg2://postgres:postgres@localhost:5432/labconnect
         run: |
           python db_init.py create
 
       - name: Running pytest
         env:
-          DATABASE_URL: postgresql+psycopg2://postgres:postgres@localhost:5432/labconnect
+          DB: postgresql+psycopg2://postgres:postgres@localhost:5432/labconnect
         run: |
           python -m pytest tests/

config.py

@@ -3,9 +3,15 @@
 from os import getenv, path
 
 from dotenv import load_dotenv
+from sqlalchemy.pool import StaticPool
 
 basedir = path.abspath(path.dirname(__file__))
 
+_test_db_uri = getenv(
+    "DATABASE_URL",
+    getenv("DB", "postgresql+psycopg2://postgres:root@localhost/labconnect"),
+)
+
 
 class Config:
     load_dotenv()
@@ -25,9 +31,7 @@ class Config:
     SENTRY_TRACES_SAMPLE_RATE = float(getenv("SENTRY_TRACES_SAMPLE_RATE", 1.0))
     SENTRY_PROFILES_SAMPLE_RATE = float(getenv("SENTRY_PROFILES_SAMPLE_RATE", 1.0))
 
-    SQLALCHEMY_DATABASE_URI = getenv(
-        "DB", "postgresql+psycopg2://postgres:root@localhost/labconnect"
-    )
+    SQLALCHEMY_DATABASE_URI = _test_db_uri
 
     JWT_SECRET_KEY = getenv("JWT_SECRET_KEY", "jwt-secret")
     JWT_ACCESS_TOKEN_EXPIRES = timedelta(hours=1)
@@ -47,6 +51,12 @@ class TestingConfig(Config):
     DEBUG = True
     JWT_COOKIE_SECURE = False
 
+    if _test_db_uri.startswith("sqlite"):
+        SQLALCHEMY_ENGINE_OPTIONS = {
+            "connect_args": {"check_same_thread": False},
+            "poolclass": StaticPool,
+        }
+
 
 class ProductionConfig(Config):
     TESTING = False