43 lines
1.0 KiB
Python
43 lines
1.0 KiB
Python
"""Helpers for lab module tests."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import json
|
|
from pathlib import Path
|
|
|
|
from app import db
|
|
from app.lab.services.component_nutrients import upsert_from_api_dict
|
|
from app.models import Component
|
|
from app.models.base import default_uuid
|
|
|
|
FIXTURES_DIR = Path(__file__).resolve().parent.parent / "fixtures" / "lab"
|
|
|
|
|
|
def load_fixture(name: str) -> list[dict]:
|
|
path = FIXTURES_DIR / name
|
|
return json.loads(path.read_text(encoding="utf-8"))
|
|
|
|
|
|
def make_component(
|
|
*,
|
|
name: str = "Ячмень",
|
|
external_no: int | None = 1001,
|
|
dry_matter: float = 86.0,
|
|
nutrients: dict | None = None,
|
|
) -> Component:
|
|
comp = Component(
|
|
id=default_uuid(),
|
|
name=name,
|
|
type="Зерновые",
|
|
dry_matter=dry_matter,
|
|
protein=10.0,
|
|
energy=12.0,
|
|
price=15.0,
|
|
external_no=external_no,
|
|
)
|
|
db.session.add(comp)
|
|
db.session.flush()
|
|
upsert_from_api_dict(comp.id, nutrients or {"Сыр. Протеин": 115})
|
|
db.session.flush()
|
|
return comp
|