def test_count(self):
q = SQSQueueMock('test')
m = Message()
m.set_body('this is a test')
status = q.write(m)
self.assertTrue(status)
self.assertTrue(q.count() == 1)
status = q.write(m)
self.assertTrue(status)
self.assertTrue(q.count() == 2)
status = q.write(m)
self.assertTrue(status)
self.assertTrue(q.count() == 3)
messages = q.get_messages(num_messages=1)
self.assertTrue(q.delete_message(messages[0]))
self.assertTrue(q.count() == 2)
messages = q.get_messages(num_messages=1)
self.assertTrue(q.delete_message(messages[0]))
self.assertTrue(q.count() == 1)
messages = q.get_messages(num_messages=1)
self.assertTrue(q.delete_message(messages[0]))
self.assertTrue(q.count() == 0)
def test_read(self):
m = Message()
m.set_body('this is a test')
q = SQSQueueMock('test')
status = q.write(m)
self.assertTrue(status)
messages = q.get_messages(num_messages=2)
self.assertTrue(len(messages) == 1)
self.assertTrue(messages[0].get_body() == 'this is a test')
def test_clear(self):
q = SQSQueueMock('test')
m = Message()
m.set_body('this is a test')
status = q.write(m)
self.assertTrue(q.count() == 1)
status = q.write(m)
status = q.write(m)
status = q.write(m)
self.assertTrue(q.count() == 4)
self.assertTrue(q.clear())
self.assertTrue(q.count() == 0)
def test_write(self):
m = Message()
m.set_body('this is a test')
q = SQSQueueMock('test')
status = q.write(m)
self.assertTrue(status)
def test_delete(self):
q = SQSQueueMock('test')
self.assertTrue(q.delete())
self.assertFalse(os.path.exists('test.sqs'))
def test_create(self):
q = SQSQueueMock('test')
self.assertTrue(isinstance(q, SQSQueueMock))
self.assertTrue(q.name == 'test')
