def test_get_unique_connection_configs(self): connection_params_1 = { 'HOST': 'localhost', 'PORT': 6379, 'DB': 0, } connection_params_2 = { 'HOST': 'localhost', 'PORT': 6379, 'DB': 1, } config = { 'default': connection_params_1, 'test': connection_params_2 } unique_configs = get_unique_connection_configs(config) self.assertEqual(len(unique_configs), 2) self.assertIn(connection_params_1, unique_configs) self.assertIn(connection_params_2, unique_configs) # self.assertEqual(get_unique_connection_configs(config), # [connection_params_1, connection_params_2]) config = { 'default': connection_params_1, 'test': connection_params_1 } # Should return one connection config since it filters out duplicates self.assertEqual(get_unique_connection_configs(config), [connection_params_1])
def test_get_unique_connection_configs(self): connection_params_1 = { 'HOST': 'localhost', 'PORT': 6379, 'DB': 0, } connection_params_2 = { 'HOST': 'localhost', 'PORT': 6379, 'DB': 1, } config = { 'default': connection_params_1, 'test': connection_params_2 } unique_configs = get_unique_connection_configs(config) self.assertEqual(len(unique_configs), 2) self.assertIn(connection_params_1, unique_configs) self.assertIn(connection_params_2, unique_configs) # self.assertEqual(get_unique_connection_configs(config), # [connection_params_1, connection_params_2]) config = { 'default': connection_params_1, 'test': connection_params_1 } # Should return one connection config since it filters out duplicates self.assertEqual(get_unique_connection_configs(config), [connection_params_1])
def test_get_unique_connection_configs(self): connection_params_1 = {"HOST": "localhost", "PORT": 6379, "DB": 0} connection_params_2 = {"HOST": "localhost", "PORT": 6379, "DB": 1} config = {"default": connection_params_1, "test": connection_params_2} self.assertEqual(get_unique_connection_configs(config), [connection_params_1, connection_params_2]) config = {"default": connection_params_1, "test": connection_params_1} # Should return one connection config since it filters out duplicates self.assertEqual(get_unique_connection_configs(config), [connection_params_1])
def test_get_unique_connection_configs_with_different_timeout(self): connection_params_1 = { 'HOST': 'localhost', 'PORT': 6379, 'DB': 0, } connection_params_2 = { 'HOST': 'localhost', 'PORT': 6379, 'DB': 1, } queue_params_a = dict(connection_params_1) queue_params_b = dict(connection_params_2) queue_params_c = dict(connection_params_2) queue_params_c["DEFAULT_TIMEOUT"] = 1 config = { 'default': queue_params_a, 'test_b': queue_params_b, 'test_c': queue_params_c, } unique_configs = get_unique_connection_configs(config) self.assertEqual(len(unique_configs), 2) self.assertIn(connection_params_1, unique_configs) self.assertIn(connection_params_2, unique_configs)
def test_get_unique_connection_configs_with_different_timeout(self): connection_params_1 = { 'HOST': 'localhost', 'PORT': 6379, 'DB': 0, } connection_params_2 = { 'HOST': 'localhost', 'PORT': 6379, 'DB': 1, } queue_params_a = dict(connection_params_1) queue_params_b = dict(connection_params_2) queue_params_c = dict(connection_params_2) queue_params_c["DEFAULT_TIMEOUT"] = 1 config = { 'default': queue_params_a, 'test_b': queue_params_b, 'test_c': queue_params_c, } unique_configs = get_unique_connection_configs(config) self.assertEqual(len(unique_configs), 2) self.assertIn(connection_params_1, unique_configs) self.assertIn(connection_params_2, unique_configs)
from .queues import get_unique_connection_configs SHOW_ADMIN_LINK = getattr(settings, "RQ_SHOW_ADMIN_LINK", False) QUEUES = getattr(settings, "RQ_QUEUES", None) if QUEUES is None: raise ImproperlyConfigured("You have to define RQ_QUEUES in settings.py") NAME = getattr(settings, "RQ_NAME", "default") BURST = getattr(settings, "RQ_BURST", False) # All queues in list format so we can get them by index, includes failed queues QUEUES_LIST = [] for key, value in QUEUES.items(): QUEUES_LIST.append({"name": key, "connection_config": value}) for config in get_unique_connection_configs(): QUEUES_LIST.append({"name": "failed", "connection_config": config}) ########NEW FILE######## __FILENAME__ = tests from django.contrib.auth.models import User from django.core.exceptions import ImproperlyConfigured from django.core.management import call_command from django.core.urlresolvers import reverse from django.test import TestCase from django.utils.unittest import skipIf from django.test.client import Client from django.test.utils import override_settings from django.conf import settings from rq import get_current_job, Queue