"""API skip компонентов и групп выгрузки.""" from __future__ import annotations import unittest from datetime import date from app import create_app, db from app.services.daily_plan.skips import skip_ingredient, skip_unloading_group from app.services.setup_state import mark_setup_complete from tests.helpers.dispenser_recipe_fixtures import ( E2E_DISP_RECIPE_1, E2E_PERIOD_A, seed_dispenser_period_recipes, ) from tests.helpers.zootech_test_helpers import ZootechTestConfig class DailyPlanPartSkipApiTests(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_get_skips_returns_grouped_payload(self) -> None: skip_ingredient(E2E_DISP_RECIPE_1, "e2e-disp-ing-1", "2026-06-07") skip_unloading_group(E2E_DISP_RECIPE_1, "e2e-disp-grp-1", "2026-06-07") resp = self.client.get("/api/daily-plan/skips?date=2026-06-07") self.assertEqual(resp.status_code, 200) body = resp.get_json() self.assertIn("trips", body) self.assertIn("ingredients", body) self.assertIn("unloadingGroups", body) self.assertEqual(len(body["ingredients"]), 1) self.assertEqual(len(body["unloadingGroups"]), 1) def test_post_and_delete_ingredient_skip(self) -> None: resp = self.client.post( "/api/daily-plan/skips/ingredients", json={ "recipeId": E2E_DISP_RECIPE_1, "ingredientId": "e2e-disp-ing-1", "date": "2026-06-08", }, ) self.assertEqual(resp.status_code, 200, resp.get_data(as_text=True)) deleted = self.client.delete( "/api/daily-plan/skips/ingredients" f"?recipe_id={E2E_DISP_RECIPE_1}&ingredient_id=e2e-disp-ing-1&date=2026-06-08" ) self.assertEqual(deleted.status_code, 200) def test_recipe_detail_marks_skipped_ingredient_for_zootech(self) -> None: skip_ingredient(E2E_DISP_RECIPE_1, "e2e-disp-ing-1", "2026-06-09") resp = self.client.get(f"/api/recipes/{E2E_DISP_RECIPE_1}?date=2026-06-09") self.assertEqual(resp.status_code, 200) ingredients = resp.get_json()["ingredients"] skipped = [i for i in ingredients if i.get("skippedToday")] self.assertEqual(len(skipped), 1) self.assertEqual(skipped[0]["id"], "e2e-disp-ing-1") def test_period_recipes_show_skipped_ingredient_flag(self) -> None: from app.services.daily_plan.skips import skip_ingredient today = date.today().isoformat() skip_ingredient(E2E_DISP_RECIPE_1, "e2e-disp-ing-1", today) resp = self.client.get(f"/api/periods/{E2E_PERIOD_A}/recipes") self.assertEqual(resp.status_code, 200) recipe = next(r for r in resp.get_json() if r["id"] == E2E_DISP_RECIPE_1) self.assertTrue(recipe.get("skippedIngredientToday")) def test_post_ingredient_skip_week_duration(self) -> None: resp = self.client.post( "/api/daily-plan/skips/ingredients", json={ "recipeId": E2E_DISP_RECIPE_1, "ingredientId": "e2e-disp-ing-1", "date": "2026-06-08", "duration": "week", }, ) self.assertEqual(resp.status_code, 200, resp.get_data(as_text=True)) body = resp.get_json() self.assertEqual(body.get("validUntil"), "2026-06-14") detail = self.client.get(f"/api/recipes/{E2E_DISP_RECIPE_1}?date=2026-06-10") skipped = [i for i in detail.get_json()["ingredients"] if i.get("skippedToday")] self.assertEqual(len(skipped), 1) def test_delete_all_ingredient_skips_for_recipe(self) -> None: skip_ingredient(E2E_DISP_RECIPE_1, "e2e-disp-ing-1", "2026-06-07") resp = self.client.delete( f"/api/daily-plan/skips/ingredients?recipe_id={E2E_DISP_RECIPE_1}&date=2026-06-07&all=1" ) self.assertEqual(resp.status_code, 200) self.assertGreaterEqual(resp.get_json().get("count", 0), 1) detail = self.client.get(f"/api/recipes/{E2E_DISP_RECIPE_1}?date=2026-06-07") skipped = [i for i in detail.get_json()["ingredients"] if i.get("skippedToday")] self.assertEqual(skipped, []) def test_feed_dispenser_has_skip_today_flag(self) -> None: today = date.today().isoformat() skip_ingredient(E2E_DISP_RECIPE_1, "e2e-disp-ing-1", today) resp = self.client.get("/api/feed_dispensers?limit=50") self.assertEqual(resp.status_code, 200) dispensers = resp.get_json() flagged = [d for d in dispensers if d.get("hasSkipToday")] self.assertTrue(flagged) def test_recipe_detail_hides_skipped_ingredient_for_kiosk(self) -> None: skip_ingredient(E2E_DISP_RECIPE_1, "e2e-disp-ing-1", "2026-06-09") resp = self.client.get( f"/api/recipes/{E2E_DISP_RECIPE_1}?date=2026-06-09", headers={"X-Wesp-Kiosk": "1"}, ) self.assertEqual(resp.status_code, 200) ids = {i["id"] for i in resp.get_json()["ingredients"]} self.assertNotIn("e2e-disp-ing-1", ids) if __name__ == "__main__": unittest.main()