213 lines
8.2 KiB
Python
213 lines
8.2 KiB
Python
"""API /api/daily-plan/skips и фильтрация рейсов для терминала."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import unittest
|
|
from datetime import date
|
|
|
|
from app import create_app, db
|
|
from app.services.daily_plan.skips import skip_trip
|
|
from app.services.setup_state import mark_setup_complete
|
|
from app.services.daily_plan.builder import recipe_total_weights_by_id
|
|
from tests.helpers.dispenser_recipe_fixtures import (
|
|
E2E_DISP_ID,
|
|
E2E_DISP_RECIPE_1,
|
|
E2E_DISP_RECIPE_2,
|
|
E2E_PERIOD_A,
|
|
seed_dispenser_period_recipes,
|
|
)
|
|
from tests.helpers.zootech_test_helpers import ZootechTestConfig
|
|
|
|
|
|
class DailyTripSkipApiTests(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)
|
|
seed_dispenser_period_recipes()
|
|
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_post_and_delete_skip(self) -> None:
|
|
resp = self.client.post(
|
|
"/api/daily-plan/skips",
|
|
json={"recipeId": E2E_DISP_RECIPE_1, "date": "2026-06-07"},
|
|
)
|
|
self.assertEqual(resp.status_code, 200, resp.get_data(as_text=True))
|
|
body = resp.get_json()
|
|
self.assertEqual(body["recipeId"], E2E_DISP_RECIPE_1)
|
|
|
|
listed = self.client.get("/api/daily-plan/skips?date=2026-06-07")
|
|
self.assertEqual(listed.status_code, 200)
|
|
self.assertEqual(len(listed.get_json()["trips"]), 1)
|
|
|
|
deleted = self.client.delete(
|
|
f"/api/daily-plan/skips?recipe_id={E2E_DISP_RECIPE_1}&date=2026-06-07"
|
|
)
|
|
self.assertEqual(deleted.status_code, 200)
|
|
self.assertEqual(
|
|
self.client.get("/api/daily-plan/skips?date=2026-06-07").get_json()["trips"],
|
|
[],
|
|
)
|
|
|
|
def test_skip_requires_auth(self) -> None:
|
|
anon = self.app.test_client()
|
|
resp = anon.post(
|
|
"/api/daily-plan/skips",
|
|
json={"recipeId": E2E_DISP_RECIPE_1, "date": "2026-06-07"},
|
|
)
|
|
self.assertIn(resp.status_code, (401, 403))
|
|
|
|
def test_period_recipes_hide_skipped_for_terminal(self) -> None:
|
|
today = date.today().isoformat()
|
|
skip_trip(E2E_DISP_RECIPE_1, today)
|
|
anon = self.app.test_client()
|
|
resp = anon.get(f"/api/periods/{E2E_PERIOD_A}/recipes")
|
|
self.assertEqual(resp.status_code, 200, resp.get_data(as_text=True))
|
|
ids = {r["id"] for r in resp.get_json()}
|
|
self.assertNotIn(E2E_DISP_RECIPE_1, ids)
|
|
|
|
def test_period_recipes_show_skipped_flag_for_zootech(self) -> None:
|
|
today = date.today().isoformat()
|
|
skip_trip(E2E_DISP_RECIPE_1, today)
|
|
resp = self.client.get(f"/api/periods/{E2E_PERIOD_A}/recipes")
|
|
self.assertEqual(resp.status_code, 200)
|
|
skipped = [r for r in resp.get_json() if r.get("skippedToday")]
|
|
self.assertEqual(len(skipped), 1)
|
|
self.assertEqual(skipped[0]["id"], E2E_DISP_RECIPE_1)
|
|
|
|
def test_period_recipes_hide_skipped_for_kiosk_even_when_authenticated(self) -> None:
|
|
today = date.today().isoformat()
|
|
skip_trip(E2E_DISP_RECIPE_1, today)
|
|
resp = self.client.get(
|
|
f"/api/periods/{E2E_PERIOD_A}/recipes",
|
|
headers={"X-Wesp-Kiosk": "1"},
|
|
)
|
|
self.assertEqual(resp.status_code, 200)
|
|
ids = {r["id"] for r in resp.get_json()}
|
|
self.assertNotIn(E2E_DISP_RECIPE_1, ids)
|
|
|
|
def test_period_recipes_include_plan_total_weight_for_kiosk(self) -> None:
|
|
resp = self.app.test_client().get(
|
|
f"/api/periods/{E2E_PERIOD_A}/recipes",
|
|
headers={"X-Wesp-Kiosk": "1"},
|
|
)
|
|
self.assertEqual(resp.status_code, 200)
|
|
by_id = {r["id"]: r for r in resp.get_json()}
|
|
self.assertEqual(by_id[E2E_DISP_RECIPE_1]["total_weight"], 20.0)
|
|
self.assertEqual(by_id[E2E_DISP_RECIPE_2]["total_weight"], 30.0)
|
|
|
|
def test_kiosk_recipe_detail_uses_daily_plan_weights(self) -> None:
|
|
today = date.today().isoformat()
|
|
plan_weights = recipe_total_weights_by_id(dispenser_id=E2E_DISP_ID, plan_date=today)
|
|
resp = self.client.get(
|
|
f"/api/recipes/{E2E_DISP_RECIPE_1}?date={today}",
|
|
headers={"X-Wesp-Kiosk": "1"},
|
|
)
|
|
self.assertEqual(resp.status_code, 200)
|
|
data = resp.get_json()
|
|
self.assertEqual(data.get("total_weight"), plan_weights[E2E_DISP_RECIPE_1])
|
|
sum_ing = sum(float(i.get("amount") or 0) for i in data.get("ingredients") or [])
|
|
self.assertAlmostEqual(sum_ing, plan_weights[E2E_DISP_RECIPE_1], places=2)
|
|
|
|
def test_kiosk_recipe_detail_includes_plan_unloading_group_weights(self) -> None:
|
|
today = date.today().isoformat()
|
|
resp = self.client.get(
|
|
f"/api/recipes/{E2E_DISP_RECIPE_1}?date={today}",
|
|
headers={"X-Wesp-Kiosk": "1"},
|
|
)
|
|
self.assertEqual(resp.status_code, 200)
|
|
data = resp.get_json()
|
|
groups = data.get("unloadingGroups") or []
|
|
self.assertEqual(len(groups), 1)
|
|
self.assertEqual(groups[0]["weight"], data.get("total_weight"))
|
|
|
|
def test_kiosk_recipe_detail_respects_trip_percent_from_plan(self) -> None:
|
|
from app.models import Recipe
|
|
|
|
today = date.today().isoformat()
|
|
recipe = db.session.get(Recipe, E2E_DISP_RECIPE_1)
|
|
recipe.trip_percent = 50.0
|
|
db.session.commit()
|
|
try:
|
|
plan_weights = recipe_total_weights_by_id(dispenser_id=E2E_DISP_ID, plan_date=today)
|
|
resp = self.client.get(
|
|
f"/api/recipes/{E2E_DISP_RECIPE_1}?date={today}",
|
|
headers={"X-Wesp-Kiosk": "1"},
|
|
)
|
|
data = resp.get_json()
|
|
self.assertEqual(data.get("total_weight"), plan_weights[E2E_DISP_RECIPE_1])
|
|
self.assertEqual(data.get("total_weight"), 10.0)
|
|
self.assertAlmostEqual(
|
|
sum(float(i.get("amount") or 0) for i in data.get("ingredients") or []),
|
|
10.0,
|
|
places=2,
|
|
)
|
|
finally:
|
|
recipe.trip_percent = 100.0
|
|
db.session.commit()
|
|
|
|
def test_weight_display_uses_daily_plan_weights(self) -> None:
|
|
from app.models import Component, Recipe
|
|
from app.services.daily_plan.replacements import replace_ingredient
|
|
|
|
alt = Component(
|
|
id="e2e-disp-comp-alt-loading",
|
|
name="Alt Silo",
|
|
type="silage",
|
|
dry_matter=35.0,
|
|
protein=3.0,
|
|
energy=6.0,
|
|
price=0.0,
|
|
)
|
|
db.session.add(alt)
|
|
recipe = db.session.get(Recipe, E2E_DISP_RECIPE_1)
|
|
recipe.dry_matter_locked = True
|
|
db.session.commit()
|
|
plan_date = date.today().isoformat()
|
|
replace_ingredient(
|
|
E2E_DISP_RECIPE_1,
|
|
"e2e-disp-ing-1",
|
|
"e2e-disp-comp-alt-loading",
|
|
plan_date,
|
|
)
|
|
try:
|
|
kiosk = self.app.test_client()
|
|
plan_resp = kiosk.get(
|
|
f"/api/recipes/{E2E_DISP_RECIPE_1}?date={plan_date}",
|
|
headers={"X-Wesp-Kiosk": "1"},
|
|
)
|
|
self.assertEqual(plan_resp.status_code, 200)
|
|
plan_amount = float(plan_resp.get_json()["ingredients"][0]["amount"])
|
|
self.assertGreater(plan_amount, 10.0)
|
|
|
|
set_resp = kiosk.post(
|
|
"/api/set_current_recipe",
|
|
json={"recipe_id": E2E_DISP_RECIPE_1},
|
|
)
|
|
self.assertEqual(set_resp.status_code, 200)
|
|
display = kiosk.get("/api/weight_display_data")
|
|
self.assertEqual(display.status_code, 200, display.get_data(as_text=True))
|
|
body = display.get_json()
|
|
self.assertEqual(body["status"], "active")
|
|
self.assertEqual(body["total_component"], int(round(plan_amount)))
|
|
self.assertEqual(body["total_mixture"], int(round(plan_amount)))
|
|
self.assertIn("Alt Silo", body["component_name"])
|
|
finally:
|
|
recipe.dry_matter_locked = False
|
|
db.session.commit()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|