def test_dequeue_connection_error(self, mock_redis):
queue = ReliableQueue(mock_redis)
qname = md5(self.user_id + "_queue").hexdigest()
mock_redis.rpoplpush.side_effect = ConnectionError
data = None
with self.assertRaises(ConnectionError) as ar:
data = queue.dequeue(qname, self.uuid)
self.assertIsNone(data)
self.assertIsInstance(ar.exception, ConnectionError)
mock_redis.reset_mock()
mock_redis.rpoplpush.return_value = "anan"
mock_redis.pipeline.return_value.execute.side_effect= ConnectionError
with self.assertRaises(ConnectionError) as ar:
data = queue.dequeue(qname, self.uuid)
self.assertIsNone(data)
self.assertIsInstance(ar.exception, ConnectionError)
def test_dequeue_connection_error(self, mock_redis):
queue = ReliableQueue(mock_redis)
qname = md5(self.user_id + "_queue").hexdigest()
mock_redis.rpoplpush.side_effect = ConnectionError
data = None
with self.assertRaises(ConnectionError) as ar:
data = queue.dequeue(qname, self.uuid)
self.assertIsNone(data)
self.assertIsInstance(ar.exception, ConnectionError)
mock_redis.reset_mock()
mock_redis.rpoplpush.return_value = "anan"
mock_redis.pipeline.return_value.execute.side_effect = ConnectionError
with self.assertRaises(ConnectionError) as ar:
data = queue.dequeue(qname, self.uuid)
self.assertIsNone(data)
self.assertIsInstance(ar.exception, ConnectionError)
def test_dequeue(self, mock_redis):
queue = ReliableQueue(mock_redis)
qname = md5(self.user_id + "_queue").hexdigest()
mock_redis.rpoplpush.return_value = self.uuid
mock_redis.pipeline.return_value.execute.return_value = \
self.data, pickle.dumps(Job(self.uuid))
k, data = queue.dequeue(qname, self.uuid)
self.assertEqual(k, self.uuid)
self.assertEqual(data, self.data)
def test_dequeue(self, mock_redis):
queue = ReliableQueue(mock_redis)
qname = md5(self.user_id + "_queue").hexdigest()
mock_redis.rpoplpush.return_value = self.uuid
mock_redis.pipeline.return_value.execute.return_value = \
self.data, pickle.dumps(Job(self.uuid))
k, data = queue.dequeue(qname, self.uuid)
self.assertEqual(k, self.uuid)
self.assertEqual(data, self.data)
def test_dequeue_key_not_found(self, mock_redis):
queue = ReliableQueue(mock_redis)
qname = md5(self.user_id + "_queue").hexdigest()
mock_redis.rpoplpush.return_value = "anan"
k = data = None
mock_redis.pipeline.return_value.execute.return_value = \
None, None
with self.assertRaises(ERedisKeyNotFound) as ar:
k, data = queue.dequeue(qname, self.uuid)
self.assertIsNone(k)
self.assertIsNone(data)
self.assertIsInstance(ar.exception, ERedisKeyNotFound)
def test_dequeue_data_missing(self, mock_redis):
queue = ReliableQueue(mock_redis)
qname = md5(self.user_id + "_queue").hexdigest()
mock_redis.rpoplpush.return_value = "anan"
k = data = None
mock_redis.pipeline.return_value.execute.return_value = \
None, pickle.dumps(Job(self.uuid))
with self.assertRaises(ERedisDataMissing) as ar:
k, data = queue.dequeue(qname, "anan")
self.assertIsNone(k)
self.assertIsNone(data)
self.assertIsInstance(ar.exception, ERedisDataMissing)
def test_dequeue_empty_queue(self, mock_redis):
queue = ReliableQueue(mock_redis)
qname = md5(self.user_id + "_queue").hexdigest()
mock_redis.rpoplpush.return_value = None
k = data = None
with self.assertRaises(ERedisEmptyQueue) as ar:
k, data = queue.dequeue(qname, self.uuid)
self.assertIsNone(k)
self.assertIsNone(data)
self.assertIsInstance(ar.exception, ERedisEmptyQueue)
self.assertEqual("Redis Warn: {} queue is empty!".format(qname),
str(ar.exception))
def test_dequeue_key_not_found(self, mock_redis):
queue = ReliableQueue(mock_redis)
qname = md5(self.user_id + "_queue").hexdigest()
mock_redis.rpoplpush.return_value = "anan"
k = data = None
mock_redis.pipeline.return_value.execute.return_value = \
None, None
with self.assertRaises(ERedisKeyNotFound) as ar:
k, data = queue.dequeue(qname, self.uuid)
self.assertIsNone(k)
self.assertIsNone(data)
self.assertIsInstance(ar.exception, ERedisKeyNotFound)
def test_dequeue_data_missing(self, mock_redis):
queue = ReliableQueue(mock_redis)
qname = md5(self.user_id + "_queue").hexdigest()
mock_redis.rpoplpush.return_value = "anan"
k = data = None
mock_redis.pipeline.return_value.execute.return_value = \
None, pickle.dumps(Job(self.uuid))
with self.assertRaises(ERedisDataMissing) as ar:
k, data = queue.dequeue(qname, "anan")
self.assertIsNone(k)
self.assertIsNone(data)
self.assertIsInstance(ar.exception, ERedisDataMissing)
def test_dequeue_empty_queue(self, mock_redis):
queue = ReliableQueue(mock_redis)
qname = md5(self.user_id + "_queue").hexdigest()
mock_redis.rpoplpush.return_value = None
k = data = None
with self.assertRaises(ERedisEmptyQueue) as ar:
k, data = queue.dequeue(qname, self.uuid)
self.assertIsNone(k)
self.assertIsNone(data)
self.assertIsInstance(ar.exception, ERedisEmptyQueue)
self.assertEqual("Redis Warn: {} queue is empty!".format(qname),
str(ar.exception))
