Testing

Testing Patterns

Agent Test

import pytest
from unittest.mock import AsyncMock, MagicMock
from src.agents.quality_agent import QualityAgent

@pytest.fixture
def mock_retriever():
    retriever = AsyncMock()
    retriever.retrieve.return_value = [
        RetrievedSegment(content="Built REST API with 99.9% uptime", score=0.89)
    ]
    return retriever

@pytest.fixture
def quality_agent(mock_retriever):
    return QualityAgent(retriever=mock_retriever, llm_client=AsyncMock())

@pytest.mark.asyncio
async def test_quality_agent_strengthens_weak_bullet(quality_agent):
    # Arrange
    input = AgentInput(content="Worked on API")

    # Act
    result = await quality_agent.execute(input)

    # Assert
    assert result.quality_score >= 0.7
    assert any(char.isdigit() for char in result.content)

API Test