From fa6a52c7179ea8413d5086a3526c4c5dfcaf8e43 Mon Sep 17 00:00:00 2001 From: Tom Date: Thu, 8 Jan 2026 15:57:09 +0100 Subject: [PATCH] [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. --- test_queue_job/models/test_models.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/test_queue_job/models/test_models.py b/test_queue_job/models/test_models.py index 03e8e8a8f9..494512cb49 100644 --- a/test_queue_job/models/test_models.py +++ b/test_queue_job/models/test_models.py @@ -139,6 +139,13 @@ def _register_hook(self): ) return super()._register_hook() + def _unregister_hook(self): + """Remove the patches installed by _register_hook()""" + self._revert_method("delay_me") + self._revert_method("delay_me_options") + self._revert_method("delay_me_context_key") + return super()._unregister_hook() + def _job_store_values(self, job): value = "JUST_TESTING" if job.state == "failed":