135 lines
5.6 KiB
Python
135 lines
5.6 KiB
Python
"""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()
|