108 lines
4.1 KiB
Python
108 lines
4.1 KiB
Python
import pytest
|
|
from fastapi.testclient import TestClient
|
|
from unittest.mock import MagicMock, patch
|
|
from ea_chatbot.api.main import app
|
|
from ea_chatbot.history.models import User
|
|
|
|
# We will need to mock HistoryManager and get_db dependencies later
|
|
# For now, we define the expected behavior of the auth endpoints.
|
|
|
|
client = TestClient(app)
|
|
|
|
@pytest.fixture
|
|
def mock_user():
|
|
return User(
|
|
id="user-123",
|
|
username="test@example.com",
|
|
display_name="Test User",
|
|
password_hash="hashed_password"
|
|
)
|
|
|
|
def test_register_user_success():
|
|
"""Test successful user registration."""
|
|
# We mock it where it is used in the router
|
|
with patch("ea_chatbot.api.routers.auth.history_manager") as mock_hm:
|
|
mock_hm.get_user.return_value = None
|
|
mock_hm.create_user.return_value = User(id="1", username="new@example.com", display_name="New")
|
|
|
|
response = client.post(
|
|
"/auth/register",
|
|
json={"email": "new@example.com", "password": "password123", "display_name": "New"}
|
|
)
|
|
|
|
assert response.status_code == 201
|
|
assert response.json()["email"] == "new@example.com"
|
|
|
|
def test_login_success():
|
|
"""Test successful login and JWT return."""
|
|
with patch("ea_chatbot.api.routers.auth.history_manager") as mock_hm:
|
|
mock_hm.authenticate_user.return_value = User(id="1", username="test@example.com")
|
|
|
|
response = client.post(
|
|
"/auth/login",
|
|
data={"username": "test@example.com", "password": "password123"}
|
|
)
|
|
|
|
assert response.status_code == 200
|
|
assert "access_token" in response.json()
|
|
assert response.json()["token_type"] == "bearer"
|
|
|
|
def test_login_invalid_credentials():
|
|
"""Test login with wrong password."""
|
|
with patch("ea_chatbot.api.routers.auth.history_manager") as mock_hm:
|
|
mock_hm.authenticate_user.return_value = None
|
|
|
|
response = client.post(
|
|
"/auth/login",
|
|
data={"username": "test@example.com", "password": "wrongpassword"}
|
|
)
|
|
|
|
assert response.status_code == 401
|
|
assert "detail" in response.json()
|
|
|
|
def test_protected_route_without_token():
|
|
"""Test that protected routes require a token."""
|
|
response = client.get("/auth/me")
|
|
assert response.status_code == 401
|
|
|
|
def test_oidc_login_redirect():
|
|
"""Test that OIDC login returns a redirect URL."""
|
|
with patch("ea_chatbot.api.routers.auth.oidc_client") as mock_oidc:
|
|
mock_oidc.get_login_url.return_value = "https://oidc-provider.com/auth"
|
|
|
|
response = client.get("/auth/oidc/login")
|
|
assert response.status_code == 200
|
|
assert response.json()["url"] == "https://oidc-provider.com/auth"
|
|
|
|
def test_oidc_callback_success():
|
|
"""Test successful OIDC callback and JWT issuance."""
|
|
with patch("ea_chatbot.api.routers.auth.oidc_client") as mock_oidc, \
|
|
patch("ea_chatbot.api.routers.auth.history_manager") as mock_hm:
|
|
|
|
mock_oidc.exchange_code_for_token.return_value = {"access_token": "oidc-token"}
|
|
mock_oidc.get_user_info.return_value = {"email": "sso@example.com", "name": "SSO User"}
|
|
mock_hm.sync_user_from_oidc.return_value = User(id="sso-123", username="sso@example.com", display_name="SSO User")
|
|
|
|
response = client.get("/auth/oidc/callback?code=some-code")
|
|
|
|
assert response.status_code == 200
|
|
assert "access_token" in response.json()
|
|
assert response.json()["token_type"] == "bearer"
|
|
|
|
def test_get_me_success():
|
|
"""Test getting current user with a valid token."""
|
|
from ea_chatbot.api.utils import create_access_token
|
|
token = create_access_token(data={"sub": "123"})
|
|
|
|
with patch("ea_chatbot.api.dependencies.history_manager") as mock_hm:
|
|
mock_hm.get_user_by_id.return_value = User(id="123", username="test@example.com", display_name="Test")
|
|
|
|
response = client.get(
|
|
"/auth/me",
|
|
headers={"Authorization": f"Bearer {token}"}
|
|
)
|
|
|
|
assert response.status_code == 200
|
|
assert response.json()["email"] == "test@example.com"
|
|
assert response.json()["id"] == "123"
|