33
44from dateutil .relativedelta import relativedelta
55
6- from odoo import _ , fields
6+ from odoo import fields
77
88from .common import TestMaintenancePlanBase
99
@@ -17,15 +17,15 @@ def setUpClass(cls):
1717 def test_display_name (self ):
1818 self .assertEqual (
1919 self .maintenance_plan_1 .display_name ,
20- _ (
20+ self . env . _ (
2121 "Unnamed %(void)s plan (%(eqpmnt)s)" ,
2222 void = "" ,
2323 eqpmnt = self .maintenance_plan_1 .equipment_id .name ,
2424 ),
2525 )
2626 self .assertEqual (
2727 self .maintenance_plan_2 .display_name ,
28- _ (
28+ self . env . _ (
2929 "Unnamed %(kind)s plan (%(eqpmnt)s)" ,
3030 kind = self .maintenance_plan_2 .maintenance_kind_id .name ,
3131 eqpmnt = self .maintenance_plan_2 .equipment_id .name ,
@@ -46,7 +46,8 @@ def test_next_maintenance_date_01(self):
4646 )
4747
4848 def test_next_maintenance_date_02 (self ):
49- self .cron .method_direct_trigger ()
49+ with self .enter_registry_test_mode ():
50+ self .cron .method_direct_trigger ()
5051 # Check maintenance plan dates
5152 self .assertEqual (
5253 self .maintenance_plan_1 .start_maintenance_date , self .today_date
@@ -98,7 +99,8 @@ def test_next_maintenance_date_02(self):
9899 )
99100
100101 def test_generate_requests (self ):
101- self .cron .method_direct_trigger ()
102+ with self .enter_registry_test_mode ():
103+ self .cron .method_direct_trigger ()
102104 generated_requests = self .maintenance_request_obj .search (
103105 [("maintenance_plan_id" , "=" , self .maintenance_plan_1 .id )],
104106 order = "schedule_date asc" ,
@@ -120,15 +122,16 @@ def test_generate_requests(self):
120122 )
121123 self .assertEqual (
122124 generated_request .name ,
123- _ (
125+ self . env . _ (
124126 "Preventive Maintenance (%(kind)s) - %(plan)s" ,
125127 kind = self .weekly_kind .name ,
126128 plan = self .maintenance_plan_4 .name ,
127129 ),
128130 )
129131
130132 def test_generate_requests2 (self ):
131- self .cron .method_direct_trigger ()
133+ with self .enter_registry_test_mode ():
134+ self .cron .method_direct_trigger ()
132135 generated_requests = self .maintenance_request_obj .search (
133136 [("maintenance_plan_id" , "=" , self .maintenance_plan_1 .id )],
134137 order = "schedule_date asc" ,
@@ -140,7 +143,8 @@ def test_generate_requests2(self):
140143 new_date = fields .Date .from_string ("2023-04-25" )
141144 self .maintenance_plan_1 .next_maintenance_date = new_date
142145 self .maintenance_plan_1 .maintenance_plan_horizon = 3
143- self .cron .method_direct_trigger ()
146+ with self .enter_registry_test_mode ():
147+ self .cron .method_direct_trigger ()
144148 generated_requests = self .maintenance_request_obj .search (
145149 [("maintenance_plan_id" , "=" , self .maintenance_plan_1 .id )],
146150 order = "schedule_date asc" ,
@@ -149,7 +153,8 @@ def test_generate_requests2(self):
149153 self .assertEqual (generated_requests [- 1 ].request_date , new_date )
150154
151155 def test_generate_requests_no_equipment (self ):
152- self .cron .method_direct_trigger ()
156+ with self .enter_registry_test_mode ():
157+ self .cron .method_direct_trigger ()
153158 generated_requests = self .maintenance_request_obj .search (
154159 [("maintenance_plan_id" , "=" , self .maintenance_plan_5 .id )],
155160 order = "schedule_date asc" ,
@@ -168,8 +173,8 @@ def test_generate_requests_no_equipment(self):
168173 "maintenance_plan_horizon" : 3 ,
169174 }
170175 )
171-
172- self .cron .method_direct_trigger ()
176+ with self . enter_registry_test_mode ():
177+ self .cron .method_direct_trigger ()
173178
174179 generated_requests = self .maintenance_request_obj .search (
175180 [("maintenance_plan_id" , "=" , self .maintenance_plan_5 .id )],
@@ -196,22 +201,25 @@ def test_get_relativedelta(self):
196201
197202 def test_generate_requests_inactive_equipment (self ):
198203 self .equipment_1 .active = False
199- self .cron .method_direct_trigger ()
204+ with self .enter_registry_test_mode ():
205+ self .cron .method_direct_trigger ()
200206 generated_requests = self .maintenance_request_obj .search (
201207 [("maintenance_plan_id" , "=" , self .maintenance_plan_1 .id )],
202208 order = "schedule_date asc" ,
203209 )
204210 self .assertEqual (len (generated_requests ), 0 )
205211 self .equipment_1 .active = True
206- self .cron .method_direct_trigger ()
212+ with self .enter_registry_test_mode ():
213+ self .cron .method_direct_trigger ()
207214 generated_requests = self .maintenance_request_obj .search (
208215 [("maintenance_plan_id" , "=" , self .maintenance_plan_1 .id )],
209216 order = "schedule_date asc" ,
210217 )
211218 self .assertEqual (len (generated_requests ), 3 )
212219
213220 def test_maintenance_request_report (self ):
214- self .cron .method_direct_trigger ()
221+ with self .enter_registry_test_mode ():
222+ self .cron .method_direct_trigger ()
215223 generated_request = self .maintenance_request_obj .search (
216224 [("maintenance_plan_id" , "=" , self .maintenance_plan_1 .id )],
217225 order = "schedule_date asc" ,
0 commit comments