- feat: add custom test runner

Author: koladev32

core/settings.py

@@ -132,14 +132,28 @@
 
 AUTH_USER_MODEL = 'api_user.User'
 
+# ##################################################################### #
+# ################### REST FRAMEWORK             ###################### #
+# ##################################################################### #
+
 REST_FRAMEWORK = {
     'DEFAULT_AUTHENTICATION_CLASSES': (
         'api.authentication.backends.ActiveSessionAuthentication',
     ),
 }
 
+# ##################################################################### #
+# ################### CORS              ############################### #
+# ##################################################################### #
+
 CORS_ALLOWED_ORIGINS = [
     "http://localhost:3000",
     "http://127.0.0.1:3000"
 ]
 
+# ##################################################################### #
+# ################### TESTING           ############################### #
+# ##################################################################### #
+
+TESTING = False
+TEST_RUNNER = 'core.test_runner.CoreTestRunner'

core/test_runner.py

@@ -0,0 +1,27 @@
+from importlib import import_module
+
+from django.conf import settings
+from django.db import connections
+from django.test.runner import DiscoverRunner
+
+
+class CoreTestRunner(DiscoverRunner):
+    def setup_test_environment(self, **kwargs):
+        """We set the TESTING setting to True. By default, it's on False."""
+        super().setup_test_environment(**kwargs)
+        settings.TESTING = True
+
+    def setup_databases(self, **kwargs):
+        """We set the database"""
+        kwargs['aliases'] = connections
+        r = super().setup_databases(**kwargs)
+        self.load_fixtures()
+        return r
+
+    @classmethod
+    def load_fixtures(cls):
+        try:
+            module = import_module(f'api.fixtures')
+            getattr(module, 'run_fixtures')()
+        except ImportError:
+            return