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_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)