Strava MCP Server

from datetime import datetime from unittest.mock import AsyncMock, patch import pytest from strava_mcp.config import StravaSettings from strava_mcp.models import Activity, DetailedActivity, Segment, SegmentEffort from strava_mcp.service import StravaService @pytest.fixture def settings(): return StravaSettings( client_id="test_client_id", client_secret="test_client_secret", refresh_token="test_refresh_token", base_url="https://www.strava.com/api/v3", ) @pytest.fixture def mock_api(): mock = AsyncMock() mock.get_activities = AsyncMock() mock.get_activity = AsyncMock() mock.get_activity_segments = AsyncMock() return mock @pytest.fixture def service(settings, mock_api): with patch("strava_mcp.service.StravaAPI", return_value=mock_api): service = StravaService(settings) yield service @pytest.mark.asyncio async def test_get_activities(service, mock_api): # Setup mocked data mock_activity = Activity( id=1234567890, name="Morning Run", distance=5000, moving_time=1200, elapsed_time=1300, total_elevation_gain=50, type="Run", sport_type="Run", start_date=datetime.fromisoformat("2023-01-01T10:00:00+00:00"), start_date_local=datetime.fromisoformat("2023-01-01T10:00:00+00:00"), timezone="Europe/London", achievement_count=2, kudos_count=5, comment_count=0, athlete_count=1, photo_count=0, trainer=False, commute=False, manual=False, private=False, flagged=False, average_speed=4.167, max_speed=5.3, has_heartrate=True, average_heartrate=140, max_heartrate=160, # Add required fields with default values map=None, workout_type=None, elev_high=None, elev_low=None, ) mock_api.get_activities.return_value = [mock_activity] # Test get_activities activities = await service.get_activities() # Verify API call mock_api.get_activities.assert_called_once_with(None, None, 1, 30) # Verify response assert len(activities) == 1 assert activities[0] == mock_activity @pytest.mark.asyncio async def test_get_activity(service, mock_api): # Setup mocked data mock_activity = DetailedActivity( id=1234567890, name="Morning Run", distance=5000, moving_time=1200, elapsed_time=1300, total_elevation_gain=50, type="Run", sport_type="Run", start_date=datetime.fromisoformat("2023-01-01T10:00:00+00:00"), start_date_local=datetime.fromisoformat("2023-01-01T10:00:00+00:00"), timezone="Europe/London", achievement_count=2, kudos_count=5, comment_count=0, athlete_count=1, photo_count=0, trainer=False, commute=False, manual=False, private=False, flagged=False, average_speed=4.167, max_speed=5.3, has_heartrate=True, average_heartrate=140, max_heartrate=160, athlete={"id": 123}, description="Test description", # Add required fields with default values map=None, workout_type=None, elev_high=None, elev_low=None, calories=None, segment_efforts=None, splits_metric=None, splits_standard=None, best_efforts=None, photos=None, gear=None, device_name=None, ) mock_api.get_activity.return_value = mock_activity # Test get_activity activity = await service.get_activity(1234567890) # Verify API call mock_api.get_activity.assert_called_once_with(1234567890, False) # Verify response assert activity == mock_activity @pytest.mark.asyncio async def test_get_activity_segments(service, mock_api): # Setup mocked data mock_segment = SegmentEffort( id=67890, activity_id=1234567890, segment_id=12345, name="Test Segment", elapsed_time=180, moving_time=180, start_date=datetime.fromisoformat("2023-01-01T10:05:00+00:00"), start_date_local=datetime.fromisoformat("2023-01-01T10:05:00+00:00"), distance=1000, athlete={"id": 123}, segment=Segment( id=12345, name="Test Segment", activity_type="Run", distance=1000, average_grade=5.0, maximum_grade=10.0, elevation_high=200, elevation_low=150, total_elevation_gain=50, start_latlng=[51.5, -0.1], end_latlng=[51.5, -0.2], climb_category=0, private=False, starred=False, city=None, state=None, country=None, ), # Add required fields with default values average_watts=None, device_watts=None, average_heartrate=None, max_heartrate=None, pr_rank=None, achievements=None, ) mock_api.get_activity_segments.return_value = [mock_segment] # Test get_activity_segments segments = await service.get_activity_segments(1234567890) # Verify API call mock_api.get_activity_segments.assert_called_once_with(1234567890) # Verify response assert len(segments) == 1 assert segments[0] == mock_segment