test-writer

Test Writer

Generate comprehensive tests following testing best practices.

Test Types

Python Testing with Pytest

Unit Test Template

"""Test {module} {component}.

Tests Requirements: REQ-{XXX}-001
"""
import pytest
from unittest.mock import Mock, patch

from module import Component


@pytest.fixture
def component():
    """Create component with test dependencies."""
    return Component(
        dependency=Mock(),
        config={"test": True}
    )


class TestComponent:
    """Tests for Component class."""

    def test_method_returns_expected_when_valid_input(self, component):
        """Method returns expected result for valid input."""
        # Arrange
        input_data = {"key": "value"}
        expected = {"result": "processed"}

        # Act
        result = component.method(input_data)

        # Assert
        assert result == expected

    def test_method_raises_when_invalid_input(self, component):
        """Method raises ValueError for invalid input."""
        with pytest.raises(ValueError, match="Invalid input"):
            component.method(None)

    @pytest.mark.parametrize("input_val,expected", [
        ("a", 1),
        ("b", 2),
        ("c", 3),
    ])
    def test_method_handles_various_inputs(self, component, input_val, expected):
        """Method handles different input values correctly."""
        assert component.method(input_val) == expected

Integration Test Template

"""Integration tests for {feature}.

Tests end-to-end flow of {workflow}.
"""
import pytest
from httpx import AsyncClient

from app.main import app
from app.database import get_db


@pytest.fixture
async def client():
    """Create test client with database."""
    async with AsyncClient(app=app, base_url="http://test") as client:
        yield client


@pytest.fixture
async def db_session():
    """Create isolated database session."""
    # Setup
    session = await create_test_session()
    yield session
    # Teardown
    await session.rollback()


class TestFeatureWorkflow:
    """Integration tests for feature workflow."""

    @pytest.mark.asyncio
    async def test_complete_workflow(self, client, db_session):
        """Test complete create-read-update-delete workflow."""
        # Create
        response = await client.post("/api/items", json={"name": "test"})
        assert response.status_code == 201
        item_id = response.json()["id"]

        # Read
        response = await client.get(f"/api/items/{item_id}")
        assert response.status_code == 200
        assert response.json()["name"] == "test"

        # Update
        response = await client.put(
            f"/api/items/{item_id}",
            json={"name": "updated"}
        )
        assert response.status_code == 200

        # Delete
        response = await client.delete(f"/api/items/{item_id}")
        assert response.status_code == 204