mirror of
https://github.com/fastapi-users/fastapi-users.git
synced 2025-11-04 14:45:50 +08:00
c4de66b81c
* Implement Transport classes * Implement authentication strategy classes * Revamp authentication with Transport and Strategy * Revamp strategy and OAuth so that they can use a callable dependency * Update docstring * Make ErrorCode a proper Enum and cleanup unused OpenAPI utils * Remove useless check * Tweak typing in authenticator * Update docs * Improve logout/destroy token logic * Update docs * Update docs * Update docs and full examples * Apply formatting to examples * Update OAuth doc and examples * Add migration doc * Implement Redis session token * Add Redis Session documentation * RedisSession -> Redis * Fix links in docs
120 lines
3.9 KiB
Python
120 lines
3.9 KiB
Python
import httpx
|
|
import pytest
|
|
from fastapi import FastAPI, status
|
|
|
|
from fastapi_users.fastapi_users import FastAPIUsers
|
|
from tests.conftest import User, UserCreate, UserDB, UserUpdate
|
|
|
|
|
|
@pytest.fixture
|
|
def fastapi_users(get_user_manager, mock_authentication) -> FastAPIUsers:
|
|
return FastAPIUsers[User, UserCreate, UserUpdate, UserDB](
|
|
get_user_manager,
|
|
[mock_authentication],
|
|
User,
|
|
UserCreate,
|
|
UserUpdate,
|
|
UserDB,
|
|
)
|
|
|
|
|
|
@pytest.fixture
|
|
def test_app(
|
|
fastapi_users: FastAPIUsers, secret, mock_authentication, oauth_client
|
|
) -> FastAPI:
|
|
app = FastAPI()
|
|
app.include_router(fastapi_users.get_register_router())
|
|
app.include_router(fastapi_users.get_reset_password_router())
|
|
app.include_router(fastapi_users.get_auth_router(mock_authentication))
|
|
app.include_router(
|
|
fastapi_users.get_oauth_router(oauth_client, mock_authentication, secret)
|
|
)
|
|
app.include_router(fastapi_users.get_users_router())
|
|
app.include_router(fastapi_users.get_verify_router())
|
|
|
|
return app
|
|
|
|
|
|
@pytest.fixture
|
|
@pytest.mark.asyncio
|
|
async def test_app_client(test_app, get_test_client):
|
|
async for client in get_test_client(test_app):
|
|
yield client
|
|
|
|
|
|
@pytest.fixture
|
|
def openapi_dict(test_app: FastAPI):
|
|
return test_app.openapi()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
@pytest.mark.openapi
|
|
async def test_openapi_route(test_app_client: httpx.AsyncClient):
|
|
response = await test_app_client.get("/openapi.json")
|
|
assert response.status_code == status.HTTP_200_OK
|
|
|
|
|
|
class TestReset:
|
|
def test_reset_password_status_codes(self, openapi_dict):
|
|
route = openapi_dict["paths"]["/reset-password"]["post"]
|
|
assert list(route["responses"].keys()) == ["200", "400", "422"]
|
|
|
|
def test_forgot_password_status_codes(self, openapi_dict):
|
|
route = openapi_dict["paths"]["/forgot-password"]["post"]
|
|
assert list(route["responses"].keys()) == ["202", "422"]
|
|
|
|
|
|
class TestUsers:
|
|
def test_patch_id_status_codes(self, openapi_dict):
|
|
route = openapi_dict["paths"]["/{id}"]["patch"]
|
|
assert list(route["responses"].keys()) == [
|
|
"200",
|
|
"401",
|
|
"403",
|
|
"404",
|
|
"400",
|
|
"422",
|
|
]
|
|
|
|
def test_delete_id_status_codes(self, openapi_dict):
|
|
route = openapi_dict["paths"]["/{id}"]["delete"]
|
|
assert list(route["responses"].keys()) == ["204", "401", "403", "404", "422"]
|
|
|
|
def test_get_id_status_codes(self, openapi_dict):
|
|
route = openapi_dict["paths"]["/{id}"]["get"]
|
|
assert list(route["responses"].keys()) == ["200", "401", "403", "404", "422"]
|
|
|
|
def test_patch_me_status_codes(self, openapi_dict):
|
|
route = openapi_dict["paths"]["/me"]["patch"]
|
|
assert list(route["responses"].keys()) == ["200", "401", "400", "422"]
|
|
|
|
def test_get_me_status_codes(self, openapi_dict):
|
|
route = openapi_dict["paths"]["/me"]["get"]
|
|
assert list(route["responses"].keys()) == ["200", "401"]
|
|
|
|
|
|
class TestRegister:
|
|
def test_register_status_codes(self, openapi_dict):
|
|
route = openapi_dict["paths"]["/register"]["post"]
|
|
assert list(route["responses"].keys()) == ["201", "400", "422"]
|
|
|
|
|
|
class TestVerify:
|
|
def test_verify_status_codes(self, openapi_dict):
|
|
route = openapi_dict["paths"]["/verify"]["post"]
|
|
assert list(route["responses"].keys()) == ["200", "400", "422"]
|
|
|
|
def test_request_verify_status_codes(self, openapi_dict):
|
|
route = openapi_dict["paths"]["/request-verify-token"]["post"]
|
|
assert list(route["responses"].keys()) == ["202", "422"]
|
|
|
|
|
|
class TestOAuth2:
|
|
def test_oauth_authorize_status_codes(self, openapi_dict):
|
|
route = openapi_dict["paths"]["/authorize"]["get"]
|
|
assert list(route["responses"].keys()) == ["200", "422"]
|
|
|
|
def test_oauth_callback_status_codes(self, openapi_dict):
|
|
route = openapi_dict["paths"]["/callback"]["get"]
|
|
assert list(route["responses"].keys()) == ["200", "400", "422"]
|