初始化提交

backend/tests/repositories/test_skills_repo.py

@@ -0,0 +1,65 @@
+"""Skill 数据访问测试。"""
+
+import pytest
+from sqlalchemy.exc import IntegrityError
+
+from app.db.session import AsyncSessionLocal
+from app.models.skills import Skill
+from app.repositories import skills as skills_repo
+
+
+def _skill(name: str = "coding", **overrides) -> Skill:
+    base = {
+        "name": name,
+        "description": "代码辅助",
+        "trigger": "代码,编程",
+        "prompt_fragment": "你是代码助手",
+        "tool_bindings_json": "[]",
+        "risk_level": "L1",
+        "status": "enabled",
+    }
+    base.update(overrides)
+    return Skill(**base)
+
+
+@pytest.mark.asyncio
+async def test_add_and_get_skill() -> None:
+    async with AsyncSessionLocal() as session:
+        s = _skill()
+        await skills_repo.add_skill(session, s)
+        await session.commit()
+        sid = s.id
+
+    async with AsyncSessionLocal() as session:
+        fetched = await skills_repo.get_skill(session, sid)
+    assert fetched is not None
+    assert fetched.name == "coding"
+    assert fetched.trigger == "代码,编程"
+    assert fetched.status == "enabled"
+
+
+@pytest.mark.asyncio
+async def test_list_skills_with_status_filter() -> None:
+    async with AsyncSessionLocal() as session:
+        await skills_repo.add_skill(session, _skill("a", status="enabled"))
+        await skills_repo.add_skill(session, _skill("b", status="disabled"))
+        await session.commit()
+
+    async with AsyncSessionLocal() as session:
+        enabled = await skills_repo.list_skills(session, status="enabled")
+        all_skills = await skills_repo.list_skills(session)
+    assert len(enabled) == 1
+    assert enabled[0].name == "a"
+    assert len(all_skills) == 2
+
+
+@pytest.mark.asyncio
+async def test_name_unique_conflict() -> None:
+    async with AsyncSessionLocal() as session:
+        await skills_repo.add_skill(session, _skill("dup"))
+        await session.commit()
+
+    async with AsyncSessionLocal() as session:
+        with pytest.raises(IntegrityError):
+            await skills_repo.add_skill(session, _skill("dup"))
+            await session.commit()