class TestDanmakuQueue(TestCase):
def setUp(self):
room_id = NUMBER
self.danmaku_queue = DanmakuQueue(room_id)
def test_count(self):
count = NUMBER
self.count = count
eq_(self.count, count)
def test_enqueue(self):
self.assertFalse(self.danmaku_queue.enqueue([]))
danmaku = Mock(DanmakuModel)
self.assertTrue(self.danmaku_queue.enqueue(danmaku))
def test_dequeue(self):
self.assertIsNone(self.danmaku_queue.dequeue())
danmaku = Mock(spec=DanmakuModel)
self.danmaku_queue._queue.append(danmaku)
eq_(self.danmaku_queue.dequeue(), danmaku)
class TestDanmakuQueue(TestCase):
def setUp(self):
room_id = NUMBER
self.danmaku_queue = DanmakuQueue(room_id)
def test_count(self):
count = NUMBER
self.count = count
eq_(self.count, count)
def test_enqueue(self):
self.assertFalse(self.danmaku_queue.enqueue([]))
danmaku = Mock(DanmakuModel)
self.assertTrue(self.danmaku_queue.enqueue(danmaku))
def test_dequeue(self):
self.assertIsNone(self.danmaku_queue.dequeue())
danmaku = Mock(spec=DanmakuModel)
self.danmaku_queue._queue.append(danmaku)
eq_(self.danmaku_queue.dequeue(), danmaku)
def setUp(self):
room_id = NUMBER
self.danmaku_queue = DanmakuQueue(room_id)
def test_dequeue(self):
test_danmaku_queue = DanmakuQueue(self.room_id)
assert test_danmaku_queue.dequeue() == None
danmaku = Mock(spec=DanmakuModel)
test_danmaku_queue.enqueue(danmaku)
assert test_danmaku_queue.dequeue() == danmaku
def test_enqueue(self):
test_danmaku_queue = DanmakuQueue(self.room_id)
assert test_danmaku_queue.enqueue([]) == False
danmaku = Mock(spec=DanmakuModel)
assert test_danmaku_queue.enqueue(danmaku) == True
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
def setUp(self):
room_id = NUMBER
self.danmaku_queue = DanmakuQueue(room_id)
