Портал технической службы: наряды и чек-листы, задачи, отказы и неисправности, анализ надёжности, карты, склады, приложение для Android

tests/test_y_assistant.py

@@ -0,0 +1,83 @@
+"""AI-помощник: настройки (ключ не утекает), фильтр тематики, инструменты с правами пользователя, лимиты."""
+import os
+import tempfile
+
+os.environ.setdefault("DATABASE_URL", os.environ.get("TEST_DATABASE_URL") or "sqlite:///" + os.path.join(tempfile.mkdtemp(), "as.db").replace("\\", "/"))
+os.environ.setdefault("SCHEDULER_INTERVAL", "3600")
+os.environ.setdefault("SECRET_KEY", "test-secret-key-for-pytest-0123456789abcdef")
+os.environ.setdefault("DATA_DIR", tempfile.mkdtemp())
+
+from fastapi.testclient import TestClient  # noqa: E402
+
+from app import assistant as A  # noqa: E402
+from app.main import app  # noqa: E402
+
+V1 = "/api/v1"
+
+
+def test_assistant(monkeypatch):
+    with TestClient(app) as c:
+        adm = {"Authorization": "Bearer " + c.post(f"{V1}/auth/login", json={"username": "admin", "password": "admin"}).json()["access_token"]}
+        c.post(f"{V1}/users", headers=adm, json={"username": "as_tech", "password": "Pass1234567", "full_name": "Помощников Тест"})
+        tech = {"Authorization": "Bearer " + c.post(f"{V1}/auth/login", json={"username": "as_tech", "password": "Pass1234567"}).json()["access_token"]}
+        ask = lambda q, h=tech: c.post(f"{V1}/assistant/chat", headers=h, json={"messages": [{"role": "user", "content": q}]})
+
+        # выключен по умолчанию
+        assert c.get(f"{V1}/assistant/status", headers=tech).json() == {"enabled": False}
+        assert ask("где задания?").status_code == 503
+        # настройки — только админ; без ключа включить нельзя
+        assert c.put(f"{V1}/assistant/config", headers=tech, json={"enabled": True}).status_code == 403
+        assert c.put(f"{V1}/assistant/config", headers=adm, json={"enabled": True, "model": "m"}).status_code == 422
+        assert c.put(f"{V1}/assistant/config", headers=adm, json={"base_url": "ftp://x"}).status_code == 422
+        r = c.put(f"{V1}/assistant/config", headers=adm, json={"enabled": True, "provider": "openai", "model": "test-model",
+                                                               "api_key": "sk-secret-value-1234", "base_url": "https://llm.example/v1"})
+        assert r.status_code == 200 and r.json()["api_key_set"] and "sk-secret" not in r.text and r.json()["api_key_hint"] == "…1234"
+        assert "sk-secret" not in c.get(f"{V1}/settings", headers=adm).text
+        assert c.get(f"{V1}/assistant/status", headers=tech).json() == {"enabled": True}
+
+        calls = []
+
+        async def fake(cfg, system, messages):
+            calls.append((system, messages))
+            q = messages[-1]["content"]
+            if "стих" in q:
+                return "OFFTOPIC"
+            if "задани" in q and not q.startswith("РЕЗУЛЬТАТ"):
+                return 'CALL my_tasks {}'
+            if "склад" in q and not q.startswith("РЕЗУЛЬТАТ"):
+                return 'CALL warehouses {}'
+            if q.startswith("РЕЗУЛЬТАТ"):
+                return "Ответ по данным: " + q[:200]
+            return "Откройте [Суточные задания](#/tasks)."
+        monkeypatch.setattr(A, "complete", fake)
+
+        # системный промт знает пользователя и его роль; справка в нём
+        r = ask("Как поставить задание?")
+        assert r.status_code == 200, r.text
+        sys = calls[-1][0]
+        assert "Помощников Тест" in sys and "техник" in sys and "Задачи и поручения" in sys and "OFFTOPIC" in sys
+        # не по теме — вежливый отказ, а не ответ модели
+        r = ask("Напиши стих про весну").json()
+        assert r["offtopic"] and r["reply"] == A.REFUSAL
+        # инструмент вызывается от имени пользователя: его задания
+        c.post(f"{V1}/tasks", headers=tech, json={"title": "Проверить насос помощником"})
+        r = ask("Какие у меня задания?").json()
+        assert r["tools"] == ["my_tasks"] and "Проверить насос помощником" in r["reply"]
+        # нет прав — помощник получает отказ, а не данные
+        out = __import__("asyncio").run(A.run_tool(app, tech["Authorization"][7:], "equipment_units", {"id": "abc"}))
+        assert "числовой" in out
+        # длина вопроса и частота
+        assert ask("x" * 1500).status_code == 422
+        A._hits.clear()
+        monkeypatch.setattr(A, "RATE", (2, 3600))
+        assert ask("где наряды?").status_code == 200 and ask("где наряды?").status_code == 200
+        assert ask("где наряды?").status_code == 429
+        A._hits.clear()
+        # ключ не попадает в резервную копию
+        b = c.post(f"{V1}/backups", headers=adm, json={}).json()
+        import io, json, zipfile
+        z = zipfile.ZipFile(io.BytesIO(c.get(f"{V1}/backups/{b['name']}", headers=adm).content))
+        assert "sk-secret" not in z.read("db/settings.json").decode()
+        # выключение
+        c.put(f"{V1}/assistant/config", headers=adm, json={"enabled": False})
+        assert c.get(f"{V1}/assistant/status", headers=tech).json() == {"enabled": False}