feat: Initial commit of FitTrack_GarminSync project

backend/tests/unit/test_auth_service.py

@@ -0,0 +1,137 @@
+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