@@ -0,0 +1,110 @@
|
||||
"""Golden path: seed → recalculate → diff → apply_from_master."""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import json
|
||||
import unittest
|
||||
|
||||
from app import create_app, db
|
||||
from app.lab.services.component_nutrients import upsert_from_api_dict
|
||||
from app.models import Component, Ingredient, Recipe
|
||||
from app.models.base import default_uuid
|
||||
from app.services.setup_state import mark_setup_complete
|
||||
from tests.helpers.zootech_test_helpers import ZootechTestConfig
|
||||
|
||||
|
||||
class LabGoldenPathTests(unittest.TestCase):
|
||||
def setUp(self) -> None:
|
||||
self.app = create_app(ZootechTestConfig)
|
||||
self.client = self.app.test_client()
|
||||
self.ctx = self.app.app_context()
|
||||
self.ctx.push()
|
||||
db.create_all()
|
||||
mark_setup_complete(self.app)
|
||||
|
||||
self.comp = Component(
|
||||
id=default_uuid(),
|
||||
name="Ячмень",
|
||||
type="Зерновые",
|
||||
dry_matter=86.0,
|
||||
protein=12.0,
|
||||
energy=12.5,
|
||||
price=15.0,
|
||||
)
|
||||
db.session.add(self.comp)
|
||||
db.session.flush()
|
||||
upsert_from_api_dict(self.comp.id, {"Сыр. Протеин": 120, "СВ": 860})
|
||||
self.recipe = Recipe(
|
||||
id=default_uuid(),
|
||||
name="Golden path",
|
||||
heads_per_trip=20,
|
||||
mixing_time=5,
|
||||
ration_type="BEEF",
|
||||
)
|
||||
self.ingredient = Ingredient(
|
||||
id=default_uuid(),
|
||||
recipe_id=self.recipe.id,
|
||||
component_id=self.comp.id,
|
||||
name="Ячмень",
|
||||
amount=4.0,
|
||||
weight_per_head=4.0,
|
||||
dry_matter=86.0,
|
||||
order=0,
|
||||
)
|
||||
db.session.add_all([self.recipe, self.ingredient])
|
||||
db.session.commit()
|
||||
self.recipe_id = self.recipe.id
|
||||
|
||||
self.client.post(
|
||||
"/api/auth/login",
|
||||
json={"login": "zootech-test-admin", "password": "zootech-test-secret"},
|
||||
)
|
||||
|
||||
def tearDown(self) -> None:
|
||||
db.session.remove()
|
||||
db.drop_all()
|
||||
self.ctx.pop()
|
||||
|
||||
def test_full_http_golden_path(self) -> None:
|
||||
r = self.client.post(f"/api/lab/rations/{self.recipe_id}/seed-from-execution")
|
||||
self.assertEqual(r.status_code, 200, r.get_data(as_text=True))
|
||||
|
||||
r = self.client.post(f"/api/lab/rations/{self.recipe_id}/recalculate")
|
||||
self.assertEqual(r.status_code, 200, r.get_data(as_text=True))
|
||||
calc = r.get_json()
|
||||
self.assertTrue(calc.get("indicators"))
|
||||
|
||||
r = self.client.get(f"/api/lab/rations/{self.recipe_id}")
|
||||
self.assertTrue(r.get_json()["exists"])
|
||||
|
||||
r = self.client.put(
|
||||
f"/api/lab/rations/{self.recipe_id}",
|
||||
json={
|
||||
"lines": [
|
||||
{
|
||||
"componentId": self.comp.id,
|
||||
"dailyKg": 100.0,
|
||||
"inRation": True,
|
||||
"inCompound": True,
|
||||
}
|
||||
],
|
||||
"rationType": "BEEF",
|
||||
},
|
||||
)
|
||||
self.assertEqual(r.status_code, 200)
|
||||
|
||||
r = self.client.get(f"/api/lab/rations/{self.recipe_id}/diff")
|
||||
self.assertEqual(r.status_code, 200)
|
||||
self.assertTrue(r.get_json()["hasChanges"])
|
||||
|
||||
r = self.client.post(f"/api/lab/rations/{self.recipe_id}/apply-from-master")
|
||||
self.assertEqual(r.status_code, 200)
|
||||
|
||||
db.session.refresh(self.ingredient)
|
||||
self.assertAlmostEqual(float(self.ingredient.weight_per_head), 5.0, places=2)
|
||||
|
||||
r = self.client.get(f"/api/lab/rations/diff?recipe_ids={self.recipe_id}")
|
||||
self.assertEqual(r.status_code, 200)
|
||||
batch = r.get_json()["results"]
|
||||
self.assertEqual(len(batch), 1)
|
||||
self.assertFalse(batch[0]["hasChanges"])
|
||||
Reference in New Issue
Block a user