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

add dedup for data migration

Browse Source
This commit is contained in:
felinae98
2022-10-09 21:25:11 +08:00
parent 8db0ed3fe1
commit f6e392e8db
2 changed files with 39 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/config/test_data_migration.py
+30 -4
View File
@@ -1,9 +1,6 @@
import pytest
async def test_migration(use_legacy_config):
from nonebot_bison.config.config_legacy import config as config_legacy
from nonebot_bison.config.db import data_migrate, upgrade_db
from nonebot_bison.config.db import upgrade_db
from nonebot_bison.config.db_config import config
config_legacy.add_subscribe(
@@ -54,3 +51,32 @@ async def test_migration(use_legacy_config):
assert user234_config[0].target.target == "weibo_id"
assert user234_config[0].target.target_name == "weibo_name"
assert user234_config[0].tags == []
async def test_migrate_dup(use_legacy_config):
from nonebot_bison.config.config_legacy import config as config_legacy
from nonebot_bison.config.db import upgrade_db
from nonebot_bison.config.db_config import config
config_legacy.add_subscribe(
user=123,
user_type="group",
target="weibo_id",
target_name="weibo_name",
target_type="weibo",
cats=[2, 3],
tags=[],
)
config_legacy.add_subscribe(
user=123,
user_type="group",
target="weibo_id",
target_name="weibo_name",
target_type="weibo",
cats=[2, 3],
tags=[],
)
# await data_migrate()
await upgrade_db()
user123_config = await config.list_subscribe(123, "group")
assert len(user123_config) == 1