added aoi factory

hotosm · Apr 2, 2024 · f8d78da · f8d78da
1 parent 4aba08c
commit f8d78da
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,9 @@
 import factory
 
 from login.models import OsmUser
-from core.models import Dataset
+from core.models import Dataset, AOI
+
+from django.contrib.gis.geos import GEOSGeometry
 
 
 class OsmUserFactory(factory.django.DjangoModelFactory):
@@ -19,3 +21,12 @@ class Meta:
     name = "Test Dataset"
     created_by = factory.SubFactory(OsmUserFactory)
     status = -1
+
+
+class AoiFactory(factory.django.DjangoModelFactory):
+    class Meta:
+        model = AOI
+
+    geom = GEOSGeometry("POLYGON ((0 0, 0 1, 1 1, 1 0, 0 0))", srid=4326)
+    dataset = factory.SubFactory(DatasetFactory)
+    label_status = -1