|
|
'''
|
|
|
对getBarrage中的函数进行测试python -m unittest
|
|
|
由于测试的是爬取b站数据，为了防止被ban，所以手动添加了一定的延时
|
|
|
代码覆盖率测试coverage run -m unittest discover
|
|
|
'''
|
|
|
|
|
|
import unittest
|
|
|
import time
|
|
|
import random
|
|
|
from barrage import getBarrage as bg
|
|
|
|
|
|
class TestBarrage(unittest.TestCase):
|
|
|
headers = {
|
|
|
'Cookie': '',
|
|
|
'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/122.0.6261.95 Safari/537.36'
|
|
|
}
|
|
|
# 以下是用到的测试数据
|
|
|
searchWord = '2024巴黎奥运会'
|
|
|
content_list = [
|
|
|
'//www.bilibili.com/video/BV1NtYZe2EFH',
|
|
|
'//www.bilibili.com/video/BV1CSYoerEHy',
|
|
|
'//www.bilibili.com/video/BV1mE4m1R71o',
|
|
|
'//www.bilibili.com/video/BV17zpxeQE9A',
|
|
|
'//www.bilibili.com/video/BV1Di421h7Q4',
|
|
|
'//www.bilibili.com/video/BV1tn4y1f7sR',
|
|
|
'//www.bilibili.com/video/BV1e1p2eMEKX',
|
|
|
'//www.bilibili.com/video/BV1NgiVecEns',
|
|
|
'//www.bilibili.com/video/BV1RQHcekEfU',
|
|
|
'//www.bilibili.com/video/BV1JbexeUELF'
|
|
|
]
|
|
|
cid_list = [
|
|
|
'25751522644',
|
|
|
'25751522791',
|
|
|
'25751522812',
|
|
|
'25751523009',
|
|
|
'5000401',
|
|
|
'5038962',
|
|
|
'5248',
|
|
|
'579',
|
|
|
'6871235',
|
|
|
'304154'
|
|
|
]
|
|
|
|
|
|
def setUp(self):
|
|
|
print('start')
|
|
|
|
|
|
def test_GetAllSearchVideoUrl(self):
|
|
|
for i in range(10):
|
|
|
time.sleep(0.5+random.random())
|
|
|
page = i + 1
|
|
|
o = page * 30
|
|
|
url = f'https://search.bilibili.com/all?keyword={self.searchWord}&from_source=webtop_search&spm_id_from=333.934&search_source=5&page={page}&o={o}'
|
|
|
self.assertTrue(type(bg.GetAllSearchVideoUrl(url, self.headers)) == list )
|
|
|
|
|
|
def test_GetVideoCid(self):
|
|
|
for i in range(10):
|
|
|
time.sleep(0.5+random.random())
|
|
|
url = 'https:' + self.content_list[i]
|
|
|
self.assertIsNotNone(type(bg.GetVideoCid(url, self.headers)))
|
|
|
|
|
|
def test_GetVideoBarrage(self):
|
|
|
for i in range(10):
|
|
|
time.sleep(0.5+random.random())
|
|
|
url = 'https://api.bilibili.com/x/v1/dm/list.so?oid=' + self.cid_list[i]
|
|
|
self.assertTrue(type(bg.GetVideoBarrage(url, self.headers)) == list)
|
|
|
|
|
|
def tearDown(self):
|
|
|
print('end')
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
unittest.main() |
