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
102
103
104
105
106
|
"""Tests to check that usage of pydantic v1 and v2 at the same time works."""
from typing import Dict, List, Optional, Tuple, Type, Union
import pytest
from typing_extensions import Annotated
import pydantic
from polyfactory.factories.pydantic_factory import ModelFactory
if pydantic.VERSION.startswith("1"):
pytest.skip("only for pydantic v2", allow_module_level=True)
from pydantic import BaseModel as BaseModelV2
try:
from pydantic.v1 import BaseModel as BaseModelV1
except ImportError:
from pydantic import BaseModel as BaseModelV1 # type: ignore[assignment]
@pytest.mark.parametrize("base_model", [BaseModelV1, BaseModelV2])
def test_is_supported_type(base_model: Type[Union[BaseModelV1, BaseModelV2]]) -> None:
class Foo(base_model): # type: ignore[valid-type, misc]
...
assert ModelFactory.is_supported_type(Foo) is True
@pytest.mark.parametrize("base_model", [BaseModelV1, BaseModelV2])
def test_build(base_model: Type[Union[BaseModelV1, BaseModelV2]]) -> None:
class Foo(base_model): # type: ignore[valid-type, misc]
a: int
b: str
c: bool
FooFactory = ModelFactory.create_factory(Foo)
foo = FooFactory.build()
assert isinstance(foo.a, int)
assert isinstance(foo.b, str)
assert isinstance(foo.c, bool)
def test_build_v1_with_constrained_fields() -> None:
from pydantic.v1.fields import Field
ConstrainedInt = Annotated[int, Field(ge=100, le=200)]
ConstrainedStr = Annotated[str, Field(min_length=1, max_length=3)]
class Foo(BaseModelV1):
a: ConstrainedInt
b: ConstrainedStr
c: Union[ConstrainedInt, ConstrainedStr]
d: Optional[ConstrainedInt]
e: Optional[Union[ConstrainedInt, ConstrainedStr]]
f: List[ConstrainedInt]
g: Dict[ConstrainedInt, ConstrainedStr]
ModelFactory.create_factory(Foo).build()
def test_build_v2_with_constrained_fields() -> None:
from pydantic.fields import Field
ConstrainedInt = Annotated[int, Field(ge=100, le=200)]
ConstrainedStr = Annotated[str, Field(min_length=1, max_length=3)]
class Foo(pydantic.BaseModel): # pyright: ignore[reportGeneralTypeIssues]
a: ConstrainedInt
b: ConstrainedStr
c: Union[ConstrainedInt, ConstrainedStr]
d: Optional[ConstrainedInt]
e: Optional[Union[ConstrainedInt, ConstrainedStr]]
f: List[ConstrainedInt]
g: Dict[ConstrainedInt, ConstrainedStr]
ModelFactory.create_factory(Foo).build()
def test_variadic_tuple_length() -> None:
class Foo(pydantic.BaseModel):
bar: Tuple[int, ...]
class Factory(ModelFactory[Foo]):
__randomize_collection_length__ = True
__min_collection_length__ = 7
__max_collection_length__ = 8
result = Factory.build()
assert 7 <= len(result.bar) <= 8
def test_build_v1_with_url_and_email_types() -> None:
from pydantic.v1 import AnyHttpUrl, AnyUrl, EmailStr, HttpUrl, NameEmail
class Foo(BaseModelV1):
http_url: HttpUrl
any_http_url: AnyHttpUrl
any_url: AnyUrl
email_str: EmailStr
name_email: NameEmail
composed: Tuple[HttpUrl, AnyHttpUrl, AnyUrl, EmailStr, NameEmail]
ModelFactory.create_factory(Foo).build()
|
