193 lines
7.0 KiB
Python
193 lines
7.0 KiB
Python
"""Tests for OAuth service."""
|
|
|
|
from typing import Any
|
|
from unittest.mock import AsyncMock, Mock, patch
|
|
|
|
import pytest
|
|
from httpx import AsyncClient
|
|
|
|
from app.services.oauth import (
|
|
GitHubOAuthProvider,
|
|
GoogleOAuthProvider,
|
|
OAuthService,
|
|
OAuthUserInfo,
|
|
)
|
|
|
|
|
|
class TestOAuthService:
|
|
"""Test OAuth service functionality."""
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_oauth_service_initialization(self, test_session: Any) -> None:
|
|
"""Test OAuth service initialization."""
|
|
oauth_service = OAuthService(test_session)
|
|
|
|
assert "google" in oauth_service.providers
|
|
assert "github" in oauth_service.providers
|
|
assert isinstance(oauth_service.providers["google"], GoogleOAuthProvider)
|
|
assert isinstance(oauth_service.providers["github"], GitHubOAuthProvider)
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_generate_state(self, test_session: Any) -> None:
|
|
"""Test state generation."""
|
|
oauth_service = OAuthService(test_session)
|
|
state = oauth_service.generate_state()
|
|
|
|
assert isinstance(state, str)
|
|
assert len(state) > 10 # Should be a reasonable length
|
|
|
|
# Generate another to ensure they're different
|
|
state2 = oauth_service.generate_state()
|
|
assert state != state2
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_provider_valid(self, test_session: Any) -> None:
|
|
"""Test getting valid OAuth provider."""
|
|
oauth_service = OAuthService(test_session)
|
|
|
|
google_provider = oauth_service.get_provider("google")
|
|
assert isinstance(google_provider, GoogleOAuthProvider)
|
|
assert google_provider.provider_name == "google"
|
|
|
|
github_provider = oauth_service.get_provider("github")
|
|
assert isinstance(github_provider, GitHubOAuthProvider)
|
|
assert github_provider.provider_name == "github"
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_provider_invalid(self, test_session: Any) -> None:
|
|
"""Test getting invalid OAuth provider."""
|
|
oauth_service = OAuthService(test_session)
|
|
|
|
with pytest.raises(Exception) as exc_info:
|
|
oauth_service.get_provider("invalid")
|
|
|
|
assert "Unsupported OAuth provider" in str(exc_info.value)
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_authorization_url(self, test_session: Any) -> None:
|
|
"""Test authorization URL generation."""
|
|
oauth_service = OAuthService(test_session)
|
|
state = "test_state_123"
|
|
|
|
# Test Google
|
|
google_url = oauth_service.get_authorization_url("google", state)
|
|
assert "accounts.google.com" in google_url
|
|
assert "client_id=" in google_url
|
|
assert f"state={state}" in google_url
|
|
|
|
# Test GitHub
|
|
github_url = oauth_service.get_authorization_url("github", state)
|
|
assert "github.com" in github_url
|
|
assert "client_id=" in github_url
|
|
assert f"state={state}" in github_url
|
|
|
|
|
|
class TestGoogleOAuthProvider:
|
|
"""Test Google OAuth provider."""
|
|
|
|
def test_provider_properties(self) -> None:
|
|
"""Test Google provider properties."""
|
|
provider = GoogleOAuthProvider("test_client_id", "test_secret")
|
|
|
|
assert provider.provider_name == "google"
|
|
assert "accounts.google.com" in provider.authorization_url
|
|
assert "oauth2.googleapis.com" in provider.token_url
|
|
assert "googleapis.com" in provider.user_info_url
|
|
assert "openid email profile" in provider.scope
|
|
|
|
def test_authorization_url_generation(self) -> None:
|
|
"""Test authorization URL generation."""
|
|
provider = GoogleOAuthProvider("test_client_id", "test_secret")
|
|
state = "test_state"
|
|
|
|
auth_url = provider.get_authorization_url(state)
|
|
|
|
assert "accounts.google.com" in auth_url
|
|
assert "client_id=test_client_id" in auth_url
|
|
assert f"state={state}" in auth_url
|
|
assert "scope=openid+email+profile" in auth_url
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_user_info_success(self) -> None:
|
|
"""Test successful user info retrieval."""
|
|
provider = GoogleOAuthProvider("test_client_id", "test_secret")
|
|
|
|
mock_response_data = {
|
|
"id": "google_user_123",
|
|
"email": "test@gmail.com",
|
|
"name": "Test User",
|
|
"picture": "https://example.com/avatar.jpg",
|
|
}
|
|
|
|
with patch("httpx.AsyncClient.get") as mock_get:
|
|
mock_response = Mock()
|
|
mock_response.status_code = 200
|
|
mock_response.json.return_value = mock_response_data
|
|
mock_get.return_value = mock_response
|
|
|
|
user_info = await provider.get_user_info("test_access_token")
|
|
|
|
assert user_info.provider == "google"
|
|
assert user_info.provider_user_id == "google_user_123"
|
|
assert user_info.email == "test@gmail.com"
|
|
assert user_info.name == "Test User"
|
|
assert user_info.picture == "https://example.com/avatar.jpg"
|
|
|
|
|
|
class TestGitHubOAuthProvider:
|
|
"""Test GitHub OAuth provider."""
|
|
|
|
def test_provider_properties(self) -> None:
|
|
"""Test GitHub provider properties."""
|
|
provider = GitHubOAuthProvider("test_client_id", "test_secret")
|
|
|
|
assert provider.provider_name == "github"
|
|
assert "github.com" in provider.authorization_url
|
|
assert "github.com" in provider.token_url
|
|
assert "api.github.com" in provider.user_info_url
|
|
assert "user:email" in provider.scope
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_user_info_success(self) -> None:
|
|
"""Test successful user info retrieval."""
|
|
provider = GitHubOAuthProvider("test_client_id", "test_secret")
|
|
|
|
mock_user_data = {
|
|
"id": 123456,
|
|
"login": "testuser",
|
|
"name": "Test User",
|
|
"avatar_url": "https://github.com/avatar.jpg",
|
|
}
|
|
|
|
mock_emails_data = [
|
|
{"email": "test@example.com", "primary": True, "verified": True},
|
|
{"email": "secondary@example.com", "primary": False, "verified": True},
|
|
]
|
|
|
|
with patch("httpx.AsyncClient.get") as mock_get:
|
|
# Mock user profile response
|
|
mock_user_response = Mock()
|
|
mock_user_response.status_code = 200
|
|
mock_user_response.json.return_value = mock_user_data
|
|
|
|
# Mock emails response
|
|
mock_emails_response = Mock()
|
|
mock_emails_response.status_code = 200
|
|
mock_emails_response.json.return_value = mock_emails_data
|
|
|
|
# Return different responses based on URL
|
|
def side_effect(url, **kwargs):
|
|
if "user/emails" in str(url):
|
|
return mock_emails_response
|
|
return mock_user_response
|
|
|
|
mock_get.side_effect = side_effect
|
|
|
|
user_info = await provider.get_user_info("test_access_token")
|
|
|
|
assert user_info.provider == "github"
|
|
assert user_info.provider_user_id == "123456"
|
|
assert user_info.email == "test@example.com"
|
|
assert user_info.name == "Test User"
|
|
assert user_info.picture == "https://github.com/avatar.jpg"
|