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.
226 lines
8.1 KiB
Python
226 lines
8.1 KiB
Python
# -*- coding: utf-8 -*-
|
|
from __future__ import (absolute_import, division, print_function,
|
|
unicode_literals)
|
|
|
|
from click.testing import CliRunner
|
|
from rq import get_failed_queue, Queue
|
|
from rq.job import Job
|
|
from rq.cli import main
|
|
from rq.cli.helpers import read_config_file, CliConfig
|
|
import pytest
|
|
|
|
from tests import RQTestCase
|
|
from tests.fixtures import div_by_zero
|
|
|
|
try:
|
|
from unittest import TestCase
|
|
except ImportError:
|
|
from unittest2 import TestCase # noqa
|
|
|
|
|
|
class TestRQCli(RQTestCase):
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def set_tmpdir(self, tmpdir):
|
|
self.tmpdir = tmpdir
|
|
|
|
def assert_normal_execution(self, result):
|
|
if result.exit_code == 0:
|
|
return True
|
|
else:
|
|
print("Non normal execution")
|
|
print("Exit Code: {}".format(result.exit_code))
|
|
print("Output: {}".format(result.output))
|
|
print("Exception: {}".format(result.exception))
|
|
self.assertEqual(result.exit_code, 0)
|
|
|
|
"""Test rq_cli script"""
|
|
def setUp(self):
|
|
super(TestRQCli, self).setUp()
|
|
db_num = self.testconn.connection_pool.connection_kwargs['db']
|
|
self.redis_url = 'redis://127.0.0.1:6379/%d' % db_num
|
|
|
|
job = Job.create(func=div_by_zero, args=(1, 2, 3))
|
|
job.origin = 'fake'
|
|
job.save()
|
|
get_failed_queue().quarantine(job, Exception('Some fake error')) # noqa
|
|
|
|
def test_config_file(self):
|
|
settings = read_config_file('tests.dummy_settings')
|
|
self.assertIn('REDIS_HOST', settings)
|
|
self.assertEqual(settings['REDIS_HOST'], 'testhost.example.com')
|
|
|
|
def test_config_file_option(self):
|
|
""""""
|
|
cli_config = CliConfig(config='tests.dummy_settings')
|
|
self.assertEqual(
|
|
cli_config.connection.connection_pool.connection_kwargs['host'],
|
|
'testhost.example.com',
|
|
)
|
|
runner = CliRunner()
|
|
result = runner.invoke(main, ['info', '--config', cli_config.config])
|
|
self.assertEqual(result.exit_code, 1)
|
|
|
|
def test_config_file_default_options(self):
|
|
""""""
|
|
cli_config = CliConfig(config='tests.dummy_settings')
|
|
|
|
self.assertEqual(
|
|
cli_config.connection.connection_pool.connection_kwargs['host'],
|
|
'testhost.example.com',
|
|
)
|
|
self.assertEqual(
|
|
cli_config.connection.connection_pool.connection_kwargs['port'],
|
|
6379
|
|
)
|
|
self.assertEqual(
|
|
cli_config.connection.connection_pool.connection_kwargs['db'],
|
|
0
|
|
)
|
|
self.assertEqual(
|
|
cli_config.connection.connection_pool.connection_kwargs['password'],
|
|
None
|
|
)
|
|
|
|
def test_config_file_default_options_override(self):
|
|
""""""
|
|
cli_config = CliConfig(config='tests.dummy_settings_override')
|
|
|
|
self.assertEqual(
|
|
cli_config.connection.connection_pool.connection_kwargs['host'],
|
|
'testhost.example.com',
|
|
)
|
|
self.assertEqual(
|
|
cli_config.connection.connection_pool.connection_kwargs['port'],
|
|
6378
|
|
)
|
|
self.assertEqual(
|
|
cli_config.connection.connection_pool.connection_kwargs['db'],
|
|
2
|
|
)
|
|
self.assertEqual(
|
|
cli_config.connection.connection_pool.connection_kwargs['password'],
|
|
'123'
|
|
)
|
|
|
|
def test_empty_nothing(self):
|
|
"""rq empty -u <url>"""
|
|
runner = CliRunner()
|
|
result = runner.invoke(main, ['empty', '-u', self.redis_url])
|
|
self.assert_normal_execution(result)
|
|
self.assertEqual(result.output.strip(), 'Nothing to do')
|
|
|
|
def test_empty_failed(self):
|
|
"""rq empty -u <url> failed"""
|
|
runner = CliRunner()
|
|
result = runner.invoke(main, ['empty', '-u', self.redis_url, 'failed'])
|
|
self.assert_normal_execution(result)
|
|
self.assertEqual(result.output.strip(), '1 jobs removed from failed queue')
|
|
|
|
def test_empty_all(self):
|
|
"""rq empty -u <url> failed --all"""
|
|
runner = CliRunner()
|
|
result = runner.invoke(main, ['empty', '-u', self.redis_url, '--all'])
|
|
self.assert_normal_execution(result)
|
|
self.assertEqual(result.output.strip(), '1 jobs removed from failed queue')
|
|
|
|
def test_requeue(self):
|
|
"""rq requeue -u <url> --all"""
|
|
runner = CliRunner()
|
|
result = runner.invoke(main, ['requeue', '-u', self.redis_url, '--all'])
|
|
self.assert_normal_execution(result)
|
|
self.assertEqual(result.output.strip(), 'Requeueing 1 jobs from failed queue')
|
|
|
|
result = runner.invoke(main, ['requeue', '-u', self.redis_url, '--all'])
|
|
self.assert_normal_execution(result)
|
|
self.assertEqual(result.output.strip(), 'Nothing to do')
|
|
|
|
def test_info(self):
|
|
"""rq info -u <url>"""
|
|
runner = CliRunner()
|
|
result = runner.invoke(main, ['info', '-u', self.redis_url])
|
|
self.assert_normal_execution(result)
|
|
self.assertIn('1 queues, 1 jobs total', result.output)
|
|
|
|
def test_info_only_queues(self):
|
|
"""rq info -u <url> --only-queues (-Q)"""
|
|
runner = CliRunner()
|
|
result = runner.invoke(main, ['info', '-u', self.redis_url, '--only-queues'])
|
|
self.assert_normal_execution(result)
|
|
self.assertIn('1 queues, 1 jobs total', result.output)
|
|
|
|
def test_info_only_workers(self):
|
|
"""rq info -u <url> --only-workers (-W)"""
|
|
runner = CliRunner()
|
|
result = runner.invoke(main, ['info', '-u', self.redis_url, '--only-workers'])
|
|
self.assert_normal_execution(result)
|
|
self.assertIn('0 workers, 1 queues', result.output)
|
|
|
|
def test_worker(self):
|
|
"""rq worker -u <url> -b"""
|
|
runner = CliRunner()
|
|
result = runner.invoke(main, ['worker', '-u', self.redis_url, '-b'])
|
|
self.assert_normal_execution(result)
|
|
|
|
def test_worker_pid(self):
|
|
"""rq worker -u <url> /tmp/.."""
|
|
pid = self.tmpdir.join('rq.pid')
|
|
runner = CliRunner()
|
|
result = runner.invoke(main, ['worker', '-u', self.redis_url, '-b', '--pid', str(pid)])
|
|
self.assertTrue(len(pid.read()) > 0)
|
|
self.assert_normal_execution(result)
|
|
|
|
def test_exception_handlers(self):
|
|
"""rq worker -u <url> -b --exception-handler <handler>"""
|
|
q = Queue()
|
|
failed_q = get_failed_queue()
|
|
failed_q.empty()
|
|
|
|
runner = CliRunner()
|
|
|
|
# If exception handler is not given, failed job goes to FailedQueue
|
|
q.enqueue(div_by_zero)
|
|
runner.invoke(main, ['worker', '-u', self.redis_url, '-b'])
|
|
self.assertEqual(failed_q.count, 1)
|
|
|
|
# Black hole exception handler doesn't add failed jobs to FailedQueue
|
|
q.enqueue(div_by_zero)
|
|
runner.invoke(main, ['worker', '-u', self.redis_url, '-b',
|
|
'--exception-handler', 'tests.fixtures.black_hole'])
|
|
self.assertEqual(failed_q.count, 1)
|
|
|
|
def test_suspend_and_resume(self):
|
|
"""rq suspend -u <url>
|
|
rq worker -u <url> -b
|
|
rq resume -u <url>
|
|
"""
|
|
runner = CliRunner()
|
|
result = runner.invoke(main, ['suspend', '-u', self.redis_url])
|
|
self.assert_normal_execution(result)
|
|
|
|
result = runner.invoke(main, ['worker', '-u', self.redis_url, '-b'])
|
|
self.assertEqual(result.exit_code, 1)
|
|
self.assertEqual(
|
|
result.output.strip(),
|
|
'RQ is currently suspended, to resume job execution run "rq resume"'
|
|
)
|
|
|
|
result = runner.invoke(main, ['resume', '-u', self.redis_url])
|
|
self.assert_normal_execution(result)
|
|
|
|
def test_suspend_with_ttl(self):
|
|
"""rq suspend -u <url> --duration=2
|
|
"""
|
|
runner = CliRunner()
|
|
result = runner.invoke(main, ['suspend', '-u', self.redis_url, '--duration', 1])
|
|
self.assert_normal_execution(result)
|
|
|
|
def test_suspend_with_invalid_ttl(self):
|
|
"""rq suspend -u <url> --duration=0
|
|
"""
|
|
runner = CliRunner()
|
|
result = runner.invoke(main, ['suspend', '-u', self.redis_url, '--duration', 0])
|
|
|
|
self.assertEqual(result.exit_code, 1)
|
|
self.assertIn("Duration must be an integer greater than 1", result.output)
|