tests: Extract TestCase class

backend/tests/__init__.py

@@ -0,0 +1,25 @@
+from django.db import connections
+from django.test import TransactionTestCase
+
+from api.models import DataItem, Dataset, FieldLevelCheck, ProgressMonitorDataset
+
+
+class TestCase(TransactionTestCase):
+    databases = {"default", "data"}
+    unmanaged = {DataItem, Dataset, FieldLevelCheck, ProgressMonitorDataset}
+
+    def setUp(self):
+        with connections["data"].schema_editor() as schema_editor:
+            for model in self.unmanaged:
+                schema_editor.create_model(model)
+
+    def tearDown(self):
+        with connections["data"].schema_editor() as schema_editor:
+            for model in self.unmanaged:
+                schema_editor.delete_model(model)
+
+    def create(self, model, **kwargs):
+        obj = model(**kwargs)
+        obj.full_clean()
+        obj.save()
+        return obj

backend/tests/processor/test_views.py → backend/tests/controller/test_views.py

@@ -1,31 +1,10 @@
 from unittest.mock import patch
 
-from django.db import connections
-from django.test import TransactionTestCase
-
 from api.models import DataItem, Dataset, FieldLevelCheck, ProgressMonitorDataset
+from tests import TestCase
 
 
-class ViewsTests(TransactionTestCase):
-    databases = {"default", "data"}
-    unmanaged = {DataItem, Dataset, FieldLevelCheck, ProgressMonitorDataset}
-
-    def setUp(self):
-        with connections["data"].schema_editor() as schema_editor:
-            for model in self.unmanaged:
-                schema_editor.create_model(model)
-
-    def tearDown(self):
-        with connections["data"].schema_editor() as schema_editor:
-            for model in self.unmanaged:
-                schema_editor.delete_model(model)
-
-    def create(self, model, **kwargs):
-        obj = model(**kwargs)
-        obj.full_clean()
-        obj.save()
-        return obj
-
+class ViewsTests(TestCase):
     @patch("controller.views.publish")
     def test_create_invalid(self, publish):
         response = self.client.post("/datasets/", {"name": "anything", "collection_id": "xxx"}, "application/json")