mirror of
https://github.com/suyiiyii/nonebot-bison.git
synced 2025-06-02 09:26:12 +08:00
190 lines
6.5 KiB
Python
190 lines
6.5 KiB
Python
from typing import cast
|
|
from pathlib import Path
|
|
from unittest.mock import patch
|
|
|
|
from nonebug.app import App
|
|
from click.core import BaseCommand
|
|
from click.testing import CliRunner
|
|
|
|
from .utils import get_file
|
|
|
|
|
|
def test_cli_help(app: App):
|
|
from nonebot_bison.script.cli import cli
|
|
|
|
cli = cast(BaseCommand, cli)
|
|
runner = CliRunner()
|
|
|
|
result = runner.invoke(cli, ["--help"])
|
|
assert result.exit_code == 0
|
|
|
|
for arg in ["subs-export", "subs-import"]:
|
|
assert arg not in result.output
|
|
|
|
for arg in ["export", "import"]:
|
|
assert arg in result.output
|
|
|
|
result = runner.invoke(cli, ["export", "--help"])
|
|
assert result.exit_code == 0
|
|
|
|
for opt in ["--path", "-p", "导出路径", "--format", "指定导出格式[json, yaml]，默认为 json"]:
|
|
assert opt in result.output
|
|
|
|
result = runner.invoke(cli, ["import", "--help"])
|
|
assert result.exit_code == 0
|
|
|
|
for opt in ["--path", "-p", "导入文件名", "--format", "指定导入格式[json, yaml]，默认为 json"]:
|
|
assert opt in result.output
|
|
|
|
|
|
async def test_subs_export(app: App, tmp_path: Path):
|
|
from nonebot_plugin_saa import TargetQQGroup
|
|
|
|
from nonebot_bison.config.db_model import Cookie
|
|
from nonebot_bison.config.db_config import config
|
|
from nonebot_bison.types import Target as TTarget
|
|
from nonebot_bison.script.cli import cli, run_sync
|
|
|
|
cli = cast(BaseCommand, cli)
|
|
|
|
await config.add_subscribe(
|
|
TargetQQGroup(group_id=123),
|
|
target=TTarget("weibo_id"),
|
|
target_name="weibo_name",
|
|
platform_name="weibo",
|
|
cats=[],
|
|
tags=[],
|
|
)
|
|
await config.add_subscribe(
|
|
TargetQQGroup(group_id=234),
|
|
target=TTarget("weibo_id"),
|
|
target_name="weibo_name",
|
|
platform_name="weibo",
|
|
cats=[],
|
|
tags=["kaltsit", "amiya"],
|
|
)
|
|
await config.add_subscribe(
|
|
TargetQQGroup(group_id=234),
|
|
target=TTarget("bilibili_id"),
|
|
target_name="bilibili_name",
|
|
platform_name="bilibili",
|
|
cats=[1, 2],
|
|
tags=[],
|
|
)
|
|
cookie_id = await config.add_cookie(
|
|
Cookie(
|
|
site_name="weibo.com",
|
|
content='{"cookie": "test"}',
|
|
cookie_name="test cookie",
|
|
)
|
|
)
|
|
await config.add_cookie_target(TTarget("weibo_id"), "weibo", cookie_id)
|
|
|
|
assert len(await config.list_subs_with_all_info()) == 3
|
|
|
|
with patch("time.time") as mock_time:
|
|
mock_now = mock_time.now.return_value
|
|
mock_now.time.return_value = 1
|
|
|
|
runner = CliRunner()
|
|
# 是否默认导出到工作目录
|
|
with runner.isolated_filesystem(temp_dir=tmp_path):
|
|
result = await run_sync(runner.invoke)(cli, ["export"])
|
|
assert result.exit_code == 0
|
|
file_path = Path.cwd() / "bison_subscribes_export_1.json"
|
|
assert file_path.exists()
|
|
assert '"version": 3' in file_path.read_text()
|
|
assert '"group_id": 123' in file_path.read_text()
|
|
assert '"content": "{\\"cookie\\": \\"test\\"}",\n' in file_path.read_text()
|
|
|
|
# 是否导出到指定已存在文件夹
|
|
data_dir = tmp_path / "data"
|
|
data_dir.mkdir()
|
|
result = await run_sync(runner.invoke)(cli, ["export", "-p", str(data_dir)])
|
|
assert result.exit_code == 0
|
|
file_path2 = data_dir / "bison_subscribes_export_1.json"
|
|
assert file_path2.exists()
|
|
assert '"version": 3' in file_path2.read_text()
|
|
assert '"group_id": 123' in file_path2.read_text()
|
|
assert '"content": "{\\"cookie\\": \\"test\\"}",\n' in file_path.read_text()
|
|
|
|
# 是否拒绝导出到不存在的文件夹
|
|
result = await run_sync(runner.invoke)(cli, ["export", "-p", str(tmp_path / "data2")])
|
|
assert result.exit_code == 1
|
|
|
|
# 是否可以以yaml格式导出
|
|
result = await run_sync(runner.invoke)(cli, ["export", "-p", str(tmp_path), "--format", "yaml"])
|
|
assert result.exit_code == 0
|
|
file_path3 = tmp_path / "bison_subscribes_export_1.yaml"
|
|
assert file_path3.exists()
|
|
assert "version: 3" in file_path3.read_text()
|
|
assert "group_id: 123" in file_path3.read_text()
|
|
assert "platform_type: QQ Group" in file_path3.read_text()
|
|
assert '"content": "{\\"cookie\\": \\"test\\"}",\n' in file_path.read_text()
|
|
|
|
# 是否允许以未支持的格式导出
|
|
result = await run_sync(runner.invoke)(cli, ["export", "-p", str(tmp_path), "--format", "toml"])
|
|
assert result.exit_code == 2
|
|
|
|
|
|
async def test_subs_import_v1(app: App, tmp_path):
|
|
from nonebot_bison.config.db_config import config
|
|
from nonebot_bison.script.cli import cli, run_sync
|
|
|
|
cli = cast(BaseCommand, cli)
|
|
|
|
assert len(await config.list_subs_with_all_info()) == 0
|
|
|
|
mock_file: Path = tmp_path / "1.json"
|
|
mock_file.write_text(get_file("v1/subs_export.json"))
|
|
|
|
runner = CliRunner()
|
|
|
|
result = await run_sync(runner.invoke)(cli, ["import"])
|
|
assert result.exit_code == 2
|
|
|
|
result = await run_sync(runner.invoke)(cli, ["import", "-p"])
|
|
assert result.exit_code == 2
|
|
|
|
result = await run_sync(runner.invoke)(cli, ["import", "-p", str(mock_file)])
|
|
assert result.exit_code == 0
|
|
assert len(await config.list_subs_with_all_info()) == 3
|
|
|
|
mock_file2: Path = tmp_path / "2.yaml"
|
|
mock_file2.write_text(get_file("v1/subs_export.yaml"))
|
|
|
|
result = await run_sync(runner.invoke)(cli, ["import", "-p", str(mock_file2), "--format=yml"])
|
|
assert result.exit_code == 0
|
|
assert len(await config.list_subs_with_all_info()) == 6
|
|
|
|
|
|
async def test_sub_import_v2(app: App, tmp_path):
|
|
from nonebot_bison.config.db_config import config
|
|
from nonebot_bison.script.cli import cli, run_sync
|
|
|
|
cli = cast(BaseCommand, cli)
|
|
|
|
assert len(await config.list_subs_with_all_info()) == 0
|
|
|
|
mock_file: Path = tmp_path / "1.json"
|
|
mock_file.write_text(get_file("v2/subs_export.json"))
|
|
|
|
runner = CliRunner()
|
|
|
|
result = await run_sync(runner.invoke)(cli, ["import"])
|
|
assert result.exit_code == 2
|
|
|
|
result = await run_sync(runner.invoke)(cli, ["import", "-p"])
|
|
assert result.exit_code == 2
|
|
|
|
result = await run_sync(runner.invoke)(cli, ["import", "-p", str(mock_file)])
|
|
assert result.exit_code == 0
|
|
assert len(await config.list_subs_with_all_info()) == 3
|
|
|
|
mock_file2: Path = tmp_path / "2.yaml"
|
|
mock_file2.write_text(get_file("v2/subs_export.yaml"))
|
|
|
|
result = await run_sync(runner.invoke)(cli, ["import", "-p", str(mock_file2), "--format=yml"])
|
|
assert result.exit_code == 0
|
|
assert len(await config.list_subs_with_all_info()) == 6
|