def test_many_polls(self):
qs = QueueSet()
self.fill_queue(qs, 10000)
for i in xrange(0, 9000):
qs.poll()
assert len(qs) == 1000
assert qs.front() == 9000
def test_enqueue_multiple_replace_one(self):
qs = QueueSet()
qs.enqueue('a')
qs.enqueue('b')
qs.enqueue('c')
qs.enqueue('d')
qs.enqueue('b')
assert len(qs) == 4
assert qs.front() == 'a'
qs.poll()
k = qs.poll()
assert k == 'b'
assert len(qs) == 2
def test_enqueue_two(self):
qs = QueueSet()
qs.enqueue('a')
qs.enqueue('b')
assert len(qs) == 2
assert qs.front() == 'a'
k = qs.poll()
assert k == 'a'
assert len(qs) == 1
assert qs.front() == 'b'
k = qs.poll()
assert k == 'b'
assert len(qs) == 0
assert qs.front() is None
def test_one_poll(self):
qs = QueueSet(['a'])
assert len(qs) == 1
assert qs.front() == 'a'
k = qs.poll()
assert k == 'a'
assert len(qs) == 0
assert qs.front() is None
def test_poll(self):
qs = QueueSet(['one','two','three'])
assert len(qs) == 3
assert qs.front() == 'one'
k = qs.poll()
assert k == 'one'
assert len(qs) == 2
assert qs.front() == 'two'
def test_enqueue_one_thing_twice(self):
qs = QueueSet()
qs.enqueue('home')
qs.enqueue('home')
assert len(qs) == 1
assert qs.front() == 'home'
k = qs.poll()
assert k == 'home'
assert len(qs) == 0
assert qs.front() is None
