FitTrack_GarminSync/backend/tests/unit/test_auth_service.py

import pytest
from unittest.mock import AsyncMock, patch
from src.services.auth_service import AuthService
from src.schemas import UserCreate, User, TokenCreate, TokenUpdate
import uuid

@pytest.fixture
def auth_service():
    """Fixture for AuthService with mocked CentralDBService."""
    with patch('src.services.auth_service.CentralDBService') as MockCentralDBService:
        mock_central_db_instance = MockCentralDBService.return_value
        mock_central_db_instance.get_user_by_email = AsyncMock()
        mock_central_db_instance.create_user = AsyncMock()
        mock_central_db_instance.get_token = AsyncMock()
        mock_central_db_instance.create_token = AsyncMock()
        mock_central_db_instance.update_token = AsyncMock()
        service = AuthService()
        service.central_db = mock_central_db_instance
        yield service

@pytest.fixture
def mock_garth_login():
    """Fixture to mock garth.login."""
    with patch('garth.login') as mock_login:
        yield mock_login

@pytest.fixture
def mock_garth_client():
    """Fixture to mock garth.client attributes."""
    with patch('garth.client') as mock_client:
        mock_client.oauth2_token = "mock_oauth2_token"
        mock_client.refresh_token = "mock_refresh_token"
        mock_client.token_expires_at = 1234567890
        yield mock_client

@pytest.mark.asyncio
async def test_authenticate_garmin_connect_new_user_success(auth_service, mock_garth_login, mock_garth_client):
    """Test successful Garmin authentication with a new user."""
    email = "new_user@example.com"
    password = "password123"
    mock_user = User(id=uuid.uuid4(), name=email, email=email)

    auth_service.central_db.get_user_by_email.return_value = None
    auth_service.central_db.create_user.return_value = mock_user
    auth_service.central_db.get_token.return_value = None

    result = await auth_service.authenticate_garmin_connect(email, password)

    mock_garth_login.assert_called_once_with(email, password)
    auth_service.central_db.get_user_by_email.assert_called_once_with(email=email)
    auth_service.central_db.create_user.assert_called_once()
    auth_service.central_db.create_token.assert_called_once()
    auth_service.central_db.update_token.assert_not_called()

    assert result == {"message": "Garmin Connect authentication successful", "user_id": str(mock_user.id)}

@pytest.mark.asyncio
async def test_authenticate_garmin_connect_existing_user_success(auth_service, mock_garth_login, mock_garth_client):
    """Test successful Garmin authentication with an existing user and no existing token."""
    email = "existing_user@example.com"
    password = "password123"
    mock_user = User(id=uuid.uuid4(), name=email, email=email)

    auth_service.central_db.get_user_by_email.return_value = mock_user
    auth_service.central_db.get_token.return_value = None

    result = await auth_service.authenticate_garmin_connect(email, password)

    mock_garth_login.assert_called_once_with(email, password)
    auth_service.central_db.get_user_by_email.assert_called_once_with(email=email)
    auth_service.central_db.create_user.assert_not_called()
    auth_service.central_db.create_token.assert_called_once()
    auth_service.central_db.update_token.assert_not_called()

    assert result == {"message": "Garmin Connect authentication successful", "user_id": str(mock_user.id)}

@pytest.mark.asyncio
async def test_authenticate_garmin_connect_existing_user_existing_token_success(auth_service, mock_garth_login, mock_garth_client):
    """Test successful Garmin authentication with an existing user and existing token."""
    email = "existing_user_token@example.com"
    password = "password123"
    mock_user = User(id=uuid.uuid4(), name=email, email=email)
    mock_user_id = mock_user.id # Capture the generated UUID
    mock_existing_token = TokenCreate(
        access_token="old_access", refresh_token="old_refresh", expires_at=1111111111, user_id=mock_user_id
    )

    auth_service.central_db.get_user_by_email.return_value = mock_user
    auth_service.central_db.get_token.return_value = mock_existing_token

    result = await auth_service.authenticate_garmin_connect(email, password)

    mock_garth_login.assert_called_once_with(email, password)
    auth_service.central_db.get_user_by_email.assert_called_once_with(email=email)
    auth_service.central_db.create_user.assert_not_called()
    auth_service.central_db.get_token.assert_called_once_with(user_id=mock_user_id)
    auth_service.central_db.update_token.assert_called_once()
    auth_service.central_db.create_token.assert_not_called()

    assert result == {"message": "Garmin Connect authentication successful", "user_id": str(mock_user.id)}
@pytest.mark.asyncio
async def test_authenticate_garmin_connect_garmin_failure(auth_service, mock_garth_login):
    """Test Garmin authentication failure."""
    email = "fail_garmin@example.com"
    password = "password123"
    mock_garth_login.side_effect = Exception("Garmin login failed")

    result = await auth_service.authenticate_garmin_connect(email, password)

    mock_garth_login.assert_called_once_with(email, password)
    auth_service.central_db.get_user_by_email.assert_not_called()
    auth_service.central_db.create_user.assert_not_called()
    auth_service.central_db.get_token.assert_not_called()
    auth_service.central_db.create_token.assert_not_called()
    auth_service.central_db.update_token.assert_not_called()

    assert result is None

@pytest.mark.asyncio
async def test_authenticate_garmin_connect_central_db_user_creation_failure(auth_service, mock_garth_login, mock_garth_client):
    """Test CentralDB user creation failure."""
    email = "fail_user_create@example.com"
    password = "password123"

    auth_service.central_db.get_user_by_email.return_value = None
    auth_service.central_db.create_user.return_value = None

    result = await auth_service.authenticate_garmin_connect(email, password)

    mock_garth_login.assert_called_once_with(email, password)
    auth_service.central_db.get_user_by_email.assert_called_once_with(email=email)
    auth_service.central_db.create_user.assert_called_once()
    auth_service.central_db.get_token.assert_not_called()
    auth_service.central_db.create_token.assert_not_called()
    auth_service.central_db.update_token.assert_not_called()

    assert result is None