87 lines
3.6 KiB
Python
87 lines
3.6 KiB
Python
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.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)
|
|
|
|
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_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])
|