mirror of
https://github.com/suyiiyii/nonebot-bison.git
synced 2025-06-05 19:36:43 +08:00
90816796c7
* 🎨 修改 nonebot_bison 目录位置 * auto fix by pre-commit hooks * 🚚 fix frontend build target * 🚚 use soft link * Revert "🚚 use soft link" This reverts commit de21f79d5ae1bd5515b04f42a4138cb25ddf3e62. * 🚚 modify dockerfile --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: felinae98 <731499577@qq.com>
224 lines
7.3 KiB
Python
224 lines
7.3 KiB
Python
from datetime import time
|
|
|
|
from nonebug import App
|
|
from pytest_mock import MockerFixture
|
|
|
|
|
|
async def test_create_config(init_scheduler):
|
|
from nonebot_plugin_datastore.db import get_engine
|
|
|
|
from nonebot_bison.config.db_config import TimeWeightConfig, WeightConfig, config
|
|
from nonebot_bison.config.db_model import Subscribe, Target, User
|
|
from nonebot_bison.types import Target as T_Target
|
|
|
|
await config.add_subscribe(
|
|
user=123,
|
|
user_type="group",
|
|
target=T_Target("weibo_id"),
|
|
target_name="weibo_name",
|
|
platform_name="weibo",
|
|
cats=[],
|
|
tags=[],
|
|
)
|
|
await config.add_subscribe(
|
|
user=123,
|
|
user_type="group",
|
|
target=T_Target("weibo_id1"),
|
|
target_name="weibo_name1",
|
|
platform_name="weibo",
|
|
cats=[],
|
|
tags=[],
|
|
)
|
|
await config.update_time_weight_config(
|
|
target=T_Target("weibo_id"),
|
|
platform_name="weibo",
|
|
conf=WeightConfig(
|
|
default=10,
|
|
time_config=[
|
|
TimeWeightConfig(start_time=time(1, 0), end_time=time(2, 0), weight=20)
|
|
],
|
|
),
|
|
)
|
|
|
|
test_config = await config.get_time_weight_config(
|
|
target=T_Target("weibo_id"), platform_name="weibo"
|
|
)
|
|
assert test_config.default == 10
|
|
assert test_config.time_config == [
|
|
TimeWeightConfig(start_time=time(1, 0), end_time=time(2, 0), weight=20)
|
|
]
|
|
test_config1 = await config.get_time_weight_config(
|
|
target=T_Target("weibo_id1"), platform_name="weibo"
|
|
)
|
|
assert test_config1.default == 10
|
|
assert test_config1.time_config == []
|
|
|
|
|
|
async def test_get_current_weight(init_scheduler, mocker: MockerFixture):
|
|
from datetime import time
|
|
|
|
from nonebot_plugin_datastore.db import get_engine
|
|
|
|
from nonebot_bison.config import db_config
|
|
from nonebot_bison.config.db_config import TimeWeightConfig, WeightConfig, config
|
|
from nonebot_bison.config.db_model import Subscribe, Target, User
|
|
from nonebot_bison.types import Target as T_Target
|
|
|
|
await config.add_subscribe(
|
|
user=123,
|
|
user_type="group",
|
|
target=T_Target("weibo_id"),
|
|
target_name="weibo_name",
|
|
platform_name="weibo",
|
|
cats=[],
|
|
tags=[],
|
|
)
|
|
await config.add_subscribe(
|
|
user=123,
|
|
user_type="group",
|
|
target=T_Target("weibo_id1"),
|
|
target_name="weibo_name1",
|
|
platform_name="weibo",
|
|
cats=[],
|
|
tags=[],
|
|
)
|
|
await config.add_subscribe(
|
|
user=123,
|
|
user_type="group",
|
|
target=T_Target("weibo_id1"),
|
|
target_name="weibo_name2",
|
|
platform_name="bilibili",
|
|
cats=[],
|
|
tags=[],
|
|
)
|
|
await config.update_time_weight_config(
|
|
target=T_Target("weibo_id"),
|
|
platform_name="weibo",
|
|
conf=WeightConfig(
|
|
default=10,
|
|
time_config=[
|
|
TimeWeightConfig(start_time=time(1, 0), end_time=time(2, 0), weight=20),
|
|
TimeWeightConfig(start_time=time(4, 0), end_time=time(5, 0), weight=30),
|
|
],
|
|
),
|
|
)
|
|
mocker.patch.object(db_config, "_get_time", return_value=time(1, 30))
|
|
weight = await config.get_current_weight_val(["weibo", "bilibili"])
|
|
assert len(weight) == 3
|
|
assert weight["weibo-weibo_id"] == 20
|
|
assert weight["weibo-weibo_id1"] == 10
|
|
assert weight["bilibili-weibo_id1"] == 10
|
|
mocker.patch.object(db_config, "_get_time", return_value=time(4, 0))
|
|
weight = await config.get_current_weight_val(["weibo", "bilibili"])
|
|
assert len(weight) == 3
|
|
assert weight["weibo-weibo_id"] == 30
|
|
assert weight["weibo-weibo_id1"] == 10
|
|
assert weight["bilibili-weibo_id1"] == 10
|
|
mocker.patch.object(db_config, "_get_time", return_value=time(5, 0))
|
|
weight = await config.get_current_weight_val(["weibo", "bilibili"])
|
|
assert len(weight) == 3
|
|
assert weight["weibo-weibo_id"] == 10
|
|
assert weight["weibo-weibo_id1"] == 10
|
|
assert weight["bilibili-weibo_id1"] == 10
|
|
|
|
|
|
async def test_get_platform_target(app: App, init_scheduler):
|
|
from nonebot_plugin_datastore.db import get_engine
|
|
from sqlalchemy.ext.asyncio.session import AsyncSession
|
|
from sqlalchemy.sql.expression import select
|
|
|
|
from nonebot_bison.config import db_config
|
|
from nonebot_bison.config.db_config import TimeWeightConfig, WeightConfig, config
|
|
from nonebot_bison.config.db_model import Subscribe, Target, User
|
|
from nonebot_bison.types import Target as T_Target
|
|
|
|
await config.add_subscribe(
|
|
user=123,
|
|
user_type="group",
|
|
target=T_Target("weibo_id"),
|
|
target_name="weibo_name",
|
|
platform_name="weibo",
|
|
cats=[],
|
|
tags=[],
|
|
)
|
|
await config.add_subscribe(
|
|
user=123,
|
|
user_type="group",
|
|
target=T_Target("weibo_id1"),
|
|
target_name="weibo_name1",
|
|
platform_name="weibo",
|
|
cats=[],
|
|
tags=[],
|
|
)
|
|
await config.add_subscribe(
|
|
user=245,
|
|
user_type="group",
|
|
target=T_Target("weibo_id1"),
|
|
target_name="weibo_name1",
|
|
platform_name="weibo",
|
|
cats=[],
|
|
tags=[],
|
|
)
|
|
res = await config.get_platform_target("weibo")
|
|
assert len(res) == 2
|
|
await config.del_subscribe(123, "group", T_Target("weibo_id1"), "weibo")
|
|
res = await config.get_platform_target("weibo")
|
|
assert len(res) == 2
|
|
await config.del_subscribe(123, "group", T_Target("weibo_id"), "weibo")
|
|
res = await config.get_platform_target("weibo")
|
|
assert len(res) == 1
|
|
|
|
async with AsyncSession(get_engine()) as sess:
|
|
res = await sess.scalars(select(Target).where(Target.platform_name == "weibo"))
|
|
assert len(res.all()) == 2
|
|
|
|
|
|
async def test_get_platform_target_subscribers(app: App, init_scheduler):
|
|
from nonebot_plugin_datastore.db import get_engine
|
|
from sqlalchemy.ext.asyncio.session import AsyncSession
|
|
from sqlalchemy.sql.expression import select
|
|
|
|
from nonebot_bison.config import db_config
|
|
from nonebot_bison.config.db_config import TimeWeightConfig, WeightConfig, config
|
|
from nonebot_bison.config.db_model import Subscribe, Target, User
|
|
from nonebot_bison.types import Target as T_Target
|
|
from nonebot_bison.types import User as T_User
|
|
from nonebot_bison.types import UserSubInfo
|
|
|
|
await config.add_subscribe(
|
|
user=123,
|
|
user_type="group",
|
|
target=T_Target("weibo_id"),
|
|
target_name="weibo_name",
|
|
platform_name="weibo",
|
|
cats=[1],
|
|
tags=["tag1"],
|
|
)
|
|
await config.add_subscribe(
|
|
user=123,
|
|
user_type="group",
|
|
target=T_Target("weibo_id1"),
|
|
target_name="weibo_name1",
|
|
platform_name="weibo",
|
|
cats=[2],
|
|
tags=["tag2"],
|
|
)
|
|
await config.add_subscribe(
|
|
user=245,
|
|
user_type="group",
|
|
target=T_Target("weibo_id1"),
|
|
target_name="weibo_name1",
|
|
platform_name="weibo",
|
|
cats=[3],
|
|
tags=["tag3"],
|
|
)
|
|
|
|
res = await config.get_platform_target_subscribers("weibo", T_Target("weibo_id"))
|
|
assert len(res) == 1
|
|
assert res[0] == UserSubInfo(T_User(123, "group"), [1], ["tag1"])
|
|
|
|
res = await config.get_platform_target_subscribers("weibo", T_Target("weibo_id1"))
|
|
assert len(res) == 2
|
|
assert UserSubInfo(T_User(123, "group"), [2], ["tag2"]) in res
|
|
assert UserSubInfo(T_User(245, "group"), [3], ["tag3"]) in res
|