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_delete_message(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') self.assertTrue(q.delete_message(messages[0])) messages = q.get_messages(num_messages=1) self.assertTrue(len(messages) == 0)