import asyncio
from unittest.mock import AsyncMock
import pytest
from windows_mcp.analytics import with_analytics
class TestWithAnalytics:
async def test_success_path(self):
mock_analytics = AsyncMock()
@with_analytics(mock_analytics, "test_tool")
async def my_tool():
return "result"
result = await my_tool()
assert result == "result"
mock_analytics.track_tool.assert_called_once()
call_args = mock_analytics.track_tool.call_args
assert call_args[0][0] == "test_tool"
assert call_args[0][1]["success"] is True
assert "duration_ms" in call_args[0][1]
async def test_error_path(self):
mock_analytics = AsyncMock()
@with_analytics(mock_analytics, "test_tool")
async def failing_tool():
raise ValueError("something broke")
with pytest.raises(ValueError, match="something broke"):
await failing_tool()
mock_analytics.track_error.assert_called_once()
call_args = mock_analytics.track_error.call_args
assert isinstance(call_args[0][0], ValueError)
assert call_args[0][1]["tool_name"] == "test_tool"
async def test_no_analytics_instance(self):
@with_analytics(None, "test_tool")
async def my_tool():
return 42
result = await my_tool()
assert result == 42
async def test_duration_measurement(self):
mock_analytics = AsyncMock()
@with_analytics(mock_analytics, "test_tool")
async def slow_tool():
await asyncio.sleep(0.1)
return "done"
await slow_tool()
call_args = mock_analytics.track_tool.call_args
duration_ms = call_args[0][1]["duration_ms"]
assert duration_ms >= 50 # Should be ~100ms but allow margin
async def test_error_still_tracks_duration(self):
mock_analytics = AsyncMock()
@with_analytics(mock_analytics, "test_tool")
async def failing_tool():
await asyncio.sleep(0.05)
raise RuntimeError("fail")
with pytest.raises(RuntimeError):
await failing_tool()
call_args = mock_analytics.track_error.call_args
assert "duration_ms" in call_args[0][1]
async def test_return_value_preserved(self):
mock_analytics = AsyncMock()
@with_analytics(mock_analytics, "test_tool")
async def tool_with_complex_result():
return {"key": "value", "count": 42}
result = await tool_with_complex_result()
assert result == {"key": "value", "count": 42}
