remove wagtail and admin website, remove edit option, from hitung ayam, add create flock to settings

backend/apps/farms/tests.py

@@ -10,12 +10,37 @@ from apps.farms.models import Cycle, Flock, Kandang, Site
 class FarmAPITests(TestCase):
     def setUp(self):
         self.user = User.objects.create_user(user_name="tester", password="secret")
+        self.admin = User.objects.create_user(
+            user_name="admin",
+            password="secret",
+            status=User.STATUS_SUPERADMIN,
+        )
         self.client = APIClient()
-        self.client.force_authenticate(self.user)
         self.site = Site.objects.create(site_name="Sukawarna", user=self.user)
         self.kandang = Kandang.objects.create(kandang_name="Kandang 1", site=self.site)
 
-    def test_create_cycle_auto_creates_flock_and_returns_erd_fields(self):
+    def test_create_flock_under_kandang_requires_superadmin(self):
+        self.client.force_authenticate(self.admin)
+        response = self.client.post(
+            "/api/v1/flocks/",
+            {"flock_name": "Lantai 1", "kandang": self.kandang.pk},
+            format="json",
+        )
+        self.assertEqual(response.status_code, 201, response.data)
+        self.assertEqual(response.data["flock_name"], "Lantai 1")
+        self.assertEqual(response.data["kandang"], self.kandang.pk)
+
+    def test_staff_cannot_create_flock(self):
+        self.client.force_authenticate(self.user)
+        response = self.client.post(
+            "/api/v1/flocks/",
+            {"flock_name": "Lantai 1", "kandang": self.kandang.pk},
+            format="json",
+        )
+        self.assertEqual(response.status_code, 403, response.data)
+
+    def test_create_cycle_without_flock(self):
+        self.client.force_authenticate(self.user)
         start = date.today()
         response = self.client.post(
             "/api/v1/cycles/",
@@ -31,56 +56,32 @@ class FarmAPITests(TestCase):
             format="json",
         )
         self.assertEqual(response.status_code, 201, response.data)
-        self.assertEqual(response.data["chick_in_weight"], 42)
-        self.assertEqual(response.data["kandang_name"], "Kandang 1")
-        self.assertEqual(response.data["flock_name"], "Kandang 1")
-        self.assertIsNotNone(response.data["flock"])
-        cycle = Cycle.objects.get(pk=response.data["id"])
-        flock = Flock.objects.get(pk=response.data["flock"])
-        self.assertEqual(cycle.doc_in_weight, 42)
-        self.assertEqual(flock.cycle_id, cycle.pk)
+        self.assertNotIn("flock", response.data)
+        self.assertNotIn("flock_name", response.data)
 
-    def test_list_cycles_by_kandang(self):
-        cycle = Cycle.objects.create(
-            kandang=self.kandang,
-            total_days=35,
-            start_date=date.today(),
-            end_date=date.today() + timedelta(days=34),
-            doc_in_weight=40,
-            doc_in_count=9000,
-            status=Cycle.STATUS_ACTIVE,
-        )
-        flock = Flock.objects.create(flock_name="Flock A", cycle=cycle)
-        response = self.client.get("/api/v1/cycles/", {"kandang_id": self.kandang.pk})
-        self.assertEqual(response.status_code, 200)
-        results = response.data["results"] if isinstance(response.data, dict) else response.data
-        self.assertEqual(len(results), 1)
-        self.assertEqual(results[0]["flock"], flock.pk)
-        self.assertEqual(results[0]["kandang_name"], "Kandang 1")
-        self.assertEqual(results[0]["flock_name"], "Flock A")
+    def test_flock_persists_across_cycles(self):
+        flock = Flock.objects.create(flock_name="Lantai 1", kandang=self.kandang)
+        start = date.today()
+        for offset in (0, 40):
+            Cycle.objects.create(
+                kandang=self.kandang,
+                total_days=35,
+                start_date=start + timedelta(days=offset),
+                end_date=start + timedelta(days=offset + 34),
+                doc_in_weight=40,
+                doc_in_count=9000,
+                status=Cycle.STATUS_ACTIVE,
+            )
+        self.assertEqual(Flock.objects.filter(kandang=self.kandang).count(), 1)
+        self.assertEqual(flock.kandang_id, self.kandang.pk)
 
-    def test_list_flocks_by_cycle(self):
-        cycle = Cycle.objects.create(
-            kandang=self.kandang,
-            total_days=35,
-            start_date=date.today(),
-            end_date=date.today() + timedelta(days=34),
-            doc_in_weight=40,
-            doc_in_count=9000,
-            status=Cycle.STATUS_ACTIVE,
-        )
-        flock = Flock.objects.create(flock_name="Flock A", cycle=cycle)
-        other_cycle = Cycle.objects.create(
-            kandang=self.kandang,
-            total_days=35,
-            start_date=date.today(),
-            end_date=date.today() + timedelta(days=34),
-            doc_in_weight=40,
-            doc_in_count=9000,
-            status=Cycle.STATUS_ACTIVE,
-        )
-        Flock.objects.create(flock_name="Flock B", cycle=other_cycle)
-        response = self.client.get("/api/v1/flocks/", {"cycle_id": cycle.pk})
+    def test_list_flocks_by_kandang(self):
+        self.client.force_authenticate(self.user)
+        Flock.objects.create(flock_name="Lantai 1", kandang=self.kandang)
+        Flock.objects.create(flock_name="Lantai 2", kandang=self.kandang)
+        other_kandang = Kandang.objects.create(kandang_name="Kandang 2", site=self.site)
+        Flock.objects.create(flock_name="Other", kandang=other_kandang)
+        response = self.client.get("/api/v1/flocks/", {"kandang_id": self.kandang.pk})
         self.assertEqual(response.status_code, 200, response.data)
         results = response.data["results"] if isinstance(response.data, dict) else response.data
-        self.assertEqual([r["id"] for r in results], [flock.pk])
+        self.assertEqual(len(results), 2)