Интегрирован 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,77 @@
+"""Профили стада — прямое чтение без legacy remap."""
+
+from __future__ import annotations
+
+import unittest
+
+from app import create_app, db
+from app.lab.commands.delete_profile import delete_animal_profile
+from app.lab.commands.upsert_profile import upsert_animal_profile
+from app.lab.loaders.profile_loader import list_animal_profiles, load_animal_profile
+from app.lab.models import LabAnimalProfile
+from app.services.setup_state import mark_setup_complete
+from tests.helpers.zootech_test_helpers import ZootechTestConfig
+
+
+class LabProfileLoaderTests(unittest.TestCase):
+    def setUp(self) -> None:
+        self.app = create_app(ZootechTestConfig)
+        self.ctx = self.app.app_context()
+        self.ctx.push()
+        db.create_all()
+        mark_setup_complete(self.app)
+
+    def tearDown(self) -> None:
+        db.session.remove()
+        db.drop_all()
+        self.ctx.pop()
+
+    def test_create_and_load_profile(self) -> None:
+        result = upsert_animal_profile(
+            None,
+            {
+                "profileKey": "test_beef_custom",
+                "label": "Тест",
+                "rationType": "BEEF",
+                "normsData": {"dry_matter": {"min": 1, "max": 2}},
+            },
+            "test",
+        )
+        pid = result["id"]
+        data = load_animal_profile(pid)
+        self.assertEqual(data["profileKey"], "test_beef_custom")
+        self.assertEqual(data["normsProfileKey"], "test_beef_custom")
+        self.assertFalse(data["legacyNormsRemapped"])
+
+    def test_list_includes_lab_math_profiles(self) -> None:
+        db.session.add(
+            LabAnimalProfile(
+                profile_key="lab_math_dairy_01",
+                label="Math 1",
+                ration_type="DAIRY",
+                created_by="test",
+                updated_by="test",
+            )
+        )
+        db.session.commit()
+        keys = [p["profileKey"] for p in list_animal_profiles("DAIRY")]
+        self.assertIn("lab_math_dairy_01", keys)
+
+    def test_delete_profile_soft_deletes_and_hides_from_list(self) -> None:
+        result = upsert_animal_profile(
+            None,
+            {
+                "profileKey": "to_delete",
+                "label": "На удаление",
+                "rationType": "BEEF",
+            },
+            "test",
+        )
+        pid = result["id"]
+        delete_animal_profile(pid, "test")
+        with self.assertRaises(LookupError):
+            load_animal_profile(pid)
+        keys = [p["profileKey"] for p in list_animal_profiles()]
+        self.assertNotIn("to_delete", keys)
+        row = LabAnimalProfile.query.get(pid)
+        self.assertTrue(row.is_deleted)