56 lines
1.9 KiB
Python
56 lines
1.9 KiB
Python
"""Components API: nutrients без обязательных protein/energy."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import json
|
|
import unittest
|
|
|
|
from app import create_app, db
|
|
from app.lab.services.component_nutrients import nutrients_api_dict
|
|
from app.models import Component
|
|
from app.services.setup_state import mark_setup_complete
|
|
from tests.helpers.zootech_test_helpers import ZootechTestConfig
|
|
|
|
|
|
class ComponentRefactorTests(unittest.TestCase):
|
|
def setUp(self) -> None:
|
|
self.app = create_app(ZootechTestConfig)
|
|
self.client = self.app.test_client()
|
|
self.ctx = self.app.app_context()
|
|
self.ctx.push()
|
|
db.create_all()
|
|
mark_setup_complete(self.app)
|
|
self.client.post(
|
|
"/api/auth/login",
|
|
json={"login": "zootech-test-admin", "password": "zootech-test-secret"},
|
|
)
|
|
|
|
def tearDown(self) -> None:
|
|
db.session.remove()
|
|
db.drop_all()
|
|
self.ctx.pop()
|
|
|
|
def test_create_component_with_nutrients(self) -> None:
|
|
r = self.client.post(
|
|
"/api/components",
|
|
json={
|
|
"name": "Тест lab nutrients",
|
|
"type": "Концентрированные",
|
|
"dry_matter": 88.0,
|
|
"price": 12.0,
|
|
"protein": 0,
|
|
"energy": 0,
|
|
"nutrients": {"Сыр. Протеин": 110, "СВ": 880},
|
|
},
|
|
)
|
|
self.assertEqual(r.status_code, 201, r.get_data(as_text=True))
|
|
comp_id = r.get_json()["id"]
|
|
nutrients = nutrients_api_dict(comp_id)
|
|
self.assertEqual(nutrients.get("Сыр. Протеин"), 110)
|
|
|
|
def test_get_component_returns_nutrients(self) -> None:
|
|
r = self.client.get("/api/components")
|
|
self.assertEqual(r.status_code, 200)
|
|
body = r.get_json()
|
|
self.assertTrue(isinstance(body, list))
|