def test3_firstnext(self): """ QueueSet.first()/next() """ q1 = Queue(self.q1, schema={'data': 'string'}) q2 = Queue(self.q2, schema={'data': 'string'}) for i in range(10): q1.add({'data': '%i A\n' % i}) q2.add({'data': '%i A\n' % i}) qs = QueueSet([q1, q2]) e = qs.first() assert isinstance(e, tuple) assert isinstance(e[0], Queue) assert isinstance(e[1], str) while e[0]: e = qs.next()
def test2_addremove(self): """ QueueSet.add()/remove() """ q1 = Queue(self.q1, schema={'data': 'string'}) q2 = Queue(self.q2, schema={'data': 'string'}) q3 = Queue(self.q3, schema={'data': 'string'}) q4 = Queue(self.q4, schema={'data': 'string'}) for i in range(10): q1.add({'data': '%i A\n' % i}) q2.add({'data': '%i A\n' % i}) q3.add({'data': '%i A\n' % i}) q4.add({'data': '%i A\n' % i}) qs = QueueSet([q1, q2]) qs.add(q3) qs.remove(q1) qs.add(q1, q4) self.failUnlessRaises(QueueError, qs.add, ([q1]))