def test_mmqrz_select():
with kauli.test.redis.RedisServer() as tredis:
connect_db('localhost', tredis.port, 0)
def setup(mmqrz):
fruits = mmqrz.qadd('fruits', 200)
fruits.put('apple')
people = mmqrz.qadd('people', 150)
people.put('Mike')
mountains = mmqrz.qadd('mountains', 100)
mountains.put('fuji', 3776)
mountains.put('takao', 599)
mmqrz = Mmqrz()
setup(mmqrz)
queue = mmqrz.qselect()
eq_(queue.name, 'fruits')
eq_(queue.get().value, 'apple')
queue = mmqrz.qselect()
eq_(queue.name, 'people')
eq_(queue.get().value, 'Mike')
queue = mmqrz.qselect()
eq_(queue.name, 'mountains')
eq_(queue.get().value, 'takao')
queue = mmqrz.qselect()
eq_(queue.name, 'mountains')
eq_(queue.get().value, 'fuji')
def test_mmqrz():
def assert_queues(got, expect):
eq_([(x.name, x.score) for x in got], expect)
with kauli.test.redis.RedisServer() as tredis:
connect_db('localhost', tredis.port, 0)
mmqrz = Mmqrz()
assert_queues(mmqrz.qall(), [])
# add queues
fruits = mmqrz.qadd('fruits')
assert_true(isinstance(fruits, Queue))
mmqrz.qadd('mountains', score=20)
mmqrz.qadd('people', score=5)
assert_queues(mmqrz.qall(),
[('mountains', 20.0), ('fruits', 10.0), ('people', 5.0)])
# get queue
people = mmqrz.qget('people')
assert_true(isinstance(people, Queue))
eq_(people.name, 'people')
assert_raises(IndexError, mmqrz.qget, 'books')
# remove queue
mmqrz.qrem('fruits')
assert_queues(mmqrz.qall(), [('mountains', 20.0), ('people', 5.0)])
# qrem also delete tasks
people.put('Mike')
eq_(people.qsize(), 1)
mmqrz.qrem('people')
eq_(mmqrz.qadd('people').qsize(), 0)
def test_queue():
with kauli.test.redis.RedisServer() as tredis:
connect_db('localhost', tredis.port, 0)
queue = Queue('fruits')
assert_true(isinstance(queue, Queue))
eq_(queue.name, 'fruits')
eq_(queue.score, 10)
queue.put('apple')
queue.put('banana')
queue.put('orange')
eq_(queue.qsize(), 3)
eq_(queue.get().value, 'apple')
eq_(queue.get().value, 'banana')
eq_(queue.get().value, 'orange')
assert_raises(IndexError, queue.get)
eq_(queue.qsize(), 0)
# discard all tasks
queue.put('apple')
queue.put('banana')
eq_(queue.qsize(), 2)
queue.clear()
eq_(queue.qsize(), 0)
def test_queue_withscore():
def assert_task(task, value, score):
eq_(task.value, value)
eq_(task.score, score)
with kauli.test.redis.RedisServer() as tredis:
connect_db('localhost', tredis.port, 0)
queue = Queue('mountains', 20)
eq_(queue.score, 20)
queue.put('fuji', 3776)
queue.put('takao', 599)
queue.put('everest', 8848)
assert_task(queue.get(), 'takao', 599)
assert_task(queue.get(), 'fuji', 3776)
assert_task(queue.get(), 'everest', 8848)
assert_raises(IndexError, queue.get)