From acab64b3c355a2aeb13c7739a639f9fe2a03bc1d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Sebasti=C3=A1n=20Ram=C3=ADrez?= <tiangolo@gmail.com>
Date: Sat, 14 May 2022 14:08:31 -0500
Subject: [PATCH] =?UTF-8?q?=E2=9C=85=20Add=20tests=20for=20required=20nona?=
 =?UTF-8?q?ble=20parameters=20and=20body=20fields=20(#4907)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 tests/test_required_noneable.py | 62 +++++++++++++++++++++++++++++++++
 1 file changed, 62 insertions(+)
 create mode 100644 tests/test_required_noneable.py

diff --git a/tests/test_required_noneable.py b/tests/test_required_noneable.py
new file mode 100644
index 000000000..5da8cd4d0
--- /dev/null
+++ b/tests/test_required_noneable.py
@@ -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"