mirror of
https://github.com/fastapi-users/fastapi-users.git
synced 2025-08-14 18:58:10 +08:00
c759bb6915
* Added login endpoint docs
* make format
* Changed login route into multiple examples.
* Added reset password router docs
* Updated /{id} routes for user
* Updated /me routes
* Fixed user already exists response description
* Updated the /register route
* Updated verify routes
* Updated oauth2 endpoints.
* Applied `make format`
* Renamed Authentication methods for getting their openapi schemas
- `get_login_responses_success` -> `get_openapi_login_responses_success`
- `get_logout_responses_success` -> `get_openapi_logout_responses_success`
* Fixed flake8 errors
* Not using `Final` to keep python37 compatibility
Co-authored-by: François Voron <fvoron@gmail.com>
137 lines
4.9 KiB
Python
137 lines
4.9 KiB
Python
import pytest
|
|
from fastapi import FastAPI
|
|
from fastapi.testclient import TestClient
|
|
from httpx_oauth.clients.google import GoogleOAuth2
|
|
|
|
import fastapi_users.authentication
|
|
from fastapi_users import models
|
|
from fastapi_users.fastapi_users import FastAPIUsers
|
|
|
|
app = FastAPI()
|
|
jwt_authentication = fastapi_users.authentication.JWTAuthentication(
|
|
secret="", lifetime_seconds=3600
|
|
)
|
|
cookie_authentication = fastapi_users.authentication.CookieAuthentication(secret="")
|
|
users = FastAPIUsers(
|
|
# dummy get_user_manager
|
|
lambda: None,
|
|
[cookie_authentication, jwt_authentication],
|
|
models.BaseUser,
|
|
models.BaseUserCreate,
|
|
models.BaseUserUpdate,
|
|
models.BaseUserDB,
|
|
)
|
|
app.include_router(users.get_verify_router())
|
|
app.include_router(users.get_register_router())
|
|
app.include_router(users.get_users_router())
|
|
app.include_router(users.get_reset_password_router())
|
|
app.include_router(
|
|
users.get_oauth_router(
|
|
GoogleOAuth2(client_id="1234", client_secret="4321"), state_secret="secret"
|
|
)
|
|
)
|
|
app.include_router(users.get_auth_router(jwt_authentication), prefix="/jwt")
|
|
app.include_router(users.get_auth_router(cookie_authentication), prefix="/cookie")
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def get_openapi_dict():
|
|
return app.openapi()
|
|
|
|
|
|
def test_openapi_generated_ok():
|
|
assert TestClient(app).get("/openapi.json").status_code == 200
|
|
|
|
|
|
class TestLogin:
|
|
def test_jwt_login_status_codes(self, get_openapi_dict):
|
|
route = get_openapi_dict["paths"]["/jwt/login"]["post"]
|
|
assert list(route["responses"].keys()) == ["200", "400", "422"]
|
|
|
|
def test_jwt_login_200_body(self, get_openapi_dict):
|
|
"""Check if example is up to date."""
|
|
example = get_openapi_dict["paths"]["/jwt/login"]["post"]["responses"]["200"][
|
|
"content"
|
|
]["application/json"]["example"]
|
|
assert (
|
|
example.keys()
|
|
== fastapi_users.authentication.jwt.JWTLoginResponse.schema()[
|
|
"properties"
|
|
].keys()
|
|
)
|
|
|
|
def test_cookie_login_status_codes(self, get_openapi_dict):
|
|
route = get_openapi_dict["paths"]["/cookie/login"]["post"]
|
|
assert ["200", "400", "422"] == list(route["responses"].keys())
|
|
|
|
|
|
class TestLogout:
|
|
def test_cookie_logout_status_codes(self, get_openapi_dict):
|
|
route = get_openapi_dict["paths"]["/cookie/logout"]["post"]
|
|
assert list(route["responses"].keys()) == ["200", "401"]
|
|
|
|
|
|
class TestReset:
|
|
def test_reset_password_status_codes(self, get_openapi_dict):
|
|
route = get_openapi_dict["paths"]["/reset-password"]["post"]
|
|
assert list(route["responses"].keys()) == ["200", "400", "422"]
|
|
|
|
def test_forgot_password_status_codes(self, get_openapi_dict):
|
|
route = get_openapi_dict["paths"]["/forgot-password"]["post"]
|
|
assert list(route["responses"].keys()) == ["202", "422"]
|
|
|
|
|
|
class TestUsers:
|
|
def test_patch_id_status_codes(self, get_openapi_dict):
|
|
route = get_openapi_dict["paths"]["/{id}"]["patch"]
|
|
assert list(route["responses"].keys()) == [
|
|
"200",
|
|
"401",
|
|
"403",
|
|
"404",
|
|
"400",
|
|
"422",
|
|
]
|
|
|
|
def test_delete_id_status_codes(self, get_openapi_dict):
|
|
route = get_openapi_dict["paths"]["/{id}"]["delete"]
|
|
assert list(route["responses"].keys()) == ["204", "401", "403", "404", "422"]
|
|
|
|
def test_get_id_status_codes(self, get_openapi_dict):
|
|
route = get_openapi_dict["paths"]["/{id}"]["get"]
|
|
assert list(route["responses"].keys()) == ["200", "401", "403", "404", "422"]
|
|
|
|
def test_patch_me_status_codes(self, get_openapi_dict):
|
|
route = get_openapi_dict["paths"]["/me"]["patch"]
|
|
assert list(route["responses"].keys()) == ["200", "401", "400", "422"]
|
|
|
|
def test_get_me_status_codes(self, get_openapi_dict):
|
|
route = get_openapi_dict["paths"]["/me"]["get"]
|
|
assert list(route["responses"].keys()) == ["200", "401"]
|
|
|
|
|
|
class TestRegister:
|
|
def test_register_status_codes(self, get_openapi_dict):
|
|
route = get_openapi_dict["paths"]["/register"]["post"]
|
|
assert list(route["responses"].keys()) == ["201", "400", "422"]
|
|
|
|
|
|
class TestVerify:
|
|
def test_verify_status_codes(self, get_openapi_dict):
|
|
route = get_openapi_dict["paths"]["/verify"]["post"]
|
|
assert list(route["responses"].keys()) == ["200", "400", "422"]
|
|
|
|
def test_request_verify_status_codes(self, get_openapi_dict):
|
|
route = get_openapi_dict["paths"]["/request-verify-token"]["post"]
|
|
assert list(route["responses"].keys()) == ["202", "422"]
|
|
|
|
|
|
class TestOAuth2:
|
|
def test_google_authorize_status_codes(self, get_openapi_dict):
|
|
route = get_openapi_dict["paths"]["/authorize"]["get"]
|
|
assert list(route["responses"].keys()) == ["200", "422"]
|
|
|
|
def test_google_callback_status_codes(self, get_openapi_dict):
|
|
route = get_openapi_dict["paths"]["/callback"]["get"]
|
|
assert list(route["responses"].keys()) == ["200", "400", "422"]
|