87 lines
2.9 KiB
Python
87 lines
2.9 KiB
Python
# ================================================================
|
|
# services/data/tests/test_external.py
|
|
# ================================================================
|
|
"""Tests for external API clients"""
|
|
|
|
import pytest
|
|
from unittest.mock import AsyncMock, patch
|
|
from datetime import datetime, timedelta
|
|
|
|
from app.external.aemet import AEMETClient
|
|
from app.external.madrid_opendata import MadridOpenDataClient
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_aemet_client_synthetic_weather():
|
|
"""Test AEMET client synthetic weather generation"""
|
|
client = AEMETClient()
|
|
|
|
weather = await client._generate_synthetic_weather()
|
|
|
|
assert weather is not None
|
|
assert 'temperature' in weather
|
|
assert 'precipitation' in weather
|
|
assert 'humidity' in weather
|
|
assert weather['source'] == 'synthetic'
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_aemet_client_synthetic_forecast():
|
|
"""Test AEMET client synthetic forecast"""
|
|
client = AEMETClient()
|
|
|
|
forecast = await client._generate_synthetic_forecast(7)
|
|
|
|
assert len(forecast) == 7
|
|
assert all('forecast_date' in item for item in forecast)
|
|
assert all('temperature' in item for item in forecast)
|
|
assert all(item['source'] == 'synthetic' for item in forecast)
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_madrid_client_synthetic_traffic():
|
|
"""Test Madrid Open Data client synthetic traffic"""
|
|
client = MadridOpenDataClient()
|
|
|
|
traffic = await client._generate_synthetic_traffic(40.4168, -3.7038)
|
|
|
|
assert traffic is not None
|
|
assert 'traffic_volume' in traffic
|
|
assert 'pedestrian_count' in traffic
|
|
assert 'congestion_level' in traffic
|
|
assert traffic['congestion_level'] in ['low', 'medium', 'high']
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_madrid_client_historical_traffic():
|
|
"""Test Madrid Open Data client historical traffic"""
|
|
client = MadridOpenDataClient()
|
|
|
|
start_date = datetime.now() - timedelta(days=7)
|
|
end_date = datetime.now()
|
|
|
|
historical = await client._generate_historical_traffic(
|
|
40.4168, -3.7038, start_date, end_date
|
|
)
|
|
|
|
assert len(historical) > 0
|
|
assert all('traffic_volume' in item for item in historical)
|
|
assert all('date' in item for item in historical)
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_distance_calculation():
|
|
"""Test distance calculation in AEMET client"""
|
|
client = AEMETClient()
|
|
|
|
# Distance between Madrid Centro and Madrid Norte (approximately)
|
|
distance = client._calculate_distance(40.4168, -3.7038, 40.4518, -3.7246)
|
|
|
|
# Should be roughly 4-5 km
|
|
assert 3 < distance < 6
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_madrid_area_detection():
|
|
"""Test Madrid area detection"""
|
|
client = AEMETClient()
|
|
|
|
# Madrid coordinates should be detected as Madrid area
|
|
assert client._is_in_madrid_area(40.4168, -3.7038) == True
|
|
|
|
# Barcelona coordinates should not
|
|
assert client._is_in_madrid_area(41.3851, 2.1734) == False |