Coverage for tests / test_inherited_custom_class.py: 100%

1import uuid 4 ctx1defg

2 

3import pytest 4 ctx1defg

4from fastapi import FastAPI 4 ctx1defg

5from fastapi.testclient import TestClient 4 ctx1defg

6from pydantic import BaseModel 4 ctx1defg

7 

8 

9class MyUuid: 4 ctx1defg

10 def __init__(self, uuid_string: str): 4 ctx1defg

11 self.uuid = uuid_string 3 ctx1abc

12 

13 def __str__(self): 4 ctx1defg

14 return self.uuid 3 ctx1abc

15 

16 @property # type: ignore 4 ctx1defg

17 def __class__(self): 4 ctx1defg

18 return uuid.UUID 3 ctx1abc

19 

20 @property 4 ctx1defg

21 def __dict__(self): 4 ctx1defg

22 """Spoof a missing __dict__ by raising TypeError, this is how 

23 asyncpg.pgroto.pgproto.UUID behaves""" 

24 raise TypeError("vars() argument must have __dict__ attribute") 3 ctx1abc

25 

26 

27def test_pydanticv2(): 4 ctx1defg

28 from pydantic import field_serializer 3 ctx1abc

29 

30 app = FastAPI() 3 ctx1abc

31 

32 @app.get("/fast_uuid") 3 ctx1abc

33 def return_fast_uuid(): 3 ctx1abc

34 asyncpg_uuid = MyUuid("a10ff360-3b1e-4984-a26f-d3ab460bdb51") 3 ctx1abc

35 assert isinstance(asyncpg_uuid, uuid.UUID) 3 ctx1abc

36 assert type(asyncpg_uuid) is not uuid.UUID 3 ctx1abc

37 with pytest.raises(TypeError): 3 ctx1abc

38 vars(asyncpg_uuid) 3 ctx1abc

39 return {"fast_uuid": asyncpg_uuid} 3 ctx1abc

40 

41 class SomeCustomClass(BaseModel): 3 ctx1abc

42 model_config = {"arbitrary_types_allowed": True} 3 ctx1abc

43 

44 a_uuid: MyUuid 3 ctx1abc

45 

46 @field_serializer("a_uuid") 3 ctx1abc

47 def serialize_a_uuid(self, v): 3 ctx1abc

48 return str(v) 3 ctx1abc

49 

50 @app.get("/get_custom_class") 3 ctx1abc

51 def return_some_user(): 3 ctx1abc

52 # Test that the fix also works for custom pydantic classes 

53 return SomeCustomClass(a_uuid=MyUuid("b8799909-f914-42de-91bc-95c819218d01")) 3 ctx1abc

54 

55 client = TestClient(app) 3 ctx1abc

56 

57 with client: 3 ctx1abc

58 response_simple = client.get("/fast_uuid") 3 ctx1abc

59 response_pydantic = client.get("/get_custom_class") 3 ctx1abc

60 

61 assert response_simple.json() == { 3 ctx1abc

62 "fast_uuid": "a10ff360-3b1e-4984-a26f-d3ab460bdb51" 

63 } 

64 

65 assert response_pydantic.json() == { 3 ctx1abc

66 "a_uuid": "b8799909-f914-42de-91bc-95c819218d01" 

67 }