57 lines
1.8 KiB
Python
57 lines
1.8 KiB
Python
"""ration_recalc service."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import unittest
|
|
|
|
from app import create_app, db
|
|
from app.lab.models import LabRationLine, LabRecipeRation
|
|
from app.lab.services.ration_recalc import find_rations_by_component
|
|
from app.models import Component, 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 RationRecalcServiceTests(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.recipe = Recipe(id=default_uuid(), name="R", heads_per_trip=10, mixing_time=5)
|
|
self.comp = Component(
|
|
id=default_uuid(),
|
|
name="C",
|
|
dry_matter=88.0,
|
|
created_by="test",
|
|
updated_by="test",
|
|
)
|
|
db.session.add_all([self.recipe, self.comp])
|
|
db.session.add(
|
|
LabRecipeRation(recipe_id=self.recipe.id, created_by="test", updated_by="test")
|
|
)
|
|
db.session.add(
|
|
LabRationLine(
|
|
id=default_uuid(),
|
|
recipe_id=self.recipe.id,
|
|
component_id=self.comp.id,
|
|
daily_kg=10.0,
|
|
in_ration=True,
|
|
row_index=0,
|
|
created_by="test",
|
|
updated_by="test",
|
|
)
|
|
)
|
|
db.session.commit()
|
|
|
|
def tearDown(self) -> None:
|
|
db.session.remove()
|
|
db.drop_all()
|
|
self.ctx.pop()
|
|
|
|
def test_find_rations_by_component(self) -> None:
|
|
ids = find_rations_by_component(self.comp.id)
|
|
self.assertEqual(ids, [self.recipe.id])
|