Coverage for tests/test_required_noneable.py: 100%

1from typing import Union 6 ctx1abcdef

2 

3from fastapi import Body, FastAPI, Query 6 ctx1abcdef

4from fastapi.testclient import TestClient 6 ctx1abcdef

5 

6app = FastAPI() 6 ctx1abcdef

7 

8 

9@app.get("/query") 6 ctx1abcdef

10def read_query(q: Union[str, None]): 6 ctx1abcdef

11 return q 6 ctx1ghijkl

12 

13 

14@app.get("/explicit-query") 6 ctx1abcdef

15def read_explicit_query(q: Union[str, None] = Query()): 6 ctx1abcdef

16 return q 6 ctx1mnopqr

17 

18 

19@app.post("/body-embed") 6 ctx1abcdef

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

21 return b 6 ctx1stuvwx

22 

23 

24client = TestClient(app) 6 ctx1abcdef

25 

26 

27def test_required_nonable_query_invalid(): 6 ctx1abcdef

28 response = client.get("/query") 6 ctx1yzABCD

29 assert response.status_code == 422 6 ctx1yzABCD

30 

31 

32def test_required_noneable_query_value(): 6 ctx1abcdef

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

34 assert response.status_code == 200 6 ctx1ghijkl

35 assert response.json() == "foo" 6 ctx1ghijkl

36 

37 

38def test_required_nonable_explicit_query_invalid(): 6 ctx1abcdef

39 response = client.get("/explicit-query") 6 ctx1EFGHIJ

40 assert response.status_code == 422 6 ctx1EFGHIJ

41 

42 

43def test_required_nonable_explicit_query_value(): 6 ctx1abcdef

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

45 assert response.status_code == 200 6 ctx1mnopqr

46 assert response.json() == "foo" 6 ctx1mnopqr

47 

48 

49def test_required_nonable_body_embed_no_content(): 6 ctx1abcdef

50 response = client.post("/body-embed") 6 ctx1KLMNOP

51 assert response.status_code == 422 6 ctx1KLMNOP

52 

53 

54def test_required_nonable_body_embed_invalid(): 6 ctx1abcdef

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

56 assert response.status_code == 422 6 ctx1QRSTUV

57 

58 

59def test_required_noneable_body_embed_value(): 6 ctx1abcdef

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

61 assert response.status_code == 200 6 ctx1stuvwx

62 assert response.json() == "foo" 6 ctx1stuvwx