committed by
Sebastián Ramírez
2 changed files with 71 additions and 1 deletions
@ -0,0 +1,70 @@ |
|||||
|
import pytest |
||||
|
from fastapi import Depends, FastAPI |
||||
|
from starlette.testclient import TestClient |
||||
|
|
||||
|
app = FastAPI() |
||||
|
|
||||
|
|
||||
|
class CallableDependency: |
||||
|
def __call__(self, value: str) -> str: |
||||
|
return value |
||||
|
|
||||
|
|
||||
|
class AsyncCallableDependency: |
||||
|
async def __call__(self, value: str) -> str: |
||||
|
return value |
||||
|
|
||||
|
|
||||
|
class MethodsDependency: |
||||
|
def synchronous(self, value: str) -> str: |
||||
|
return value |
||||
|
|
||||
|
async def asynchronous(self, value: str) -> str: |
||||
|
return value |
||||
|
|
||||
|
|
||||
|
callable_dependency = CallableDependency() |
||||
|
async_callable_dependency = AsyncCallableDependency() |
||||
|
methods_dependency = MethodsDependency() |
||||
|
|
||||
|
|
||||
|
@app.get("/callable-dependency") |
||||
|
async def get_callable_dependency(value: str = Depends(callable_dependency)): |
||||
|
return value |
||||
|
|
||||
|
|
||||
|
@app.get("/async-callable-dependency") |
||||
|
async def get_callable_dependency(value: str = Depends(async_callable_dependency)): |
||||
|
return value |
||||
|
|
||||
|
|
||||
|
@app.get("/synchronous-method-dependency") |
||||
|
async def get_synchronous_method_dependency( |
||||
|
value: str = Depends(methods_dependency.synchronous), |
||||
|
): |
||||
|
return value |
||||
|
|
||||
|
|
||||
|
@app.get("/asynchronous-method-dependency") |
||||
|
async def get_asynchronous_method_dependency( |
||||
|
value: str = Depends(methods_dependency.asynchronous), |
||||
|
): |
||||
|
return value |
||||
|
|
||||
|
|
||||
|
client = TestClient(app) |
||||
|
|
||||
|
|
||||
|
@pytest.mark.parametrize( |
||||
|
"route,value", |
||||
|
[ |
||||
|
("/callable-dependency", "callable-dependency"), |
||||
|
("/async-callable-dependency", "async-callable-dependency"), |
||||
|
("/synchronous-method-dependency", "synchronous-method-dependency"), |
||||
|
("/asynchronous-method-dependency", "asynchronous-method-dependency"), |
||||
|
], |
||||
|
) |
||||
|
def test_class_dependency(route, value): |
||||
|
response = client.get(route, params={"value": value}) |
||||
|
assert response.status_code == 200 |
||||
|
assert response.json() == value |
Loading…
Reference in new issue