Merge pull request #516 from jayvdb/allow-sqlite-testing

Allow running tests using sqlite
jazzband · Mar 21, 2024 · 6e7158b · 6e7158b
2 parents 89b7662 + 7096314
commit 6e7158b
Showing 1 changed file with 19 additions and 10 deletions.
diff --git a/tests/settings.py b/tests/settings.py
@@ -4,16 +4,25 @@
     'model_utils',
     'tests',
 )
-DATABASES = {
-    "default": {
-        "ENGINE": "django.db.backends.postgresql_psycopg2",
-        "NAME": os.environ.get("DB_NAME", "modelutils"),
-        "USER": os.environ.get("DB_USER", 'postgres'),
-        "PASSWORD": os.environ.get("DB_PASSWORD", ""),
-        "HOST": os.environ.get("DB_HOST", "localhost"),
-        "PORT": os.environ.get("DB_PORT", 5432)
-    },
-}
+
+if os.environ.get('SQLITE'):
+    DATABASES = {
+        'default': {
+            'ENGINE': 'django.db.backends.sqlite3',
+        },
+    }
+else:
+    DATABASES = {
+        "default": {
+            "ENGINE": "django.db.backends.postgresql",
+            "NAME": os.environ.get("DB_NAME", "modelutils"),
+            "USER": os.environ.get("DB_USER", 'postgres'),
+            "PASSWORD": os.environ.get("DB_PASSWORD", ""),
+            "HOST": os.environ.get("DB_HOST", "localhost"),
+            "PORT": os.environ.get("DB_PORT", 5432)
+        },
+    }
+
 SECRET_KEY = 'dummy'
 
 CACHES = {