suyiiyii/nonebot-bison
1
0
Fork 0
mirror of https://github.com/suyiiyii/nonebot-bison.git synced 2026-05-09 18:27:56 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix platform store init error

Browse Source
This commit is contained in:
felinae98
2022-10-17 19:06:30 +08:00
parent 7449a5519b
commit 57c1e504ce
2 changed files with 57 additions and 43 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/platforms/test_platform.py
+49 -37
View File
@@ -2,6 +2,7 @@ from time import time
from typing import Any, Optional
import pytest
from httpx import AsyncClient
from nonebug.app import App
now = time()
@@ -53,8 +54,9 @@ def mock_platform_without_cats_tags(app: App):
categories = {}
has_target = True
sub_index = 0
def __init__(self, client):
self.sub_index = 0
super().__init__(client)
@classmethod
@@ -75,14 +77,15 @@ def mock_platform_without_cats_tags(app: App):
target_name="Mock",
)
async def get_sub_list(self, _: "Target"):
if self.sub_index == 0:
self.sub_index += 1
@classmethod
async def get_sub_list(cls, _: "Target"):
if cls.sub_index == 0:
cls.sub_index += 1
return raw_post_list_1
else:
return raw_post_list_2
return MockPlatform(None)
return MockPlatform
@pytest.fixture
@@ -112,8 +115,9 @@ def mock_platform(app: App):
Category(2): "视频",
}
sub_index = 0
def __init__(self, client):
self.sub_index = 0
super().__init__(client)
@staticmethod
@@ -140,14 +144,15 @@ def mock_platform(app: App):
target_name="Mock",
)
async def get_sub_list(self, _: "Target"):
if self.sub_index == 0:
self.sub_index += 1
@classmethod
async def get_sub_list(cls, _: "Target"):
if cls.sub_index == 0:
cls.sub_index += 1
return raw_post_list_1
else:
return raw_post_list_2
return MockPlatform(None)
return MockPlatform
@pytest.fixture
@@ -180,8 +185,9 @@ def mock_platform_no_target(app: App, mock_scheduler_conf):
has_target = False
categories = {Category(1): "转发", Category(2): "视频", Category(3): "不支持"}
sub_index = 0
def __init__(self, client):
self.sub_index = 0
super().__init__(client)
@staticmethod
@@ -210,9 +216,10 @@ def mock_platform_no_target(app: App, mock_scheduler_conf):
target_name="Mock",
)
async def get_sub_list(self, _: "Target"):
if self.sub_index == 0:
self.sub_index += 1
@classmethod
async def get_sub_list(cls, _: "Target"):
if cls.sub_index == 0:
cls.sub_index += 1
return raw_post_list_1
else:
return raw_post_list_2
@@ -241,8 +248,9 @@ def mock_platform_no_target_2(app: App, mock_scheduler_conf):
Category(5): "leixing5",
}
sub_index = 0
def __init__(self, client):
self.sub_index = 0
super().__init__(client)
@classmethod
@@ -269,7 +277,8 @@ def mock_platform_no_target_2(app: App, mock_scheduler_conf):
target_name="Mock",
)
async def get_sub_list(self, _: "Target"):
@classmethod
async def get_sub_list(cls, _: "Target"):
list_1 = [
{"id": 5, "text": "p5", "date": now, "tags": ["tag1"], "category": 4}
]
@@ -278,8 +287,8 @@ def mock_platform_no_target_2(app: App, mock_scheduler_conf):
{"id": 6, "text": "p6", "date": now, "tags": ["tag1"], "category": 4},
{"id": 7, "text": "p7", "date": now, "tags": ["tag2"], "category": 5},
]
if self.sub_index == 0:
self.sub_index += 1
if cls.sub_index == 0:
cls.sub_index += 1
return list_1
else:
return list_2
@@ -308,16 +317,18 @@ def mock_status_change(app: App):
Category(2): "视频",
}
sub_index = 0
def __init__(self, client):
self.sub_index = 0
super().__init__(client)
async def get_status(self, _: "Target"):
if self.sub_index == 0:
self.sub_index += 1
@classmethod
async def get_status(cls, _: "Target"):
if cls.sub_index == 0:
cls.sub_index += 1
return {"s": False}
elif self.sub_index == 1:
self.sub_index += 1
elif cls.sub_index == 1:
cls.sub_index += 1
return {"s": True}
else:
return {"s": False}
@@ -335,18 +346,18 @@ def mock_status_change(app: App):
def get_category(self, raw_post):
return raw_post["cat"]
return MockPlatform(None)
return MockPlatform
@pytest.mark.asyncio
async def test_new_message_target_without_cats_tags(
mock_platform_without_cats_tags, user_info_factory
):
res1 = await mock_platform_without_cats_tags.fetch_new_post(
res1 = await mock_platform_without_cats_tags(AsyncClient()).fetch_new_post(
"dummy", [user_info_factory([1, 2], [])]
)
assert len(res1) == 0
res2 = await mock_platform_without_cats_tags.fetch_new_post(
res2 = await mock_platform_without_cats_tags(AsyncClient()).fetch_new_post(
"dummy",
[
user_info_factory([], []),
@@ -361,9 +372,11 @@ async def test_new_message_target_without_cats_tags(
@pytest.mark.asyncio
async def test_new_message_target(mock_platform, user_info_factory):
res1 = await mock_platform.fetch_new_post("dummy", [user_info_factory([1, 2], [])])
res1 = await mock_platform(AsyncClient()).fetch_new_post(
"dummy", [user_info_factory([1, 2], [])]
)
assert len(res1) == 0
res2 = await mock_platform.fetch_new_post(
res2 = await mock_platform(AsyncClient()).fetch_new_post(
"dummy",
[
user_info_factory([1, 2], []),
@@ -388,12 +401,11 @@ async def test_new_message_target(mock_platform, user_info_factory):
@pytest.mark.asyncio
async def test_new_message_no_target(mock_platform_no_target, user_info_factory):
mock_platform_no_target = mock_platform_no_target(None)
res1 = await mock_platform_no_target.fetch_new_post(
res1 = await mock_platform_no_target(AsyncClient()).fetch_new_post(
"dummy", [user_info_factory([1, 2], [])]
)
assert len(res1) == 0
res2 = await mock_platform_no_target.fetch_new_post(
res2 = await mock_platform_no_target(AsyncClient()).fetch_new_post(
"dummy",
[
user_info_factory([1, 2], []),
@@ -414,7 +426,7 @@ async def test_new_message_no_target(mock_platform_no_target, user_info_factory)
assert "p2" in id_set_1 and "p3" in id_set_1
assert "p2" in id_set_2
assert "p2" in id_set_3
res3 = await mock_platform_no_target.fetch_new_post(
res3 = await mock_platform_no_target(AsyncClient()).fetch_new_post(
"dummy", [user_info_factory([1, 2], [])]
)
assert len(res3) == 0
@@ -422,18 +434,18 @@ async def test_new_message_no_target(mock_platform_no_target, user_info_factory)
@pytest.mark.asyncio
async def test_status_change(mock_status_change, user_info_factory):
res1 = await mock_status_change.fetch_new_post(
res1 = await mock_status_change(AsyncClient()).fetch_new_post(
"dummy", [user_info_factory([1, 2], [])]
)
assert len(res1) == 0
res2 = await mock_status_change.fetch_new_post(
res2 = await mock_status_change(AsyncClient()).fetch_new_post(
"dummy", [user_info_factory([1, 2], [])]
)
assert len(res2) == 1
posts = res2[0][1]
assert len(posts) == 1
assert posts[0].text == "on"
res3 = await mock_status_change.fetch_new_post(
res3 = await mock_status_change(AsyncClient()).fetch_new_post(
"dummy",
[
user_info_factory([1, 2], []),
@@ -444,7 +456,7 @@ async def test_status_change(mock_status_change, user_info_factory):
assert len(res3[0][1]) == 1
assert res3[0][1][0].text == "off"
assert len(res3[1][1]) == 0
res4 = await mock_status_change.fetch_new_post(
res4 = await mock_status_change(AsyncClient()).fetch_new_post(
"dummy", [user_info_factory([1, 2], [])]
)
assert len(res4) == 0