committed by
GitHub
1 changed files with 62 additions and 0 deletions
@ -0,0 +1,62 @@ |
|||
from typing import Union |
|||
|
|||
from fastapi import Body, FastAPI, Query |
|||
from fastapi.testclient import TestClient |
|||
|
|||
app = FastAPI() |
|||
|
|||
|
|||
@app.get("/query") |
|||
def read_query(q: Union[str, None]): |
|||
return q |
|||
|
|||
|
|||
@app.get("/explicit-query") |
|||
def read_explicit_query(q: Union[str, None] = Query()): |
|||
return q |
|||
|
|||
|
|||
@app.post("/body-embed") |
|||
def send_body_embed(b: Union[str, None] = Body(embed=True)): |
|||
return b |
|||
|
|||
|
|||
client = TestClient(app) |
|||
|
|||
|
|||
def test_required_nonable_query_invalid(): |
|||
response = client.get("/query") |
|||
assert response.status_code == 422 |
|||
|
|||
|
|||
def test_required_noneable_query_value(): |
|||
response = client.get("/query", params={"q": "foo"}) |
|||
assert response.status_code == 200 |
|||
assert response.json() == "foo" |
|||
|
|||
|
|||
def test_required_nonable_explicit_query_invalid(): |
|||
response = client.get("/explicit-query") |
|||
assert response.status_code == 422 |
|||
|
|||
|
|||
def test_required_nonable_explicit_query_value(): |
|||
response = client.get("/explicit-query", params={"q": "foo"}) |
|||
assert response.status_code == 200 |
|||
assert response.json() == "foo" |
|||
|
|||
|
|||
def test_required_nonable_body_embed_no_content(): |
|||
response = client.post("/body-embed") |
|||
assert response.status_code == 422 |
|||
|
|||
|
|||
def test_required_nonable_body_embed_invalid(): |
|||
response = client.post("/body-embed", json={"invalid": "invalid"}) |
|||
assert response.status_code == 422 |
|||
|
|||
|
|||
def test_required_noneable_body_embed_value(): |
|||
response = client.post("/body-embed", json={"b": "foo"}) |
|||
assert response.status_code == 200 |
|||
assert response.json() == "foo" |
Loading…
Reference in new issue