|
1 | 1 | import datetime |
2 | | -from unittest import skipIf |
3 | 2 |
|
4 | 3 | from odoo import fields |
5 | 4 | from odoo.tests import TransactionCase |
6 | 5 |
|
7 | | -from pydantic import BaseModel, Field |
| 6 | +from pydantic import Field |
8 | 7 |
|
9 | | -from ..utils import PYDANTIC_V2 |
10 | | - |
11 | | -if PYDANTIC_V2: |
12 | | - from ..utils import PydanticOdooBaseModel as PydanticOrmBaseModel |
13 | | - |
14 | | -else: |
15 | | - from ..utils import GenericOdooGetter |
16 | | - |
17 | | - class PydanticOrmBaseModel(BaseModel): |
18 | | - class Config: |
19 | | - orm_mode = True |
20 | | - getter_dict = GenericOdooGetter |
| 8 | +from ..utils import PydanticOdooBaseModel as PydanticOrmBaseModel |
21 | 9 |
|
22 | 10 |
|
23 | 11 | class OdooBaseModel(PydanticOrmBaseModel): |
@@ -60,80 +48,6 @@ def setUpClass(cls): |
60 | 48 | cls.user_demo.share = False |
61 | 49 |
|
62 | 50 |
|
63 | | -@skipIf(PYDANTIC_V2, "Ignore because Pydantic >= 2.0 is installed") |
64 | | -class TestGenericOdooGetterPydanticV1Case(CommonPydanticCase): |
65 | | - def test_user_model_serialization(self): |
66 | | - self.user_demo.partner_id.date = None |
67 | | - self.assertEqual( |
68 | | - UserModel.from_orm(self.user_demo).dict(), |
69 | | - { |
70 | | - "id": self.user_demo.id, |
71 | | - "partner": { |
72 | | - "id": self.user_demo.partner_id.id, |
73 | | - "name": self.user_demo.partner_id.name, |
74 | | - "date": None, |
75 | | - }, |
76 | | - }, |
77 | | - ) |
78 | | - |
79 | | - def test_user_model_serialization_date(self): |
80 | | - self.user_demo.partner_id.date = fields.Date.today() |
81 | | - self.assertEqual( |
82 | | - UserModel.from_orm(self.user_demo).partner.date, |
83 | | - self.user_demo.partner_id.date, |
84 | | - ) |
85 | | - |
86 | | - def test_user_model_details_serialization_datetime(self): |
87 | | - user_demo = self.user_demo.with_context(tz="Asia/Tokyo") |
88 | | - self.assertEqual( |
89 | | - UserDetailsModel.from_orm(user_demo).write_date, |
90 | | - fields.Datetime.context_timestamp(user_demo, user_demo.write_date), |
91 | | - ) |
92 | | - self.assertNotEqual( |
93 | | - UserDetailsModel.from_orm(user_demo).write_date.tzinfo, |
94 | | - fields.Datetime.context_timestamp( |
95 | | - self.user_demo, user_demo.write_date |
96 | | - ).tzinfo, |
97 | | - ) |
98 | | - |
99 | | - def test_user_details_model_serialization(self): |
100 | | - self.assertEqual( |
101 | | - UserDetailsModel.from_orm(self.user_demo).dict(), |
102 | | - { |
103 | | - "id": self.user_demo.id, |
104 | | - "partner": { |
105 | | - "id": self.user_demo.partner_id.id, |
106 | | - "name": self.user_demo.partner_id.name, |
107 | | - "date": None, |
108 | | - }, |
109 | | - "groups": [ |
110 | | - { |
111 | | - "id": group.id, |
112 | | - "name": group.name, |
113 | | - } |
114 | | - for group in self.user_demo.groups_id |
115 | | - ], |
116 | | - "action_id": None, |
117 | | - "signature": None, |
118 | | - "active": True, |
119 | | - "share": False, |
120 | | - "write_date": fields.Datetime.context_timestamp( |
121 | | - self.user_demo, self.user_demo.write_date |
122 | | - ), |
123 | | - }, |
124 | | - ) |
125 | | - |
126 | | - def test_user_flat_model_serialization(self): |
127 | | - self.assertEqual( |
128 | | - UserFlatModel.from_orm(self.user_demo).dict(), |
129 | | - { |
130 | | - "id": self.user_demo.id, |
131 | | - "partner_id": self.user_demo.partner_id.id, |
132 | | - }, |
133 | | - ) |
134 | | - |
135 | | - |
136 | | -@skipIf(not PYDANTIC_V2, "Ignore because Pydantic < 2.0 is installed") |
137 | 51 | class TestGenericOdooGetterPydanticV2Case(CommonPydanticCase): |
138 | 52 | def test_user_model_serialization(self): |
139 | 53 | self.user_demo.partner_id.date = None |
@@ -204,3 +118,18 @@ def test_user_flat_model_serialization(self): |
204 | 118 | "partner_id": self.user_demo.partner_id.id, |
205 | 119 | }, |
206 | 120 | ) |
| 121 | + |
| 122 | + def test_not_an_odoo_record(self): |
| 123 | + user = UserDetailsModel( |
| 124 | + id=666, |
| 125 | + partner_id={"id": 66, "name": "test"}, |
| 126 | + groups_id=[{"id": 33, "name": "group 1"}], |
| 127 | + action_id={"id": 55}, |
| 128 | + signature=None, |
| 129 | + active=True, |
| 130 | + share=False, |
| 131 | + write_date=fields.Datetime.now(), |
| 132 | + ) |
| 133 | + self.assertEqual( |
| 134 | + UserDetailsModel.model_validate(user).model_dump(), user.model_dump() |
| 135 | + ) |
0 commit comments