Coverage for tests/test_filter_pydantic_sub_model/app_pv1.py: 100%

1from typing import Optional 5 ctx1abcde

2 

3from fastapi import Depends, FastAPI 5 ctx1abcde

4from pydantic import BaseModel, validator 5 ctx1abcde

5 

6app = FastAPI() 5 ctx1abcde

7 

8 

9class ModelB(BaseModel): 5 ctx1abcde

10 username: str 5 ctx1abcde

11 

12 

13class ModelC(ModelB): 5 ctx1abcde

14 password: str 5 ctx1abcde

15 

16 

17class ModelA(BaseModel): 5 ctx1abcde

18 name: str 5 ctx1abcde

19 description: Optional[str] = None 5 ctx1abcde

20 model_b: ModelB 5 ctx1abcde

21 

22 @validator("name") 5 ctx1abcde

23 def lower_username(cls, name: str, values): 5 ctx1abcde

24 if not name.endswith("A"): 5 ctx1abcde

25 raise ValueError("name must end in A") 5 ctx1abcde

26 return name 5 ctx1abcde

27 

28 

29async def get_model_c() -> ModelC: 5 ctx1abcde

30 return ModelC(username="test-user", password="test-password") 5 ctx1abcde

31 

32 

33@app.get("/model/{name}", response_model=ModelA) 5 ctx1abcde

34async def get_model_a(name: str, model_c=Depends(get_model_c)): 5 ctx1abcde

35 return {"name": name, "description": "model-a-desc", "model_b": model_c} 5 ctx1abcde