Coverage for tests/test_required_noneable.py: 100%

1from typing import Union 5 ctx1abcde

2 

3from fastapi import Body, FastAPI, Query 5 ctx1abcde

4from fastapi.testclient import TestClient 5 ctx1abcde

5 

6app = FastAPI() 5 ctx1abcde

7 

8 

9@app.get("/query") 5 ctx1abcde

10def read_query(q: Union[str, None]): 5 ctx1abcde

11 return q 5 ctx1abcde

12 

13 

14@app.get("/explicit-query") 5 ctx1abcde

15def read_explicit_query(q: Union[str, None] = Query()): 5 ctx1abcde

16 return q 5 ctx1abcde

17 

18 

19@app.post("/body-embed") 5 ctx1abcde

20def send_body_embed(b: Union[str, None] = Body(embed=True)): 5 ctx1abcde

21 return b 5 ctx1abcde

22 

23 

24client = TestClient(app) 5 ctx1abcde

25 

26 

27def test_required_nonable_query_invalid(): 5 ctx1abcde

28 response = client.get("/query") 5 ctx1abcde

29 assert response.status_code == 422 5 ctx1abcde

30 

31 

32def test_required_noneable_query_value(): 5 ctx1abcde

33 response = client.get("/query", params={"q": "foo"}) 5 ctx1abcde

34 assert response.status_code == 200 5 ctx1abcde

35 assert response.json() == "foo" 5 ctx1abcde

36 

37 

38def test_required_nonable_explicit_query_invalid(): 5 ctx1abcde

39 response = client.get("/explicit-query") 5 ctx1abcde

40 assert response.status_code == 422 5 ctx1abcde

41 

42 

43def test_required_nonable_explicit_query_value(): 5 ctx1abcde

44 response = client.get("/explicit-query", params={"q": "foo"}) 5 ctx1abcde

45 assert response.status_code == 200 5 ctx1abcde

46 assert response.json() == "foo" 5 ctx1abcde

47 

48 

49def test_required_nonable_body_embed_no_content(): 5 ctx1abcde

50 response = client.post("/body-embed") 5 ctx1abcde

51 assert response.status_code == 422 5 ctx1abcde

52 

53 

54def test_required_nonable_body_embed_invalid(): 5 ctx1abcde

55 response = client.post("/body-embed", json={"invalid": "invalid"}) 5 ctx1abcde

56 assert response.status_code == 422 5 ctx1abcde

57 

58 

59def test_required_noneable_body_embed_value(): 5 ctx1abcde

60 response = client.post("/body-embed", json={"b": "foo"}) 5 ctx1abcde

61 assert response.status_code == 200 5 ctx1abcde

62 assert response.json() == "foo" 5 ctx1abcde