from unittest import mock
from unittest.mock import AsyncMock

import pytest
from fastapi import FastAPI, Request
from fastapi.testclient import TestClient

from prometheus.app.api.routes import user
from prometheus.app.entity.user import User
from prometheus.app.exception_handler import register_exception_handlers

app = FastAPI()
register_exception_handlers(app)
app.include_router(user.router, prefix="/user", tags=["user"])


@app.middleware("mock_jwt_middleware")
async def add_user_id(request: Request, call_next):
    request.state.user_id = 1  # Set user_id to 1 for testing purposes
    response = await call_next(request)
    return response


client = TestClient(app)


@pytest.fixture
def mock_service():
    service = mock.MagicMock()
    app.state.service = service
    yield service


def test_list(mock_service):
    # Mock user as admin and return a list of users
    mock_service["user_service"].list_users = AsyncMock(
        return_value=[
            User(
                id=1,
                username="testuser",
                email="test@gmail.com",
                password_hash="hashedpassword",
                github_token="ghp_1234567890abcdef1234567890abcdef1234",
                issue_credit=10,
                is_superuser=False,
            )
        ]
    )
    mock_service["user_service"].is_admin = AsyncMock(return_value=True)

    # Test the list endpoint
    response = client.get("user/list/")
    assert response.status_code == 200
    assert response.json() == {
        "code": 200,
        "message": "success",
        "data": [
            {
                "id": 1,
                "username": "testuser",
                "email": "test@gmail.com",
                "issue_credit": 10,
                "is_superuser": False,
            }
        ],
    }


def test_set_github_token(mock_service):
    # Mock user as admin and return a list of users
    mock_service["user_service"].set_github_token = AsyncMock(return_value=None)

    # Test the list endpoint
    response = client.put(
        "user/set-github-token/", json={"github_token": "ghp_1234567890abcdef1234567890abcdef1234"}
    )
    assert response.status_code == 200
    assert response.json() == {
        "code": 200,
        "message": "success",
        "data": None,
    }