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
|
import pytest
from pydantic import BaseModel, Field
from beanie import Link, PydanticObjectId
from beanie.odm.utils.pydantic import IS_PYDANTIC_V2
from tests.odm.models import DocumentTestModelWithSoftDelete
class TestModel(BaseModel):
my_id: PydanticObjectId = Field(default_factory=PydanticObjectId)
fake_doc: Link[DocumentTestModelWithSoftDelete]
def data_maker():
return TestModel(
my_id="5f4e3f3b7c0c9d001f7d4c8e",
fake_doc=DocumentTestModelWithSoftDelete(
test_int=1, test_str="test", id="5f4e3f3b7c0c9d001f7d4c8f"
),
)
@pytest.mark.skipif(
not IS_PYDANTIC_V2,
reason="model dumping support is more complete with pydantic v2",
)
def test_id_types_preserved_when_dumping_to_python():
dumped = data_maker().model_dump(mode="python")
assert isinstance(dumped["my_id"], PydanticObjectId)
assert isinstance(dumped["fake_doc"]["id"], PydanticObjectId)
@pytest.mark.skipif(
not IS_PYDANTIC_V2,
reason="model dumping support is more complete with pydantic v2",
)
def test_id_types_serialized_when_dumping_to_json():
dumped = data_maker().model_dump(mode="json")
assert isinstance(dumped["my_id"], str)
assert isinstance(dumped["fake_doc"]["id"], str)
|
