fix: handle subscibe dup exception

This commit is contained in:
felinae98
2022-10-09 00:36:54 +08:00
parent c944218e7b
commit ebcf1cc300
4 changed files with 69 additions and 25 deletions
+27
View File
@@ -1,4 +1,5 @@
from nonebug.app import App
import pytest
from sqlalchemy.ext.asyncio.session import AsyncSession
from sqlalchemy.sql.functions import func
from sqlmodel.sql.expression import select
@@ -71,6 +72,32 @@ async def test_add_subscribe(app: App, init_scheduler):
assert conf.categories == [1]
assert conf.tags == ["tag"]
async def test_add_dup_sub(init_scheduler):
from nonebot_bison.config.db_config import config
from nonebot_bison.types import Target as TTarget
from nonebot_bison.config.db_config import SubscribeDupException
await config.add_subscribe(
user=123,
user_type="group",
target=TTarget("weibo_id"),
target_name="weibo_name",
platform_name="weibo",
cats=[],
tags=[],
)
with pytest.raises(SubscribeDupException):
await config.add_subscribe(
user=123,
user_type="group",
target=TTarget("weibo_id"),
target_name="weibo_name",
platform_name="weibo",
cats=[],
tags=[],
)
async def test_del_subsribe(init_scheduler):
from nonebot_bison.config.db_config import config