def test_qsize(self):
"""测试qsize()"""
queue_qsize = RedisQueue("queue_qsize_" + self.key_rand)
self.keys.append(queue_qsize.key)
self.del_key(queue_qsize.key)
num = random.randint(1, 10)
for i in range(num):
queue_qsize.put(i)
self.assertEqual(queue_qsize.qsize(), num)
def test_put(self):
"""测试put()"""
queue_put = RedisQueue("queue_put_" + self.key_rand)
self.keys.append(queue_put.key)
items = [x for x in range(random.randint(1, 10))]
for i in items:
qsize = queue_put.put(i)
self.assertEqual(queue_put.qsize(), qsize)
for i in items:
item = queue_put.get()
self.assertEqual(item, i)