added Model factory

hotosm · Apr 2, 2024 · b8672fc · b8672fc
1 parent e0d5574
commit b8672fc
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/backend/tests/factories.py b/backend/tests/factories.py
@@ -1,7 +1,7 @@
 import factory
 
 from login.models import OsmUser
-from core.models import Dataset, AOI, Label
+from core.models import Dataset, AOI, Label, Model
 
 
 class OsmUserFactory(factory.django.DjangoModelFactory):
@@ -37,3 +37,14 @@ class Meta:
 
     aoi = factory.SubFactory(AoiFactory)
     geom = "POLYGON ((0 0, 0 1, 1 1, 1 0, 0 0))"
+
+
+class ModelFactory(factory.django.DjangoModelFactory):
+
+    class Meta:
+        model = Model
+
+    dataset = factory.SubFactory(DatasetFactory)
+    name = "Test Model"
+    created_by = factory.SubFactory(OsmUserFactory)
+    status = -1