@@ -0,0 +1,42 @@
|
||||
"""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
|
||||
Reference in New Issue
Block a user