105 lines
3.4 KiB
Python
105 lines
3.4 KiB
Python
from __future__ import annotations
|
|
|
|
import pytest
|
|
import pytest_asyncio
|
|
from sqlalchemy import select
|
|
from sqlalchemy.ext.asyncio import async_sessionmaker, create_async_engine
|
|
|
|
from app.models.base import Base
|
|
from app.models.project import Project
|
|
from app.models.signal import Signal
|
|
from app.models.signal_source_registry import SignalSourceRegistry
|
|
from app.models.user import User
|
|
from app.services.signal.signal_source_orchestration import SignalSourceOrchestrationService
|
|
|
|
|
|
@pytest_asyncio.fixture
|
|
async def db_session():
|
|
engine = create_async_engine("sqlite+aiosqlite:///:memory:")
|
|
async with engine.begin() as conn:
|
|
await conn.run_sync(
|
|
lambda sync_conn: Base.metadata.create_all(
|
|
sync_conn,
|
|
tables=[
|
|
User.__table__,
|
|
Project.__table__,
|
|
Signal.__table__,
|
|
SignalSourceRegistry.__table__,
|
|
],
|
|
)
|
|
)
|
|
maker = async_sessionmaker(engine, expire_on_commit=False)
|
|
async with maker() as session:
|
|
yield session
|
|
await engine.dispose()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_materialize_requirement_sources_dedupes_duplicate_items(db_session) -> None:
|
|
user = User(
|
|
tenant_id=1,
|
|
username="owner",
|
|
email="owner@example.com",
|
|
password_hash="hash",
|
|
name="负责人",
|
|
role="ADMIN",
|
|
status="active",
|
|
is_active=True,
|
|
)
|
|
project = Project(
|
|
tenant_id=1,
|
|
name="demo-project",
|
|
title="Demo Project",
|
|
status="active",
|
|
assigned_role="backend",
|
|
)
|
|
signal = Signal(
|
|
tenant_id=1,
|
|
source="AI-Steward",
|
|
content="main signal",
|
|
priority="high",
|
|
processed=True,
|
|
)
|
|
db_session.add_all([user, project, signal])
|
|
await db_session.commit()
|
|
|
|
service = SignalSourceOrchestrationService(db_session)
|
|
results = await service.materialize_requirement_sources(
|
|
tenant_id=1,
|
|
project_id=int(project.id),
|
|
signal_id=int(signal.id),
|
|
operator_user_id=int(user.id),
|
|
requested_via="ai_steward",
|
|
origin="project_analysis",
|
|
origin_text="我们要做一个 AI 客服工具，包含内容本地化和社媒运营。",
|
|
default_source_kind="custom",
|
|
requirement_items=[
|
|
{
|
|
"section": "核心痛点",
|
|
"title": "内容本地化",
|
|
"description": "翻译公司贵、慢、不地道",
|
|
},
|
|
{
|
|
"section": "核心痛点",
|
|
"title": "内容本地化",
|
|
"description": "翻译公司贵、慢、不地道",
|
|
},
|
|
{
|
|
"section": "MVP 功能",
|
|
"title": "社媒运营",
|
|
"description": "自动生成社媒文案并排期发布",
|
|
},
|
|
],
|
|
)
|
|
|
|
assert len(results) == 2
|
|
assert results[0]["section"] == "核心痛点"
|
|
assert results[0]["source_kind"] == "custom"
|
|
assert results[0]["origin"] == "project_analysis"
|
|
|
|
rows = (await db_session.execute(select(SignalSourceRegistry).order_by(SignalSourceRegistry.id))).scalars().all()
|
|
assert len(rows) == 2
|
|
assert rows[0].raw_context["origin"] == "project_analysis"
|
|
assert rows[0].raw_context["requirement_item"]["title"] == "内容本地化"
|
|
assert rows[1].raw_context["requirement_item"]["section"] == "MVP 功能"
|