class TestQueue(unittest.TestCase):
def setUp(self):
connection = SQSConnStub()
self.queue = Queue(connection, 'pysqes_test')
def test_enqueue_task(self):
"""
Here we're actually testing both enqueue and dequeue methods
from the queue.
"""
task = Task(add_func, [1, 2])
self.queue.enqueue_task(task)
messages = self.queue.dequeue()
self.assertEqual(messages[0][1]._args, [1, 2])
task2 = Task(data={
"key": 3
})
self.queue.enqueue_task(task2)
messages2 = self.queue.dequeue()
self.assertEqual(messages2[0][1].data, {"key": 3})
def test_enqueue(self):
self.queue.enqueue(add_func, 1, 2)
messages = self.queue.dequeue()
self.assertEqual(messages[0][1]._args, [1, 2])
def test_get_queue(self):
"""Test getting queue name from dictionary"""
connection = SQSConnStub()
queue = Queue(connection, 'pysqes_test', {'pysqes_test': 'test'})
queue_stub = queue.queue
self.assertEqual(queue_stub.name, 'test')
def test_not_queue_lookup(self):
connection = SQSConnStub()
queue = Queue(connection, False)
queue_stub = queue.queue
self.assertIsInstance(queue_stub, SQSQueueStub)
def test_delete_messages(self):
messages = self.queue.delete_message_batch(['message'])
self.assertEqual(messages, ['message'])
def test_delete_message(self):
messages = self.queue.delete_message('message')
self.assertEqual(messages, 'message')
def test_enqueue_validation(self):
def fn():
pass
fn.__module__ = '__main__'
with self.assertRaises(ValueError):
self.queue.enqueue(fn)
class TestQueue(unittest.TestCase):
def setUp(self):
connection = SQSConnStub()
self.queue = Queue(connection, 'pysqes_test')
def test_enqueue_task(self):
"""
Here we're actually testing both enqueue and dequeue methods
from the queue.
"""
task = Task(add_func, [1, 2])
self.queue.enqueue_task(task)
messages = self.queue.dequeue()
self.assertEqual(messages[0][1]._args, [1, 2])
task2 = Task(data={"key": 3})
self.queue.enqueue_task(task2)
messages2 = self.queue.dequeue()
self.assertEqual(messages2[0][1].data, {"key": 3})
def test_enqueue(self):
self.queue.enqueue(add_func, 1, 2)
messages = self.queue.dequeue()
self.assertEqual(messages[0][1]._args, [1, 2])
def test_get_queue(self):
"""Test getting queue name from dictionary"""
connection = SQSConnStub()
queue = Queue(connection, 'pysqes_test', {'pysqes_test': 'test'})
queue_stub = queue.queue
self.assertEqual(queue_stub.name, 'test')
def test_not_queue_lookup(self):
connection = SQSConnStub()
queue = Queue(connection, False)
queue_stub = queue.queue
self.assertIsInstance(queue_stub, SQSQueueStub)
def test_delete_messages(self):
messages = self.queue.delete_message_batch(['message'])
self.assertEqual(messages, ['message'])
def test_delete_message(self):
messages = self.queue.delete_message('message')
self.assertEqual(messages, 'message')
def test_enqueue_validation(self):
def fn():
pass
fn.__module__ = '__main__'
with self.assertRaises(ValueError):
self.queue.enqueue(fn)
