mirror of
https://github.com/fastapi/fastapi.git
synced 2026-02-01 18:01:14 -05:00
73c411e1b9
Co-authored-by: Motov Yurii <109919500+YuriiMotov@users.noreply.github.com> Co-authored-by: Yurii Motov <yurii.motov.monte@gmail.com> Co-authored-by: Sebastián Ramírez <tiangolo@gmail.com>
78 lines
1.6 KiB
Python
78 lines
1.6 KiB
Python
from functools import wraps
|
|
from typing import AsyncGenerator, Generator
|
|
|
|
import pytest
|
|
from fastapi import Depends, FastAPI
|
|
from fastapi.testclient import TestClient
|
|
|
|
|
|
def noop_wrap(func):
|
|
@wraps(func)
|
|
def wrapper(*args, **kwargs):
|
|
return func(*args, **kwargs)
|
|
|
|
return wrapper
|
|
|
|
|
|
app = FastAPI()
|
|
|
|
|
|
@noop_wrap
|
|
def wrapped_dependency() -> bool:
|
|
return True
|
|
|
|
|
|
@noop_wrap
|
|
def wrapped_gen_dependency() -> Generator[bool, None, None]:
|
|
yield True
|
|
|
|
|
|
@noop_wrap
|
|
async def async_wrapped_dependency() -> bool:
|
|
return True
|
|
|
|
|
|
@noop_wrap
|
|
async def async_wrapped_gen_dependency() -> AsyncGenerator[bool, None]:
|
|
yield True
|
|
|
|
|
|
@app.get("/wrapped-dependency/")
|
|
async def get_wrapped_dependency(value: bool = Depends(wrapped_dependency)):
|
|
return value
|
|
|
|
|
|
@app.get("/wrapped-gen-dependency/")
|
|
async def get_wrapped_gen_dependency(value: bool = Depends(wrapped_gen_dependency)):
|
|
return value
|
|
|
|
|
|
@app.get("/async-wrapped-dependency/")
|
|
async def get_async_wrapped_dependency(value: bool = Depends(async_wrapped_dependency)):
|
|
return value
|
|
|
|
|
|
@app.get("/async-wrapped-gen-dependency/")
|
|
async def get_async_wrapped_gen_dependency(
|
|
value: bool = Depends(async_wrapped_gen_dependency),
|
|
):
|
|
return value
|
|
|
|
|
|
client = TestClient(app)
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"route",
|
|
[
|
|
"/wrapped-dependency",
|
|
"/wrapped-gen-dependency",
|
|
"/async-wrapped-dependency",
|
|
"/async-wrapped-gen-dependency",
|
|
],
|
|
)
|
|
def test_class_dependency(route):
|
|
response = client.get(route)
|
|
assert response.status_code == 200, response.text
|
|
assert response.json() is True
|