71 lines
2.5 KiB
Python
71 lines
2.5 KiB
Python
"""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)
|