mirror of
https://github.com/sstent/FitTrack_GarminSync.git
synced 2026-01-25 08:35:23 +00:00
111 lines
4.2 KiB
Python
111 lines
4.2 KiB
Python
from datetime import datetime, timedelta
|
|
from unittest.mock import AsyncMock, patch
|
|
|
|
import pytest
|
|
|
|
from src.schemas import GarminCredentials
|
|
from src.services.garmin_auth_service import GarminAuthService
|
|
|
|
|
|
@pytest.fixture
|
|
def garmin_auth_service():
|
|
return GarminAuthService()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_initial_login_success(garmin_auth_service):
|
|
username = "test@example.com"
|
|
password = "password123"
|
|
|
|
with patch("src.services.garmin_auth_service.garth") as mock_garth:
|
|
mock_garth.Client.return_value = AsyncMock()
|
|
mock_garth.Client.return_value.login.return_value = (
|
|
None # garth.login doesn't return anything directly
|
|
)
|
|
# Mock the attributes that would be set on the client after login
|
|
mock_garth.Client.return_value.access_token = (
|
|
f"mock_access_token_for_{username}"
|
|
)
|
|
mock_garth.Client.return_value.access_token_secret = (
|
|
f"mock_access_token_secret_for_{username}"
|
|
)
|
|
mock_garth.Client.return_value.expires_in = 300
|
|
|
|
credentials = await garmin_auth_service.initial_login(username, password)
|
|
|
|
assert credentials is not None
|
|
assert credentials.garmin_username == username
|
|
assert credentials.garmin_password_plaintext == password
|
|
assert credentials.access_token.startswith("mock_access_token")
|
|
assert credentials.access_token_secret.startswith("mock_access_token_secret")
|
|
assert isinstance(credentials.token_expiration_date, datetime)
|
|
assert credentials.token_expiration_date > datetime.utcnow()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_initial_login_failure(garmin_auth_service):
|
|
username = "invalid@example.com"
|
|
password = "wrongpassword"
|
|
|
|
with patch("backend.src.services.garmin_auth_service.garth") as mock_garth:
|
|
mock_garth.Client.return_value = AsyncMock()
|
|
mock_garth.Client.return_value.login.side_effect = Exception(
|
|
"Garmin login failed"
|
|
)
|
|
|
|
credentials = await garmin_auth_service.initial_login(username, password)
|
|
|
|
assert credentials is None
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_refresh_tokens_success(garmin_auth_service):
|
|
credentials = GarminCredentials(
|
|
garmin_username="test@example.com",
|
|
garmin_password_plaintext="password123",
|
|
access_token="old_access_token",
|
|
access_token_secret="old_access_token_secret",
|
|
token_expiration_date=datetime.utcnow() - timedelta(minutes=1), # Expired token
|
|
)
|
|
|
|
with patch("backend.src.services.garmin_auth_service.garth") as mock_garth:
|
|
mock_garth.Client.return_value = AsyncMock()
|
|
mock_garth.Client.return_value.reauthorize.return_value = None
|
|
mock_garth.Client.return_value.access_token = "refreshed_access_token"
|
|
mock_garth.Client.return_value.access_token_secret = (
|
|
"refreshed_access_token_secret"
|
|
)
|
|
mock_garth.Client.return_value.expires_in = 300
|
|
|
|
refreshed_credentials = await garmin_auth_service.refresh_tokens(credentials)
|
|
|
|
assert refreshed_credentials is not None
|
|
assert refreshed_credentials.garmin_username == credentials.garmin_username
|
|
assert refreshed_credentials.access_token == "refreshed_access_token"
|
|
assert (
|
|
refreshed_credentials.access_token_secret == "refreshed_access_token_secret"
|
|
)
|
|
assert isinstance(refreshed_credentials.token_expiration_date, datetime)
|
|
assert refreshed_credentials.token_expiration_date > datetime.utcnow()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_refresh_tokens_failure(garmin_auth_service):
|
|
credentials = GarminCredentials(
|
|
garmin_username="test@example.com",
|
|
garmin_password_plaintext="invalid_password",
|
|
access_token="old_access_token",
|
|
access_token_secret="old_access_token_secret",
|
|
token_expiration_date=datetime.utcnow() - timedelta(minutes=1),
|
|
)
|
|
|
|
with patch("backend.src.services.garmin_auth_service.garth") as mock_garth:
|
|
mock_garth.Client.return_value = AsyncMock()
|
|
mock_garth.Client.return_value.reauthorize.side_effect = Exception(
|
|
"Garmin reauthorize failed"
|
|
)
|
|
|
|
refreshed_credentials = await garmin_auth_service.refresh_tokens(credentials)
|
|
|
|
assert refreshed_credentials is None
|