Use Worker instead of SimpleWorker in test

rq · Oct 29, 2024 · 7b1138b · 7b1138b
1 parent 50b83ce
commit 7b1138b
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/tests/test_callbacks.py b/tests/test_callbacks.py
@@ -5,7 +5,7 @@
 
 from rq import Queue, Worker
 from rq.job import Job, JobStatus, UNEVALUATED
-from rq.worker import SimpleWorker
+from rq.worker import SimpleWorker, Worker
 
 
 class QueueCallbackTestCase(RQTestCase):
@@ -81,15 +81,14 @@ def test_erroneous_success_callback(self):
     def test_failure_callback(self):
         """Test failure callback is executed only when job a fails"""
         queue = Queue(connection=self.testconn)
-        worker = SimpleWorker([queue], connection=self.testconn)
+        worker = Worker([queue], connection=self.testconn)
 
         job = queue.enqueue(div_by_zero, on_failure=save_exception)
 
         # Callback is executed when job is successfully executed
         worker.work(burst=True)
         self.assertEqual(job.get_status(), JobStatus.FAILED)
         job.refresh()
-        print(job.exc_info)
         self.assertIn('div_by_zero',
                       self.testconn.get('failure_callback:%s' % job.id).decode())