Skip to content

Commit c25e2d1

Browse files
committed
[FIX] test_queue_job: also add an _unregister_hook function that reverts patches
Otherwise monkey-patches could be added twice when instantiating new test classes.
1 parent 227a5e7 commit c25e2d1

1 file changed

Lines changed: 7 additions & 0 deletions

File tree

test_queue_job/models/test_models.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -141,6 +141,13 @@ def _register_hook(self):
141141
)
142142
return super()._register_hook()
143143

144+
def _unregister_hook(self):
145+
"""Remove the patches installed by _register_hook()"""
146+
self._revert_method("delay_me")
147+
self._revert_method("delay_me_options")
148+
self._revert_method("delay_me_context_key")
149+
return super()._unregister_hook()
150+
144151
def _job_store_values(self, job):
145152
value = "JUST_TESTING"
146153
if job.state == "failed":

0 commit comments

Comments
 (0)