def test_get_delete_messages_one(self): qpub = QueueSimple(self.test_dir, umask=0) qpub.add('123') qsub = DirectoryQueueClient(self.ch) qsub.connect() assert qsub.get_messages() == [123] assert qpub.count() == 0
def test_get_delete_messages_max(self): num_messages_half = 5 num_messages_all = 10 qpub = QueueSimple(self.test_dir) for i in range(num_messages_all): qpub.add('{"message" : "%i"}' % i) qsub = DirectoryQueueClient(self.ch) qsub.connect() messages = qsub.get_messages(num_messages_half) assert len(messages) == num_messages_half assert qpub.count() == num_messages_half messages_read = qsub.get_messages(num_messages_half) messages_test = [{'message': str(i)} for i in range(num_messages_half, num_messages_all)] assert messages_read == messages_test assert qpub.count() == 0
def test_get_messages_empty_queue(self): QueueSimple(self.test_dir) qsub = DirectoryQueueClient(self.ch) qsub.connect() assert qsub.get_messages() == []