first commit

backend/apps/farms/tests.py

@@ -0,0 +1,82 @@
+from datetime import date, timedelta
+
+from django.test import TestCase
+from rest_framework.test import APIClient
+
+from apps.accounts.models import User
+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.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):
+        start = date.today()
+        response = self.client.post(
+            "/api/v1/cycles/",
+            {
+                "kandang": self.kandang.pk,
+                "total_days": 35,
+                "start_date": start.isoformat(),
+                "end_date": (start + timedelta(days=34)).isoformat(),
+                "chick_in_weight": 42,
+                "doc_in_count": 10000,
+                "status": "active",
+            },
+            format="json",
+        )
+        self.assertEqual(response.status_code, 201, response.data)
+        self.assertEqual(response.data["chick_in_weight"], 42)
+        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)
+
+    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)
+
+    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})
+        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])