Coverage for tests / test_json_type.py: 100%

1import json 4 ctx1abcd

2from typing import Annotated 4 ctx1abcd

3 

4from fastapi import Cookie, FastAPI, Form, Header, Query 4 ctx1abcd

5from fastapi.testclient import TestClient 4 ctx1abcd

6from pydantic import Json 4 ctx1abcd

7 

8app = FastAPI() 4 ctx1abcd

9 

10 

11@app.post("/form-json-list") 4 ctx1abcd

12def form_json_list(items: Annotated[Json[list[str]], Form()]) -> list[str]: 4 ctx1abcd

13 return items 4 ctx1ijkl

14 

15 

16@app.get("/query-json-list") 4 ctx1abcd

17def query_json_list(items: Annotated[Json[list[str]], Query()]) -> list[str]: 4 ctx1abcd

18 return items 4 ctx1mnop

19 

20 

21@app.get("/header-json-list") 4 ctx1abcd

22def header_json_list(x_items: Annotated[Json[list[str]], Header()]) -> list[str]: 4 ctx1abcd

23 return x_items 4 ctx1qrst

24 

25 

26@app.get("/cookie-json-list") 4 ctx1abcd

27def cookie_json_list(items: Annotated[Json[list[str]], Cookie()]) -> list[str]: 4 ctx1abcd

28 return items 4 ctx1efgh

29 

30 

31client = TestClient(app) 4 ctx1abcd

32 

33 

34def test_form_json_list(): 4 ctx1abcd

35 response = client.post( 4 ctx1ijkl

36 "/form-json-list", data={"items": json.dumps(["abc", "def"])} 

37 ) 

38 assert response.status_code == 200, response.text 4 ctx1ijkl

39 assert response.json() == ["abc", "def"] 4 ctx1ijkl

40 

41 

42def test_query_json_list(): 4 ctx1abcd

43 response = client.get( 4 ctx1mnop

44 "/query-json-list", params={"items": json.dumps(["abc", "def"])} 

45 ) 

46 assert response.status_code == 200, response.text 4 ctx1mnop

47 assert response.json() == ["abc", "def"] 4 ctx1mnop

48 

49 

50def test_header_json_list(): 4 ctx1abcd

51 response = client.get( 4 ctx1qrst

52 "/header-json-list", headers={"x-items": json.dumps(["abc", "def"])} 

53 ) 

54 assert response.status_code == 200, response.text 4 ctx1qrst

55 assert response.json() == ["abc", "def"] 4 ctx1qrst

56 

57 

58def test_cookie_json_list(): 4 ctx1abcd

59 client.cookies.set("items", json.dumps(["abc", "def"])) 4 ctx1efgh

60 response = client.get("/cookie-json-list") 4 ctx1efgh

61 assert response.status_code == 200, response.text 4 ctx1efgh

62 assert response.json() == ["abc", "def"] 4 ctx1efgh

63 client.cookies.clear() 4 ctx1efgh