"""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()