Coverage for tests/test_validate_response_recursive/app.py: 100%

1from typing import List 7 ctx1abcdefg

2 

3from fastapi import FastAPI 7 ctx1abcdefg

4from fastapi._compat import PYDANTIC_V2 7 ctx1abcdefg

5from pydantic import BaseModel 7 ctx1abcdefg

6 

7app = FastAPI() 7 ctx1abcdefg

8 

9 

10class RecursiveItem(BaseModel): 7 ctx1abcdefg

11 sub_items: List["RecursiveItem"] = [] 7 ctx1abcdefg

12 name: str 7 ctx1abcdefg

13 

14 

15class RecursiveSubitemInSubmodel(BaseModel): 7 ctx1abcdefg

16 sub_items2: List["RecursiveItemViaSubmodel"] = [] 7 ctx1abcdefg

17 name: str 7 ctx1abcdefg

18 

19 

20class RecursiveItemViaSubmodel(BaseModel): 7 ctx1abcdefg

21 sub_items1: List[RecursiveSubitemInSubmodel] = [] 7 ctx1abcdefg

22 name: str 7 ctx1abcdefg

23 

24 

25if PYDANTIC_V2: 7 ctx1abcdefg

26 RecursiveItem.model_rebuild() 7 ctx1abcdefg

27 RecursiveSubitemInSubmodel.model_rebuild() 7 ctx1abcdefg

28 RecursiveItemViaSubmodel.model_rebuild() 7 ctx1abcdefg

29else: 

30 RecursiveItem.update_forward_refs() 6 ctx1abcdef

31 RecursiveSubitemInSubmodel.update_forward_refs() 6 ctx1abcdef

32 

33 

34@app.get("/items/recursive", response_model=RecursiveItem) 7 ctx1abcdefg

35def get_recursive(): 7 ctx1abcdefg

36 return {"name": "item", "sub_items": [{"name": "subitem", "sub_items": []}]} 7 ctx1hijklmn

37 

38 

39@app.get("/items/recursive-submodel", response_model=RecursiveItemViaSubmodel) 7 ctx1abcdefg

40def get_recursive_submodel(): 7 ctx1abcdefg

41 return { 7 ctx1hijklmn

42 "name": "item", 

43 "sub_items1": [ 

44 { 

45 "name": "subitem", 

46 "sub_items2": [ 

47 { 

48 "name": "subsubitem", 

49 "sub_items1": [{"name": "subsubsubitem", "sub_items2": []}], 

50 } 

51 ], 

52 } 

53 ], 

54 }