committed by
GitHub
2 changed files with 93 additions and 6 deletions
@ -0,0 +1,80 @@ |
|||||
|
from __future__ import annotations |
||||
|
|
||||
|
from typing import TYPE_CHECKING |
||||
|
|
||||
|
import pytest |
||||
|
from fastapi import Depends, FastAPI |
||||
|
from fastapi.testclient import TestClient |
||||
|
from inline_snapshot import snapshot |
||||
|
from typing_extensions import Annotated |
||||
|
|
||||
|
if TYPE_CHECKING: # pragma: no cover |
||||
|
from collections.abc import AsyncGenerator |
||||
|
|
||||
|
|
||||
|
class DummyClient: |
||||
|
async def get_people(self) -> list: |
||||
|
return ["John Doe", "Jane Doe"] |
||||
|
|
||||
|
async def close(self) -> None: |
||||
|
pass |
||||
|
|
||||
|
|
||||
|
async def get_client() -> AsyncGenerator[DummyClient, None]: |
||||
|
client = DummyClient() |
||||
|
yield client |
||||
|
await client.close() |
||||
|
|
||||
|
|
||||
|
Client = Annotated[DummyClient, Depends(get_client)] |
||||
|
|
||||
|
|
||||
|
@pytest.fixture(name="client") |
||||
|
def client_fixture() -> TestClient: |
||||
|
app = FastAPI() |
||||
|
|
||||
|
@app.get("/") |
||||
|
async def get_people(client: Client) -> list: |
||||
|
return await client.get_people() |
||||
|
|
||||
|
client = TestClient(app) |
||||
|
return client |
||||
|
|
||||
|
|
||||
|
def test_get(client: TestClient): |
||||
|
response = client.get("/") |
||||
|
assert response.status_code == 200, response.text |
||||
|
assert response.json() == ["John Doe", "Jane Doe"] |
||||
|
|
||||
|
|
||||
|
def test_openapi_schema(client: TestClient): |
||||
|
response = client.get("/openapi.json") |
||||
|
assert response.status_code == 200, response.text |
||||
|
assert response.json() == snapshot( |
||||
|
{ |
||||
|
"openapi": "3.1.0", |
||||
|
"info": {"title": "FastAPI", "version": "0.1.0"}, |
||||
|
"paths": { |
||||
|
"/": { |
||||
|
"get": { |
||||
|
"summary": "Get People", |
||||
|
"operationId": "get_people__get", |
||||
|
"responses": { |
||||
|
"200": { |
||||
|
"description": "Successful Response", |
||||
|
"content": { |
||||
|
"application/json": { |
||||
|
"schema": { |
||||
|
"items": {}, |
||||
|
"type": "array", |
||||
|
"title": "Response Get People Get", |
||||
|
} |
||||
|
} |
||||
|
}, |
||||
|
} |
||||
|
}, |
||||
|
} |
||||
|
} |
||||
|
}, |
||||
|
} |
||||
|
) |
||||
Loading…
Reference in new issue