yml/live-hub-py
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

type: 收窄测试与响应模型类型

Browse Source
This commit is contained in:
yml2213
2026-08-30 20:21:14 +08:00
parent 3f97dbe525
commit cbd314f4e5
7 changed files with 96 additions and 69 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_douyu_workbench_scopes.py
+10 -17
View File
@@ -1,5 +1,4 @@
import pytest
from types import SimpleNamespace
from sqlalchemy import create_engine, event
from sqlalchemy.orm import sessionmaker
@@ -54,12 +53,6 @@ class TestDouyuWorkbenchScope:
)
)
self.session.commit()
self.current = SimpleNamespace(
id=self.user.id,
username=self.user.username,
role=self.user.role,
custom_permissions=None,
)
def teardown_method(self):
self.session.close()
@@ -72,12 +65,10 @@ class TestDouyuWorkbenchScope:
handbook_scope="elite", account_ids=[self.account.id]
),
db=self.session,
current=self.current,
)
elite = list_workbench_accounts("elite", db=self.session, current=self.current)
esports = list_workbench_accounts(
"esports", db=self.session, current=self.current
current=self.user,
)
elite = list_workbench_accounts("elite", db=self.session, current=self.user)
esports = list_workbench_accounts("esports", db=self.session, current=self.user)
assert elite["account_ids"] == [self.account.id]
assert esports["account_ids"] == []
assert elite["configured"]
@@ -87,10 +78,10 @@ class TestDouyuWorkbenchScope:
update_workbench_accounts(
DouyuWorkbenchAccountsUpdate(handbook_scope="elite", account_ids=[]),
db=self.session,
current=self.current,
current=self.user,
)
empty_elite = list_workbench_accounts(
"elite", db=self.session, current=self.current
"elite", db=self.session, current=self.user
)
assert empty_elite["account_ids"] == []
assert empty_elite["configured"]
@@ -118,7 +109,7 @@ class TestDouyuWorkbenchScope:
page=None,
page_size=100,
db=self.session,
current=self.current,
current=self.user,
)
esports = list_tasks(
batch_id=None,
@@ -127,8 +118,10 @@ class TestDouyuWorkbenchScope:
page=None,
page_size=100,
db=self.session,
current=self.current,
current=self.user,
)
assert isinstance(elite, list)
assert isinstance(esports, list)
assert {item.batch_id for item in elite} == {elite_batch}
assert {item.batch_id for item in esports} == {esports_batch}
assert {
@@ -182,6 +175,6 @@ class TestDouyuWorkbenchScope:
)
self.session.commit()
delete_account(self.account.id, db=self.session, current=self.current)
delete_account(self.account.id, db=self.session, current=self.user)
assert self.session.query(DouyuWorkbenchAccount).count() == 0