initial tests
This commit is contained in:
@@ -0,0 +1,38 @@
|
||||
from unittest.mock import AsyncMock, patch
|
||||
|
||||
import pytest
|
||||
|
||||
from agent.loop import run_turn
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_run_turn_basic(mock_anthropic_client):
|
||||
"""test that run_turn calls the API and returns a message"""
|
||||
|
||||
# patch the client with our mock
|
||||
with patch("agent.loop.client", mock_anthropic_client):
|
||||
result = await run_turn("What is 2+2?")
|
||||
|
||||
# verify client was called
|
||||
mock_anthropic_client.messages.create.assert_called_once()
|
||||
|
||||
# verify message returned
|
||||
assert result.content[0].text == "42"
|
||||
|
||||
# verify call has correct parameters
|
||||
call_args = mock_anthropic_client.messages.create.call_args
|
||||
assert call_args.kwargs["messages"][0]["content"] == "What is 2+2?"
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_run_turn_uses_settings(mock_anthropic_client, settings):
|
||||
"""Test that run_turn uses settings correctly."""
|
||||
|
||||
with patch("agent.loop.client", mock_anthropic_client):
|
||||
with patch("agent.loop.settings", settings):
|
||||
await run_turn("test message")
|
||||
|
||||
# Verify settings were used
|
||||
call_args = mock_anthropic_client.messages.create.call_args
|
||||
assert call_args.kwargs["model"] == settings.model
|
||||
assert call_args.kwargs["max_tokens"] == settings.max_tokens
|
||||
Reference in New Issue
Block a user