fix initial balance logic and add edit for cycle data

backend/apps/farms/tests.py

@@ -61,6 +61,57 @@ class FarmAPITests(TestCase):
         self.assertNotIn("flock", response.data)
         self.assertNotIn("flock_name", response.data)
 
+    def test_update_cycle(self):
+        self.client.force_authenticate(self.user)
+        start = date(2026, 8, 1)
+        cycle = Cycle.objects.create(
+            kandang=self.kandang,
+            total_days=35,
+            start_date=start,
+            end_date=start + timedelta(days=34),
+            doc_in_weight=40,
+            doc_in_count=9000,
+            status=Cycle.STATUS_ACTIVE,
+        )
+        new_end = start + timedelta(days=39)
+        response = self.client.patch(
+            f"/api/v1/cycles/{cycle.pk}/",
+            {
+                "end_date": new_end.isoformat(),
+                "total_days": 40,
+                "doc_in_count": 9500,
+                "chick_in_weight": 42,
+            },
+            format="json",
+        )
+        self.assertEqual(response.status_code, 200, response.data)
+        cycle.refresh_from_db()
+        self.assertEqual(cycle.end_date, new_end)
+        self.assertEqual(cycle.total_days, 40)
+        self.assertEqual(cycle.doc_in_count, 9500)
+        self.assertEqual(cycle.doc_in_weight, 42)
+
+    @patch("apps.operations.services.visibility.visible_through_date", return_value=date(2026, 9, 10))
+    def test_cycle_status_closes_when_end_date_is_visible(self, _mock_visible):
+        self.client.force_authenticate(self.user)
+        start = date(2026, 8, 1)
+        cycle = Cycle.objects.create(
+            kandang=self.kandang,
+            total_days=35,
+            start_date=start,
+            end_date=start + timedelta(days=34),
+            doc_in_weight=40,
+            doc_in_count=9000,
+            status=Cycle.STATUS_ACTIVE,
+        )
+
+        response = self.client.get(f"/api/v1/cycles/{cycle.pk}/")
+
+        self.assertEqual(response.status_code, 200, response.data)
+        self.assertEqual(response.data["status"], Cycle.STATUS_CLOSED)
+        cycle.refresh_from_db()
+        self.assertEqual(cycle.status, Cycle.STATUS_CLOSED)
+
     def test_flock_persists_across_cycles(self):
         flock = Flock.objects.create(flock_name="Lantai 1", kandang=self.kandang)
         start = date.today()
@@ -88,7 +139,7 @@ class FarmAPITests(TestCase):
         results = response.data["results"] if isinstance(response.data, dict) else response.data
         self.assertEqual(len(results), 2)
 
-    def test_set_initial_balance_seeds_feed_sack_snapshot_when_none_exist(self):
+    def test_set_initial_balance_does_not_create_feed_sack_row(self):
         start = date.today()
         cycle = Cycle.objects.create(
             kandang=self.kandang,
@@ -108,12 +159,12 @@ class FarmAPITests(TestCase):
         )
         self.assertEqual(response.status_code, 200, response.data)
         self.assertEqual(response.data["feed_initial_balance_date"], start.isoformat())
+        cycle.refresh_from_db()
+        self.assertEqual(cycle.feed_initial_balance, 120)
+        self.assertFalse(FeedSacks.objects.filter(cycle=cycle).exists())
+        self.assertFalse(ManualInput.objects.filter(cycle=cycle).exists())
 
-        row = FeedSacks.objects.get(cycle=cycle)
-        self.assertEqual(row.date, start)
-        self.assertEqual(row.in_total, 120)
-
-    def test_set_initial_balance_recomputes_existing_feed_sack_in_total(self):
+    def test_set_initial_balance_does_not_change_existing_feed_sack_totals(self):
         start = date.today()
         cycle = Cycle.objects.create(
             kandang=self.kandang,
@@ -144,7 +195,50 @@ class FarmAPITests(TestCase):
         self.assertEqual(response.status_code, 200, response.data)
 
         feed_row.refresh_from_db()
-        self.assertEqual(feed_row.in_total, 124)
+        self.assertEqual(feed_row.in_total, 4)
+
+    def test_set_initial_balance_twice_on_different_dates_keeps_daily_totals(self):
+        start = date(2026, 8, 26)
+        cycle = Cycle.objects.create(
+            kandang=self.kandang,
+            total_days=35,
+            start_date=start,
+            end_date=start + timedelta(days=34),
+            doc_in_weight=40,
+            doc_in_count=9000,
+            status=Cycle.STATUS_ACTIVE,
+        )
+        feed_row = FeedSacks.objects.create(
+            cycle=cycle,
+            date=start + timedelta(days=2),
+            in_today=6,
+            in_total=6,
+            out_today=0,
+            out_total=0,
+            feed_use_today=0,
+            feed_use_total=0,
+        )
+
+        self.client.force_authenticate(self.user)
+        first = self.client.post(
+            f"/api/v1/cycles/{cycle.pk}/initial-balance/",
+            {"date": start.isoformat(), "feed_in_manual": 120},
+            format="json",
+        )
+        self.assertEqual(first.status_code, 200, first.data)
+        second = self.client.post(
+            f"/api/v1/cycles/{cycle.pk}/initial-balance/",
+            {"date": (start + timedelta(days=1)).isoformat(), "feed_in_manual": 80},
+            format="json",
+        )
+        self.assertEqual(second.status_code, 200, second.data)
+
+        cycle.refresh_from_db()
+        self.assertEqual(cycle.feed_initial_balance, 80)
+        self.assertEqual(cycle.feed_initial_balance_date, start + timedelta(days=1))
+        self.assertEqual(FeedSacks.objects.filter(cycle=cycle).count(), 1)
+        feed_row.refresh_from_db()
+        self.assertEqual(feed_row.in_total, 6)
 
     @patch("apps.farms.views.pull_initial_balance_iot", return_value=(115, None))
     def test_set_initial_balance_compares_manual_against_combined_api_iot(self, mock_pull):