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
