fix feed data pull from api using total instead of each kandang data

This commit is contained in:
Alberto-Audrix committed 2026-09-02 09:38:45 +07:00
1 parent 54d65724aa
commit 55ccc84386
2 files changed
+109 -7

No files matched your search

@@ -4,13 +4,37 @@ from unittest.mock import patch
from django.core.exceptions import ValidationError
from django.test import TestCase, override_settings
from apps.accounts.models import User
from apps.farms.models import Cycle, Kandang, Site
from apps.operations.services.karung_web import (
KarungWebError,
fetch_iot_masuk_for_date,
initial_balance_accuracy,
masuk_source_for_kandang,
request_karung,
source_for_kandang,
)
ACTIVE_SECTIONS_SAMPLE = {
"sections": {
"today": {
"date": "2026-09-02",
"masuk": [
{"karung": 192, "source": "K1 - In"},
{"karung": 2, "source": "K1 - Out"},
{"karung": 120, "source": "K2 - In"},
{"karung": 5, "source": "K2 - Out"},
],
"tuang": [
{"karung": 7, "source": "K1 - Use"},
{"karung": 9, "source": "K2 - Use"},
],
"totals": {"masuk": 312, "tuang": 16},
}
},
"cycle": {"saldo_awal": 240},
}
COMBINED_SAMPLE = {
"masuk": {
"history": [
@@ -31,6 +55,8 @@ class MasukSourceForKandangTests(TestCase):
def test_maps_kandang_number_to_iot_source(self):
self.assertEqual(masuk_source_for_kandang("Kandang 1"), "K1 - In")
self.assertEqual(masuk_source_for_kandang("Kandang 2"), "K2 - In")
self.assertEqual(source_for_kandang("Kandang 2", kind="Use"), "K2 - Use")
self.assertEqual(source_for_kandang("Kandang 2", kind="Out"), "K2 - Out")
def test_rejects_kandang_name_without_number(self):
with self.assertRaises(ValidationError):
@@ -97,3 +123,46 @@ class KarungWebCombinedTests(TestCase):
def test_fetch_iot_masuk_for_date_propagates_upstream_errors(self, _mock_fetch):
with self.assertRaises(KarungWebError):
fetch_iot_masuk_for_date(date(2026, 8, 27), kandang_name="Kandang 1")
@override_settings(KARUNG_WEB_ADMIN_BASE_URL="http://karung.internal")
class RequestKarungTests(TestCase):
def setUp(self):
user = User.objects.create_user(user_name="tester", password="secret")
site = Site.objects.create(site_name="Sukawarna", user=user)
self.kandang1 = Kandang.objects.create(kandang_name="Kandang 1", site=site)
self.kandang2 = Kandang.objects.create(kandang_name="Kandang 2", site=site)
self.cycle1 = Cycle.objects.create(
kandang=self.kandang1,
total_days=35,
start_date=date(2026, 9, 2),
end_date=date(2026, 10, 5),
doc_in_weight=40,
doc_in_count=10000,
status=Cycle.STATUS_ACTIVE,
)
self.cycle2 = Cycle.objects.create(
kandang=self.kandang2,
total_days=35,
start_date=date(2026, 9, 2),
end_date=date(2026, 10, 5),
doc_in_weight=40,
doc_in_count=10000,
status=Cycle.STATUS_ACTIVE,
)
@patch("apps.operations.services.karung_web.KarungWebClient.fetch_active_sections")
def test_request_karung_uses_per_kandang_sources_not_aggregate_totals(self, mock_fetch):
mock_fetch.return_value = ACTIVE_SECTIONS_SAMPLE
feed_sack1, summary1 = request_karung(self.cycle1, section="today")
feed_sack2, summary2 = request_karung(self.cycle2, section="today")
self.assertEqual(feed_sack1.in_today, 192)
self.assertEqual(feed_sack1.feed_use_today, 7)
self.assertEqual(feed_sack1.out_today, 2)
self.assertEqual(feed_sack2.in_today, 120)
self.assertEqual(feed_sack2.feed_use_today, 9)
self.assertEqual(feed_sack2.out_today, 5)
self.assertEqual(summary1["kandang_totals"]["in_today"], 192)
self.assertEqual(summary2["kandang_totals"]["in_today"], 120)