1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
|
from dataclasses import dataclass as vanilla_dataclass
from typing import List
from typing_extensions import TypedDict
from pydantic import BaseModel
from polyfactory.factories import DataclassFactory, TypedDictFactory
from polyfactory.factories.pydantic_factory import ModelFactory
class A(BaseModel):
a_text: str
class B(BaseModel):
b_text: str
a: A
class C(BaseModel):
b: B
b_list: List[B]
def test_auto_register_model_factory() -> None:
class AFactory(ModelFactory):
a_text = "const value"
__model__ = A
class BFactory(ModelFactory):
b_text = "const value"
__model__ = B
__set_as_default_factory_for_type__ = True
class CFactory(ModelFactory):
__model__ = C
c = CFactory.build()
assert c.b.b_text == BFactory.b_text
assert c.b_list[0].b_text == BFactory.b_text
assert c.b.a.a_text != AFactory.a_text
def test_auto_register_model_factory_using_create_factory() -> None:
const_value = "const value"
ModelFactory.create_factory(model=A, a_text=const_value)
ModelFactory.create_factory(model=B, b_text=const_value, __set_as_default_factory_for_type__=True)
factory = ModelFactory.create_factory(model=C)
c = factory.build()
assert c.b.b_text == const_value
assert c.b_list[0].b_text == const_value
assert c.b.a.a_text != const_value
def test_dataclass_model_factory_auto_registration() -> None:
@vanilla_dataclass
class DataClass:
text: str
class UpperModel(BaseModel):
nested_field: DataClass
nested_list_field: List[DataClass]
class UpperModelFactory(ModelFactory):
__model__ = UpperModel
class DTFactory(DataclassFactory):
text = "const value"
__model__ = DataClass
__set_as_default_factory_for_type__ = True
upper = UpperModelFactory.build()
assert upper.nested_field.text == DTFactory.text
assert upper.nested_list_field[0].text == DTFactory.text
def test_typeddict_model_factory_auto_registration() -> None:
class TD(TypedDict):
text: str
class UpperSchema(BaseModel):
nested_field: TD
nested_list_field: List[TD]
class UpperModelFactory(ModelFactory):
__model__ = UpperSchema
class TDFactory(TypedDictFactory):
text = "const value"
__model__ = TD
__set_as_default_factory_for_type__ = True
upper = UpperModelFactory.build()
assert upper.nested_field["text"] == TDFactory.text
assert upper.nested_list_field[0]["text"] == TDFactory.text
|
