nonebot-bison/tests/platforms/test_ncm_radio.py
2021-11-22 17:58:47 +08:00

54 lines
1.8 KiB
Python

from .utils import get_json
import pytest
import respx
import typing
import time
from httpx import Response
if typing.TYPE_CHECKING:
import sys
sys.path.append('./src/plugins')
import nonebot_bison
@pytest.fixture
def ncm_radio(plugin_module: 'nonebot_bison'):
return plugin_module.platform.platform_manager['ncm-radio']
@pytest.fixture(scope='module')
def ncm_radio_raw():
return get_json('ncm_radio_ark.json')
@pytest.fixture(scope='module')
def ncm_radio_0(ncm_radio_raw):
return {
**ncm_radio_raw,
'programs': ncm_radio_raw['programs'][1:]
}
@pytest.fixture(scope='module')
def ncm_radio_1(ncm_radio_raw: dict):
res = ncm_radio_raw.copy()
res['programs'] = ncm_radio_raw['programs'][:]
res['programs'][0]['createTime'] = int(time.time() * 1000)
return res
@pytest.mark.asyncio
@respx.mock
async def test_fetch_new(ncm_radio, ncm_radio_0, ncm_radio_1, dummy_user_subinfo):
ncm_router = respx.post("http://music.163.com/api/dj/program/byradio")
ncm_router.mock(return_value=Response(200, json=ncm_radio_0))
target = '793745436'
res = await ncm_radio.fetch_new_post(target, [dummy_user_subinfo])
assert(ncm_router.called)
assert(len(res) == 0)
ncm_router.mock(return_value=Response(200, json=ncm_radio_1))
res2 = await ncm_radio.fetch_new_post(target, [dummy_user_subinfo])
post = res2[0][1][0]
print(post)
assert(post.target_type == 'ncm-radio')
assert(post.text == '网易云电台更新:「松烟行动」灰齐山麓')
assert(post.url == 'https://music.163.com/#/program/2494997688')
assert(post.pics == ['http://p1.music.126.net/H5em5xUNIYXcjJhOmeaSqQ==/109951166647436789.jpg'])
assert(post.target_name == '《明日方舟》游戏原声OST')