This commit is contained in:
felinae98
2021-09-07 17:44:32 +08:00
parent 9c7b3a49bc
commit 8e61e27127
7 changed files with 58 additions and 8 deletions
File diff suppressed because one or more lines are too long
+49
View File
@@ -0,0 +1,49 @@
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_hk_reporter
@pytest.fixture
def ncm_artist(plugin_module: 'nonebot_hk_reporter'):
return plugin_module.platform.platform_manager['ncm-artist']
@pytest.fixture(scope='module')
def ncm_artist_raw():
return get_json('ncm_siren.json')
@pytest.fixture(scope='module')
def ncm_artist_0(ncm_artist_raw):
return {
**ncm_artist_raw,
'hotAlbums': ncm_artist_raw['hotAlbums'][1:]
}
@pytest.fixture(scope='module')
def ncm_artist_1(ncm_artist_raw: dict):
res = ncm_artist_raw.copy()
res['hotAlbums'] = ncm_artist_raw['hotAlbums'][:]
res['hotAlbums'][0]['publishTime'] = int(time.time() * 1000)
return res
@pytest.mark.asyncio
@respx.mock
async def test_fetch_new(ncm_artist, ncm_artist_0, ncm_artist_1, dummy_user_subinfo):
ncm_router = respx.get("https://music.163.com/api/artist/albums/32540734")
ncm_router.mock(return_value=Response(200, json=ncm_artist_0))
target = '32540734'
res = await ncm_artist.fetch_new_post(target, [dummy_user_subinfo])
assert(ncm_router.called)
assert(len(res) == 0)
ncm_router.mock(return_value=Response(200, json=ncm_artist_1))
res2 = await ncm_artist.fetch_new_post(target, [dummy_user_subinfo])
post = res2[0][1][0]
assert(post.target_type == 'ncm-artist')
assert(post.text == '新专辑发布:Y1K')
assert(post.url == 'https://music.163.com/#/album?id=131074504')