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