added factory boy

hotosm · Apr 2, 2024 · 1a9b419 · 1a9b419
1 parent 556c162
commit 1a9b419
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/backend/tests/factories.py b/backend/tests/factories.py
@@ -0,0 +1,21 @@
+import factory
+
+from login.models import OsmUser
+from core.models import Dataset
+
+
+class OsmUserFactory(factory.django.DjangoModelFactory):
+    class Meta:
+        model = OsmUser
+
+    username = "Test User"
+    osm_id = 123456
+
+
+class DatasetFactory(factory.django.DjangoModelFactory):
+    class Meta:
+        model = Dataset
+
+    name = "Test Dataset"
+    created_by = factory.SubFactory(OsmUserFactory)
+    status = -1