11# Copyright 2020 NextERP Romania SRL
22# Copyright 2021 Tecnativa - Víctor Martínez
33# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
4+
45from odoo import Command
56from odoo .exceptions import ValidationError
7+ from odoo .orm .model_classes import add_to_registry
68from odoo .tests import common
79from odoo .tools .misc import mute_logger
810
9- from .fake_models import ResUsers , setup_test_model , teardown_test_model
10-
1111
1212class TestCommentTemplate (common .TransactionCase ):
1313 @classmethod
1414 def setUpClass (cls ):
1515 super ().setUpClass ()
16- setup_test_model (cls .env , ResUsers )
17- cls .user_obj = cls .env .ref ("base.model_res_users" )
18- cls .user = cls .env .ref ("base.user_demo" )
19- cls .user2 = cls .env .ref ("base.demo_user0" )
20- cls .partner_id = cls .env .ref ("base.res_partner_12" )
21- cls .partner2_id = cls .env .ref ("base.res_partner_10" )
22- cls .ResPartnerTitle = cls .env ["res.partner.title" ]
16+
17+ from .fake_models import TestResUsers
18+
19+ add_to_registry (cls .registry , TestResUsers )
20+ cls .registry ._setup_models__ (cls .env .cr , ["test.res.users" ])
21+ cls .registry .init_models (
22+ cls .env .cr ,
23+ ["test.res.users" ],
24+ {"models_to_check" : True },
25+ )
26+ cls .addClassCleanup (cls .registry .__delitem__ , "test.res.users" )
27+
28+ cls .test_user_obj = cls .env ["ir.model" ].search (
29+ [("model" , "=" , "test.res.users" )], limit = 1
30+ )
31+ cls .user = cls .env ["test.res.users" ].create (
32+ {
33+ "name" : "Test User" ,
34+ "login" : "test_user" ,
35+ 36+ }
37+ )
38+ cls .user2 = cls .env ["test.res.users" ].create (
39+ {
40+ "name" : "Test User 2" ,
41+ "login" : "test_user_2" ,
42+ 43+ }
44+ )
45+ cls .partner_id = cls .env ["res.partner" ].create ({"name" : "Test Partner" })
46+ cls .partner2_id = cls .env ["res.partner" ].create ({"name" : "Test Partner 2" })
47+ cls .ResPartnerCategory = cls .env ["res.partner.category" ]
2348 cls .main_company = cls .env .ref ("base.main_company" )
2449 cls .company = cls .env ["res.company" ].create ({"name" : "Test company" })
2550 cls .before_template_id = cls .env ["base.comment.template" ].create (
2651 {
2752 "name" : "Top template" ,
2853 "text" : "Text before lines" ,
29- "models" : cls .user_obj .model ,
54+ "models" : cls .test_user_obj .model ,
3055 "company_id" : cls .company .id ,
3156 }
3257 )
@@ -35,7 +60,7 @@ def setUpClass(cls):
3560 "name" : "Bottom template" ,
3661 "position" : "after_lines" ,
3762 "text" : "Text after lines" ,
38- "models" : cls .user_obj .model ,
63+ "models" : cls .test_user_obj .model ,
3964 "company_id" : cls .company .id ,
4065 }
4166 )
@@ -44,18 +69,13 @@ def setUpClass(cls):
4469 (4 , cls .after_template_id .id ),
4570 ]
4671
47- @classmethod
48- def tearDownClass (cls ):
49- teardown_test_model (cls .env , ResUsers )
50- return super ().tearDownClass ()
51-
5272 def test_template_model_ids (self ):
5373 self .assertIn (
54- self .user_obj .model , self .before_template_id .mapped ("model_ids.model" )
74+ self .test_user_obj .model , self .before_template_id .mapped ("model_ids.model" )
5575 )
5676 self .assertEqual (len (self .before_template_id .model_ids ), 1 )
5777 self .assertIn (
58- self .user_obj .model , self .after_template_id .mapped ("model_ids.model" )
78+ self .test_user_obj .model , self .after_template_id .mapped ("model_ids.model" )
5979 )
6080 self .assertEqual (len (self .after_template_id .model_ids ), 1 )
6181
@@ -93,7 +113,7 @@ def test_global_template(self):
93113 {
94114 "name" : "Top template" ,
95115 "text" : "Text before lines" ,
96- "models" : self .user_obj .model ,
116+ "models" : self .test_user_obj .model ,
97117 "company_id" : self .company .id ,
98118 }
99119 )
@@ -133,10 +153,9 @@ def test_partner_template_domain(self):
133153 def test_render_comment_text (self ):
134154 expected_text = f"Test comment render { self .user .name }
135155 self .before_template_id .text = "Test comment render {{object.name}}"
136- with self .with_user (self .user .login ):
137- self .assertEqual (
138- self .user .render_comment (self .before_template_id ), expected_text
139- )
156+ self .assertEqual (
157+ self .user .render_comment (self .before_template_id ), expected_text
158+ )
140159
141160 def test_render_comment_text_ (self ):
142161 ro_RO_lang = (
@@ -157,27 +176,22 @@ def test_render_comment_text_(self):
157176 po_file = export .data
158177 self .assertIsNotNone (po_file )
159178
160- partner_title = self .ResPartnerTitle .create (
161- {"name" : "Ambassador" , "shortcut" : "Amb." }
162- )
179+ partner_category = self .ResPartnerCategory .create ({"name" : "Ambassador" })
163180 # Adding translated terms
164181 ctx = dict (lang = "ro_RO" )
165- partner_title .with_context (** ctx ).write (
166- {"name" : "Ambasador" , "shortcut" : "Amb." }
167- )
168- self .user .partner_id .title = partner_title
169- self .before_template_id .text = "Test comment render {{object.title.name}}"
182+ partner_category .with_context (** ctx ).write ({"name" : "Ambasador" })
183+ self .user .partner_id .category_id = partner_category
184+ self .before_template_id .text = "Test comment render {{object.category_id.name}}"
170185
171186 expected_en_text = "Test comment render Ambassador"
172187 expected_ro_text = "Test comment render Ambasador"
173- with self .with_user (self .user .login ):
174- self .assertEqual (
175- self .user .render_comment (self .before_template_id ), expected_en_text
176- )
177- self .assertEqual (
178- self .user .with_context (** ctx ).render_comment (self .before_template_id ),
179- expected_ro_text ,
180- )
188+ self .assertEqual (
189+ self .user .render_comment (self .before_template_id ), expected_en_text
190+ )
191+ self .assertEqual (
192+ self .user .with_context (** ctx ).render_comment (self .before_template_id ),
193+ expected_ro_text ,
194+ )
181195
182196 def test_partner_template_wizaard (self ):
183197 partner_preview = (
0 commit comments