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)