# tests/api/test_auth_endpoints.py
"""
Tests for authentication endpoints.
"""
import pytest
import pytest_asyncio
from unittest.mock import patch, MagicMock
from fastapi import status
from sqlalchemy import select

from app.models.user import User
from app.schemas.users import UserCreate


# Disable rate limiting for tests
@pytest.fixture(autouse=True)
def disable_rate_limit():
    """Disable rate limiting for all tests in this module."""
    with patch('app.api.routes.auth.limiter.enabled', False):
        yield


class TestRegisterEndpoint:
    """Tests for POST /auth/register endpoint."""

    @pytest.mark.asyncio
    async def test_register_success(self, client):
        """Test successful user registration."""
        response = await client.post(
            "/api/v1/auth/register",
            json={
                "email": "newuser@example.com",
                "password": "SecurePassword123!",
                "first_name": "New",
                "last_name": "User"
            }
        )

        assert response.status_code == status.HTTP_201_CREATED
        data = response.json()
        assert data["email"] == "newuser@example.com"
        assert data["first_name"] == "New"
        assert "password" not in data

    @pytest.mark.asyncio
    async def test_register_duplicate_email(self, client, async_test_user):
        """Test registering with existing email.

        Note: Returns 400 with generic message to prevent user enumeration.
        """
        response = await client.post(
            "/api/v1/auth/register",
            json={
                "email": async_test_user.email,
                "password": "SecurePassword123!",
                "first_name": "Duplicate",
                "last_name": "User"
            }
        )

        # Security: Returns 400 with generic message to prevent email enumeration
        assert response.status_code == status.HTTP_400_BAD_REQUEST
        data = response.json()
        assert data["success"] is False
        assert "registration failed" in data["errors"][0]["message"].lower()

    @pytest.mark.asyncio
    async def test_register_weak_password(self, client):
        """Test registration with weak password."""
        response = await client.post(
            "/api/v1/auth/register",
            json={
                "email": "weakpass@example.com",
                "password": "weak",
                "first_name": "Weak",
                "last_name": "Pass"
            }
        )

        assert response.status_code == status.HTTP_422_UNPROCESSABLE_ENTITY

    @pytest.mark.asyncio
    async def test_register_unexpected_error(self, client):
        """Test registration with unexpected error."""
        with patch('app.services.auth_service.AuthService.create_user') as mock_create:
            mock_create.side_effect = Exception("Unexpected error")

            response = await client.post(
                "/api/v1/auth/register",
                json={
                    "email": "error@example.com",
                    "password": "SecurePassword123!",
                    "first_name": "Error",
                    "last_name": "User"
                }
            )

            assert response.status_code == status.HTTP_500_INTERNAL_SERVER_ERROR


class TestLoginEndpoint:
    """Tests for POST /auth/login endpoint."""

    @pytest.mark.asyncio
    async def test_login_success(self, client, async_test_user):
        """Test successful login."""
        response = await client.post(
            "/api/v1/auth/login",
            json={
                "email": async_test_user.email,
                "password": "TestPassword123!"
            }
        )

        assert response.status_code == status.HTTP_200_OK
        data = response.json()
        assert "access_token" in data
        assert "refresh_token" in data
        assert data["token_type"] == "bearer"

    @pytest.mark.asyncio
    async def test_login_wrong_password(self, client, async_test_user):
        """Test login with wrong password."""
        response = await client.post(
            "/api/v1/auth/login",
            json={
                "email": async_test_user.email,
                "password": "WrongPassword123"
            }
        )

        assert response.status_code == status.HTTP_401_UNAUTHORIZED

    @pytest.mark.asyncio
    async def test_login_nonexistent_user(self, client):
        """Test login with non-existent email."""
        response = await client.post(
            "/api/v1/auth/login",
            json={
                "email": "nonexistent@example.com",
                "password": "Password123!"
            }
        )

        assert response.status_code == status.HTTP_401_UNAUTHORIZED

    @pytest.mark.asyncio
    async def test_login_inactive_user(self, client, async_test_user, async_test_db):
        """Test login with inactive user."""
        test_engine, AsyncTestingSessionLocal = async_test_db
        async with AsyncTestingSessionLocal() as session:
            # Get the user in this session and make it inactive
            result = await session.execute(select(User).where(User.id == async_test_user.id))
            user_in_session = result.scalar_one_or_none()
            user_in_session.is_active = False
            await session.commit()

        response = await client.post(
            "/api/v1/auth/login",
            json={
                "email": async_test_user.email,
                "password": "TestPassword123!"
            }
        )

        assert response.status_code == status.HTTP_401_UNAUTHORIZED

    @pytest.mark.asyncio
    async def test_login_unexpected_error(self, client, async_test_user):
        """Test login with unexpected error."""
        with patch('app.services.auth_service.AuthService.authenticate_user') as mock_auth:
            mock_auth.side_effect = Exception("Database error")

            response = await client.post(
                "/api/v1/auth/login",
                json={
                    "email": async_test_user.email,
                    "password": "TestPassword123!"
                }
            )

            assert response.status_code == status.HTTP_500_INTERNAL_SERVER_ERROR


class TestOAuthLoginEndpoint:
    """Tests for POST /auth/login/oauth endpoint."""

    @pytest.mark.asyncio
    async def test_oauth_login_success(self, client, async_test_user):
        """Test successful OAuth login."""
        response = await client.post(
            "/api/v1/auth/login/oauth",
            data={
                "username": async_test_user.email,
                "password": "TestPassword123!"
            }
        )

        assert response.status_code == status.HTTP_200_OK
        data = response.json()
        assert "access_token" in data
        assert "refresh_token" in data

    @pytest.mark.asyncio
    async def test_oauth_login_wrong_credentials(self, client, async_test_user):
        """Test OAuth login with wrong credentials."""
        response = await client.post(
            "/api/v1/auth/login/oauth",
            data={
                "username": async_test_user.email,
                "password": "WrongPassword"
            }
        )

        assert response.status_code == status.HTTP_401_UNAUTHORIZED

    @pytest.mark.asyncio
    async def test_oauth_login_inactive_user(self, client, async_test_user, async_test_db):
        """Test OAuth login with inactive user."""
        test_engine, AsyncTestingSessionLocal = async_test_db
        async with AsyncTestingSessionLocal() as session:
            # Get the user in this session and make it inactive
            result = await session.execute(select(User).where(User.id == async_test_user.id))
            user_in_session = result.scalar_one_or_none()
            user_in_session.is_active = False
            await session.commit()

        response = await client.post(
            "/api/v1/auth/login/oauth",
            data={
                "username": async_test_user.email,
                "password": "TestPassword123!"
            }
        )

        assert response.status_code == status.HTTP_401_UNAUTHORIZED

    @pytest.mark.asyncio
    async def test_oauth_login_unexpected_error(self, client, async_test_user):
        """Test OAuth login with unexpected error."""
        with patch('app.services.auth_service.AuthService.authenticate_user') as mock_auth:
            mock_auth.side_effect = Exception("Unexpected error")

            response = await client.post(
                "/api/v1/auth/login/oauth",
                data={
                    "username": async_test_user.email,
                    "password": "TestPassword123!"
                }
            )

            assert response.status_code == status.HTTP_500_INTERNAL_SERVER_ERROR


class TestRefreshTokenEndpoint:
    """Tests for POST /auth/refresh endpoint."""

    @pytest.mark.asyncio
    async def test_refresh_token_success(self, client, async_test_user):
        """Test successful token refresh."""
        # First, login to get a refresh token
        login_response = await client.post(
            "/api/v1/auth/login",
            json={
                "email": async_test_user.email,
                "password": "TestPassword123!"
            }
        )
        refresh_token = login_response.json()["refresh_token"]

        # Now refresh the token
        response = await client.post(
            "/api/v1/auth/refresh",
            json={"refresh_token": refresh_token}
        )

        assert response.status_code == status.HTTP_200_OK
        data = response.json()
        assert "access_token" in data
        assert "refresh_token" in data

    @pytest.mark.asyncio
    async def test_refresh_token_expired(self, client):
        """Test refresh with expired token."""
        from app.core.auth import TokenExpiredError

        with patch('app.services.auth_service.AuthService.refresh_tokens') as mock_refresh:
            mock_refresh.side_effect = TokenExpiredError("Token expired")

            response = await client.post(
                "/api/v1/auth/refresh",
                json={"refresh_token": "some_token"}
            )

            assert response.status_code == status.HTTP_401_UNAUTHORIZED

    @pytest.mark.asyncio
    async def test_refresh_token_invalid(self, client):
        """Test refresh with invalid token."""
        response = await client.post(
            "/api/v1/auth/refresh",
            json={"refresh_token": "invalid_token"}
        )

        assert response.status_code == status.HTTP_401_UNAUTHORIZED

    @pytest.mark.asyncio
    async def test_refresh_token_unexpected_error(self, client, async_test_user):
        """Test refresh with unexpected error."""
        # Get a valid refresh token first
        login_response = await client.post(
            "/api/v1/auth/login",
            json={
                "email": async_test_user.email,
                "password": "TestPassword123!"
            }
        )
        refresh_token = login_response.json()["refresh_token"]

        with patch('app.services.auth_service.AuthService.refresh_tokens') as mock_refresh:
            mock_refresh.side_effect = Exception("Unexpected error")

            response = await client.post(
                "/api/v1/auth/refresh",
                json={"refresh_token": refresh_token}
            )

            assert response.status_code == status.HTTP_500_INTERNAL_SERVER_ERROR