| | """TDD tests for Parlant guideline configuration.""" |
| |
|
| | from unittest.mock import AsyncMock, MagicMock |
| |
|
| | import pytest |
| |
|
| | from trialpath.agent.guidelines import GUIDELINE_SPECS, configure_guidelines |
| |
|
| |
|
| | class TestGuidelineConfiguration: |
| | """Test guideline setup.""" |
| |
|
| | @pytest.fixture |
| | def mock_agent(self): |
| | agent = MagicMock() |
| | agent.create_guideline = AsyncMock(return_value=MagicMock()) |
| | return agent |
| |
|
| | @pytest.mark.asyncio |
| | async def test_correct_number_of_guidelines(self, mock_agent): |
| | """Should create exactly 11 guidelines.""" |
| | guidelines = await configure_guidelines(mock_agent) |
| | assert len(guidelines) == 11 |
| | assert mock_agent.create_guideline.call_count == 11 |
| |
|
| | def test_each_guideline_has_condition_and_action(self): |
| | """Each spec should have condition and action strings.""" |
| | for spec in GUIDELINE_SPECS: |
| | assert "condition" in spec |
| | assert "action" in spec |
| | assert isinstance(spec["condition"], str) |
| | assert isinstance(spec["action"], str) |
| | assert len(spec["condition"]) > 10 |
| | assert len(spec["action"]) > 10 |
| |
|
| | def test_ingest_guidelines_count(self): |
| | """Should have 2 INGEST phase guidelines.""" |
| | ingest = [s for s in GUIDELINE_SPECS if s["phase"] == "INGEST"] |
| | assert len(ingest) == 2 |
| |
|
| | def test_prescreen_guidelines_count(self): |
| | """Should have 3 PRESCREEN phase guidelines.""" |
| | prescreen = [s for s in GUIDELINE_SPECS if s["phase"] == "PRESCREEN"] |
| | assert len(prescreen) == 3 |
| |
|
| | def test_validate_trials_guidelines_count(self): |
| | """Should have 1 VALIDATE_TRIALS guideline.""" |
| | validate = [s for s in GUIDELINE_SPECS if s["phase"] == "VALIDATE_TRIALS"] |
| | assert len(validate) == 1 |
| |
|
| | def test_gap_followup_guidelines_count(self): |
| | """Should have 1 GAP_FOLLOWUP guideline.""" |
| | gap = [s for s in GUIDELINE_SPECS if s["phase"] == "GAP_FOLLOWUP"] |
| | assert len(gap) == 1 |
| |
|
| | def test_summary_guidelines_count(self): |
| | """Should have 1 SUMMARY guideline.""" |
| | summary = [s for s in GUIDELINE_SPECS if s["phase"] == "SUMMARY"] |
| | assert len(summary) == 1 |
| |
|
| | def test_global_guidelines_count(self): |
| | """Should have 3 global guidelines.""" |
| | global_g = [s for s in GUIDELINE_SPECS if s["phase"] == "GLOBAL"] |
| | assert len(global_g) == 3 |
| |
|
| | def test_tool_associations(self): |
| | """Guidelines with tools should reference correct tool entries.""" |
| | from parlant.sdk import ToolEntry |
| |
|
| | for spec in GUIDELINE_SPECS: |
| | for tool in spec["tools"]: |
| | assert isinstance(tool, ToolEntry), ( |
| | f"Tool {tool} in guideline '{spec['condition'][:30]}...' is not a ToolEntry" |
| | ) |
| |
|
| | def test_medical_disclaimer_guideline_exists(self): |
| | """Should have a medical disclaimer guideline.""" |
| | disclaimer = [ |
| | s |
| | for s in GUIDELINE_SPECS |
| | if "disclaimer" in s["action"].lower() or "medical advice" in s["action"].lower() |
| | ] |
| | assert len(disclaimer) >= 1 |
| |
|
| | @pytest.mark.asyncio |
| | async def test_configure_passes_tools_to_agent(self, mock_agent): |
| | """configure_guidelines should pass tools list to create_guideline.""" |
| | await configure_guidelines(mock_agent) |
| |
|
| | |
| | calls_with_tools = [ |
| | c for c in mock_agent.create_guideline.call_args_list if c.kwargs.get("tools") |
| | ] |
| | assert len(calls_with_tools) > 0 |
| |
|
