mirror of https://github.com/peter4431/rq.git
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
45 lines
1.4 KiB
Python
45 lines
1.4 KiB
Python
4 years ago
|
import time
|
||
|
|
|
||
|
|
from multiprocessing import Process
|
||
|
|
|
||
|
|
from tests import RQTestCase
|
||
|
|
from tests.fixtures import long_running_job
|
||
|
|
|
||
|
|
from rq import Queue, Worker
|
||
|
|
from rq.command import send_command, send_kill_horse_command, send_shutdown_command
|
||
|
|
|
||
|
|
|
||
|
|
class TestCommands(RQTestCase):
|
||
|
|
|
||
|
|
def test_shutdown_command(self):
|
||
|
|
"""Ensure that shutdown command works properly."""
|
||
|
|
connection = self.testconn
|
||
|
|
worker = Worker('foo', connection=connection)
|
||
|
|
|
||
|
|
def _send_shutdown_command():
|
||
|
|
time.sleep(0.25)
|
||
|
|
send_shutdown_command(connection, worker.name)
|
||
|
|
|
||
|
|
p = Process(target=_send_shutdown_command)
|
||
|
|
p.start()
|
||
|
|
worker.work()
|
||
|
|
p.join(1)
|
||
|
|
|
||
|
|
def test_kill_horse_command(self):
|
||
|
|
"""Ensure that shutdown command works properly."""
|
||
|
|
connection = self.testconn
|
||
|
|
queue = Queue('foo', connection=connection)
|
||
|
|
job = queue.enqueue(long_running_job, 4)
|
||
|
|
worker = Worker('foo', connection=connection)
|
||
|
|
|
||
|
|
def _send_kill_horse_command():
|
||
|
|
"""Waits 0.25 seconds before sending kill-horse command"""
|
||
|
|
time.sleep(0.25)
|
||
|
|
send_kill_horse_command(connection, worker.name)
|
||
|
|
|
||
|
|
p = Process(target=_send_kill_horse_command)
|
||
|
|
p.start()
|
||
|
|
worker.work(burst=True)
|
||
|
|
p.join(1)
|
||
|
|
job.refresh()
|
||
|
|
self.assertTrue(job.id in queue.failed_job_registry)
|