Skip to content

Commit 4da8114

Browse files
author
Milan Topuzov
committed
tests(test_queue_job): stop using removed _patch_method; patch in _register_hook
Adjust tests to new patching approach; no functional change.
1 parent e040198 commit 4da8114

File tree

1 file changed

+13
-7
lines changed

1 file changed

+13
-7
lines changed

test_queue_job/models/test_models.py

Lines changed: 13 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
# Copyright 2016 Camptocamp SA
22
# License LGPL-3.0 or later (http://www.gnu.org/licenses/lgpl.html)
33

4+
import functools
45
from datetime import datetime, timedelta
56

67
from odoo import api, fields, models
@@ -127,15 +128,20 @@ def delay_me_context_key(self):
127128
return "ok"
128129

129130
def _register_hook(self):
130-
self._patch_method("delay_me", self._patch_job_auto_delay("delay_me"))
131-
self._patch_method(
132-
"delay_me_options", self._patch_job_auto_delay("delay_me_options")
131+
patched = self._patch_job_auto_delay("delay_me")
132+
type(self).delay_me = functools.update_wrapper(patched, type(self).delay_me)
133+
134+
patched = self._patch_job_auto_delay("delay_me_options")
135+
type(self).delay_me_options = functools.update_wrapper(
136+
patched, type(self).delay_me_options
133137
)
134-
self._patch_method(
138+
139+
patched = self._patch_job_auto_delay(
135140
"delay_me_context_key",
136-
self._patch_job_auto_delay(
137-
"delay_me_context_key", context_key="auto_delay_delay_me_context_key"
138-
),
141+
context_key="auto_delay_delay_me_context_key",
142+
)
143+
type(self).delay_me_context_key = functools.update_wrapper(
144+
patched, type(self).delay_me_context_key
139145
)
140146
return super()._register_hook()
141147

0 commit comments

Comments
 (0)