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
