Coverage for tests/test_required_noneable.py: 100%

1from typing import Union 7 ctx1abcdefg

2 

3from fastapi import Body, FastAPI, Query 7 ctx1abcdefg

4from fastapi.testclient import TestClient 7 ctx1abcdefg

5 

6app = FastAPI() 7 ctx1abcdefg

7 

8 

9@app.get("/query") 7 ctx1abcdefg

10def read_query(q: Union[str, None]): 7 ctx1abcdefg

11 return q 7 ctx1hijklmn

12 

13 

14@app.get("/explicit-query") 7 ctx1abcdefg

15def read_explicit_query(q: Union[str, None] = Query()): 7 ctx1abcdefg

16 return q 7 ctx1opqrstu

17 

18 

19@app.post("/body-embed") 7 ctx1abcdefg

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

21 return b 7 ctx1vwxyzAB

22 

23 

24client = TestClient(app) 7 ctx1abcdefg

25 

26 

27def test_required_nonable_query_invalid(): 7 ctx1abcdefg

28 response = client.get("/query") 7 ctx1CDEFGHI

29 assert response.status_code == 422 7 ctx1CDEFGHI

30 

31 

32def test_required_noneable_query_value(): 7 ctx1abcdefg

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

34 assert response.status_code == 200 7 ctx1hijklmn

35 assert response.json() == "foo" 7 ctx1hijklmn

36 

37 

38def test_required_nonable_explicit_query_invalid(): 7 ctx1abcdefg

39 response = client.get("/explicit-query") 7 ctx1JKLMNOP

40 assert response.status_code == 422 7 ctx1JKLMNOP

41 

42 

43def test_required_nonable_explicit_query_value(): 7 ctx1abcdefg

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

45 assert response.status_code == 200 7 ctx1opqrstu

46 assert response.json() == "foo" 7 ctx1opqrstu

47 

48 

49def test_required_nonable_body_embed_no_content(): 7 ctx1abcdefg

50 response = client.post("/body-embed") 7 ctx1QRSTUVW

51 assert response.status_code == 422 7 ctx1QRSTUVW

52 

53 

54def test_required_nonable_body_embed_invalid(): 7 ctx1abcdefg

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

56 assert response.status_code == 422 7 ctx1XYZ0123

57 

58 

59def test_required_noneable_body_embed_value(): 7 ctx1abcdefg

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

61 assert response.status_code == 200 7 ctx1vwxyzAB

62 assert response.json() == "foo" 7 ctx1vwxyzAB