Test/label-model

backend/tests/factories.py

@@ -0,0 +1,39 @@
+import factory
+
+from login.models import OsmUser
+from core.models import Dataset, AOI, Label
+
+
+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
+
+
+class AoiFactory(factory.django.DjangoModelFactory):
+    class Meta:
+        model = AOI
+
+    geom = "POLYGON ((0 0, 0 1, 1 1, 1 0, 0 0))"
+    dataset = factory.SubFactory(DatasetFactory)
+    label_status = -1
+
+
+class LabelFactory(factory.django.DjangoModelFactory):
+
+    class Meta:
+        model = Label
+
+    aoi = factory.SubFactory(AoiFactory)
+    geom = "POLYGON ((0 0, 0 1, 1 1, 1 0, 0 0))"

backend/tests/test_core_model.py

@@ -0,0 +1,23 @@
+from django.test import TestCase
+
+from .factories import DatasetFactory, OsmUserFactory, AoiFactory, LabelFactory
+
+
+class TestCoreModels(TestCase):
+
+    def setUp(self):
+        self.user = OsmUserFactory(username="Test User 2", osm_id=123)
+        self.dataset = DatasetFactory(created_by=self.user)
+        self.aoi = AoiFactory(dataset=self.dataset)
+        self.label = LabelFactory(aoi=self.aoi)
+
+    def test_dataset_creation(self):
+        self.assertEqual(self.dataset.name, "Test Dataset")
+        self.assertEqual(self.dataset.created_by, self.user)
+
+    def test_aoi_creation(self):
+        self.assertEqual(self.aoi.dataset, self.dataset)
+        self.assertEqual(self.aoi.label_status, -1)
+
+    def test_label_creation(self):
+        self.assertEqual(self.label.aoi, self.aoi)

backend/tests/test_login_model.py

@@ -0,0 +1,12 @@
+from django.test import TestCase
+
+from .factories import OsmUserFactory
+
+
+class TestLoginModels(TestCase):
+
+    def test_OsmUser_creation(self):
+        osm_user = OsmUserFactory()
+
+        self.assertEqual(str(osm_user), "Test User")
+        self.assertEqual(osm_user.osm_id, 123456)