@@ -0,0 +1,70 @@
|
||||
"""save_component_nutrients — единая точка записи."""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import unittest
|
||||
|
||||
from app import create_app, db
|
||||
from app.lab.models import LabComponentNutrientValue
|
||||
from app.lab.services.component_nutrients import nutrients_api_dict, save_component_nutrients
|
||||
from app.models import Component
|
||||
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 ComponentNutrientsSaveTests(unittest.TestCase):
|
||||
def setUp(self) -> None:
|
||||
self.app = create_app(ZootechTestConfig)
|
||||
self.ctx = self.app.app_context()
|
||||
self.ctx.push()
|
||||
db.create_all()
|
||||
mark_setup_complete(self.app)
|
||||
self.comp = Component(
|
||||
id=default_uuid(),
|
||||
name="Save test",
|
||||
dry_matter=88.0,
|
||||
created_by="test",
|
||||
updated_by="test",
|
||||
)
|
||||
db.session.add(self.comp)
|
||||
db.session.commit()
|
||||
|
||||
def tearDown(self) -> None:
|
||||
db.session.remove()
|
||||
db.drop_all()
|
||||
self.ctx.pop()
|
||||
|
||||
def test_save_returns_affected_recipes(self) -> None:
|
||||
result = save_component_nutrients(
|
||||
self.comp.id,
|
||||
{"Сыр. Протеин": 110, "Сырая клетч": 300, "Сырой жир": 20},
|
||||
)
|
||||
self.assertIn("stored", result)
|
||||
self.assertIn("affectedRecipeIds", result)
|
||||
self.assertEqual(nutrients_api_dict(self.comp.id).get("Сыр. Протеин"), 110)
|
||||
|
||||
def test_save_preserves_extra_lab_keys_after_derive(self) -> None:
|
||||
result = save_component_nutrients(
|
||||
self.comp.id,
|
||||
{
|
||||
"Сыр. Протеин": 107.5,
|
||||
"Сырая клетч": 645.5,
|
||||
"Сырой жир": 44.5,
|
||||
"Лизин": 4.4,
|
||||
"Ca": 7.3,
|
||||
"ВРХ Орг Вещ": 53.0,
|
||||
},
|
||||
)
|
||||
stored = nutrients_api_dict(self.comp.id)
|
||||
self.assertEqual(stored.get("Лизин"), 4.4)
|
||||
self.assertEqual(stored.get("Ca"), 7.3)
|
||||
self.assertIn("stored", result)
|
||||
|
||||
def test_sv_not_stored_in_eav(self) -> None:
|
||||
save_component_nutrients(
|
||||
self.comp.id,
|
||||
{"Сыр. Протеин": 100, "Сырая клетч": 280, "Сырой жир": 18, "СВ": 999},
|
||||
)
|
||||
keys = {r.nutrient_key for r in LabComponentNutrientValue.query.filter_by(component_id=self.comp.id)}
|
||||
self.assertNotIn("СВ", keys)
|
||||
Reference in New Issue
Block a user