Интегрирован wesp в сайт

2026-07-17 12:57:18 +03:00
parent 5dfa06ddbe
commit 355c0ef9f1
883 changed files with 194576 additions and 177 deletions
@@ -0,0 +1,212 @@
+"""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()