31 lines
816 B
Python
31 lines
816 B
Python
import pytest
|
|
|
|
from apps.scenarios.services import (
|
|
ScenarioNotFound,
|
|
get_scenario,
|
|
list_scenarios,
|
|
)
|
|
|
|
|
|
def test_list_scenarios_loads_five_configs():
|
|
scenarios = list_scenarios()
|
|
assert [scenario["id"] for scenario in scenarios] == [
|
|
"document_review",
|
|
"knowledge_qa",
|
|
"quality_analysis",
|
|
"risk_audit",
|
|
"ticket_assistant",
|
|
]
|
|
|
|
|
|
def test_get_scenario_returns_full_agent_config():
|
|
scenario = get_scenario("quality_analysis")
|
|
assert scenario["agent"]["role"] == "质量管理专家"
|
|
assert scenario["rag"]["enabled"] is True
|
|
assert scenario["output"]["type"] == "quality_report"
|
|
|
|
|
|
def test_get_scenario_raises_clear_error_for_missing_id():
|
|
with pytest.raises(ScenarioNotFound, match="场景不存在"):
|
|
get_scenario("missing")
|