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
|
from uuid import UUID
import pytest
from pydantic import BaseModel
from beanie import PydanticObjectId
from beanie.odm.utils.pydantic import IS_PYDANTIC_V2
from tests.odm.models import DocumentWithCustomIdInt, DocumentWithCustomIdUUID
class A(BaseModel):
id: PydanticObjectId
async def test_uuid_id():
doc = DocumentWithCustomIdUUID(name="TEST")
await doc.insert()
new_doc = await DocumentWithCustomIdUUID.get(doc.id)
assert isinstance(new_doc.id, UUID)
async def test_integer_id():
doc = DocumentWithCustomIdInt(name="TEST", id=1)
await doc.insert()
new_doc = await DocumentWithCustomIdInt.get(doc.id)
assert isinstance(new_doc.id, int)
@pytest.mark.skipif(
not IS_PYDANTIC_V2,
reason="supports only pydantic v2",
)
async def test_pydantic_object_id_validation_json():
deserialized = A.model_validate_json('{"id": "5eb7cf5a86d9755df3a6c593"}')
assert isinstance(deserialized.id, PydanticObjectId)
assert str(deserialized.id) == "5eb7cf5a86d9755df3a6c593"
assert deserialized.id == PydanticObjectId("5eb7cf5a86d9755df3a6c593")
@pytest.mark.skipif(
not IS_PYDANTIC_V2,
reason="supports only pydantic v2",
)
@pytest.mark.parametrize(
"data",
[
"5eb7cf5a86d9755df3a6c593",
PydanticObjectId("5eb7cf5a86d9755df3a6c593"),
],
)
async def test_pydantic_object_id_serialization(data):
deserialized = A(**{"id": data})
assert isinstance(deserialized.id, PydanticObjectId)
assert str(deserialized.id) == "5eb7cf5a86d9755df3a6c593"
assert deserialized.id == PydanticObjectId("5eb7cf5a86d9755df3a6c593")
|
