def test_get_count(self): test_danmaku_queue = DanmakuQueue(self.room_id) assert test_danmaku_queue.get_count() == 0 count = TEST_NUMBER assert test_danmaku_queue.set_count(count) == True assert test_danmaku_queue.get_count() == count
def test_set_count(self): count = TEST_NUMBER test_danmaku_queue = DanmakuQueue(self.room_id) assert test_danmaku_queue.set_count(count) == True assert test_danmaku_queue.set_count(count) == False