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

1from typing import List 6 ctx1abcdef

2 

3from fastapi import FastAPI 6 ctx1abcdef

4from fastapi._compat import PYDANTIC_V2 6 ctx1abcdef

5from pydantic import BaseModel 6 ctx1abcdef

6 

7app = FastAPI() 6 ctx1abcdef

8 

9 

10class RecursiveItem(BaseModel): 6 ctx1abcdef

11 sub_items: List["RecursiveItem"] = [] 6 ctx1abcdef

12 name: str 6 ctx1abcdef

13 

14 

15class RecursiveSubitemInSubmodel(BaseModel): 6 ctx1abcdef

16 sub_items2: List["RecursiveItemViaSubmodel"] = [] 6 ctx1abcdef

17 name: str 6 ctx1abcdef

18 

19 

20class RecursiveItemViaSubmodel(BaseModel): 6 ctx1abcdef

21 sub_items1: List[RecursiveSubitemInSubmodel] = [] 6 ctx1abcdef

22 name: str 6 ctx1abcdef

23 

24 

25if PYDANTIC_V2: 6 ctx1abcdef

26 RecursiveItem.model_rebuild() 6 ctx1abcdef

27 RecursiveSubitemInSubmodel.model_rebuild() 6 ctx1abcdef

28 RecursiveItemViaSubmodel.model_rebuild() 6 ctx1abcdef

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) 6 ctx1abcdef

35def get_recursive(): 6 ctx1abcdef

36 return {"name": "item", "sub_items": [{"name": "subitem", "sub_items": []}]} 6 ctx1ghijkl

37 

38 

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

40def get_recursive_submodel(): 6 ctx1abcdef

41 return { 6 ctx1ghijkl

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 }