mirror of
https://github.com/AstrBotDevs/AstrBot
synced 2026-07-15 17:30:13 +08:00
130 lines
3.9 KiB
Python
130 lines
3.9 KiB
Python
import json
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
from click.testing import CliRunner
|
|
|
|
from astrbot.cli.commands import cmd_init
|
|
from astrbot.core.utils.auth_password import verify_dashboard_password
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_init_without_initial_password_env_does_not_create_config(
|
|
monkeypatch,
|
|
tmp_path,
|
|
):
|
|
async def fake_check_dashboard(_data_path):
|
|
return None
|
|
|
|
monkeypatch.delenv("ASTRBOT_ROOT", raising=False)
|
|
monkeypatch.delenv(cmd_init.DASHBOARD_INITIAL_PASSWORD_ENV, raising=False)
|
|
monkeypatch.setattr(cmd_init, "check_dashboard", fake_check_dashboard)
|
|
(tmp_path / ".astrbot").touch()
|
|
|
|
await cmd_init.initialize_astrbot(tmp_path)
|
|
|
|
assert not (tmp_path / "data" / "cmd_config.json").exists()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_init_uses_initial_password_env_to_create_config(
|
|
monkeypatch,
|
|
tmp_path,
|
|
):
|
|
async def fake_check_dashboard(_data_path):
|
|
return None
|
|
|
|
initial_password = "AstrBotInitialPassword123"
|
|
monkeypatch.setenv("ASTRBOT_ROOT", str(tmp_path))
|
|
monkeypatch.setenv(cmd_init.DASHBOARD_INITIAL_PASSWORD_ENV, initial_password)
|
|
monkeypatch.setattr(cmd_init, "check_dashboard", fake_check_dashboard)
|
|
(tmp_path / ".astrbot").touch()
|
|
|
|
await cmd_init.initialize_astrbot(tmp_path)
|
|
|
|
config_path = tmp_path / "data" / "cmd_config.json"
|
|
config = json.loads(config_path.read_text(encoding="utf-8-sig"))
|
|
dashboard_config = config["dashboard"]
|
|
|
|
assert verify_dashboard_password(
|
|
dashboard_config["pbkdf2_password"],
|
|
initial_password,
|
|
)
|
|
assert verify_dashboard_password(
|
|
dashboard_config["password"],
|
|
initial_password,
|
|
)
|
|
assert dashboard_config["password_change_required"] is True
|
|
assert dashboard_config["password_storage_upgraded"] is True
|
|
|
|
|
|
def test_cli_main_import_does_not_create_cwd_data(tmp_path):
|
|
repo_root = Path(__file__).resolve().parents[1]
|
|
env = os.environ.copy()
|
|
env.pop("ASTRBOT_ROOT", None)
|
|
env["HOME"] = str(tmp_path / "home")
|
|
env["PYTHONPATH"] = (
|
|
str(repo_root)
|
|
if not env.get("PYTHONPATH")
|
|
else f"{repo_root}{os.pathsep}{env['PYTHONPATH']}"
|
|
)
|
|
|
|
result = subprocess.run(
|
|
[sys.executable, "-c", "import astrbot.cli.__main__"],
|
|
cwd=tmp_path,
|
|
env=env,
|
|
capture_output=True,
|
|
text=True,
|
|
check=False,
|
|
)
|
|
|
|
assert result.returncode == 0, result.stderr
|
|
assert not (tmp_path / "data").exists()
|
|
|
|
|
|
def test_init_defaults_to_user_runtime(monkeypatch, tmp_path):
|
|
async def fake_check_dashboard(_data_path):
|
|
return None
|
|
|
|
home = tmp_path / "home"
|
|
workdir = tmp_path / "workdir"
|
|
home.mkdir()
|
|
workdir.mkdir()
|
|
|
|
monkeypatch.setenv("HOME", str(home))
|
|
monkeypatch.delenv("ASTRBOT_ROOT", raising=False)
|
|
monkeypatch.chdir(workdir)
|
|
monkeypatch.setattr(cmd_init, "check_dashboard", fake_check_dashboard)
|
|
|
|
result = CliRunner().invoke(cmd_init.init, input="\n", env={"ASTRBOT_ROOT": ""})
|
|
|
|
assert result.exit_code == 0, result.output
|
|
assert (home / ".astrbot" / ".astrbot").exists()
|
|
assert (home / ".astrbot" / "data" / "config").is_dir()
|
|
assert not (workdir / "data").exists()
|
|
|
|
|
|
def test_init_can_install_to_current_directory(monkeypatch, tmp_path):
|
|
async def fake_check_dashboard(_data_path):
|
|
return None
|
|
|
|
home = tmp_path / "home"
|
|
workdir = tmp_path / "workdir"
|
|
home.mkdir()
|
|
workdir.mkdir()
|
|
|
|
monkeypatch.setenv("HOME", str(home))
|
|
monkeypatch.delenv("ASTRBOT_ROOT", raising=False)
|
|
monkeypatch.chdir(workdir)
|
|
monkeypatch.setattr(cmd_init, "check_dashboard", fake_check_dashboard)
|
|
|
|
result = CliRunner().invoke(cmd_init.init, input="2\n", env={"ASTRBOT_ROOT": ""})
|
|
|
|
assert result.exit_code == 0, result.output
|
|
assert (workdir / ".astrbot").exists()
|
|
assert (workdir / "data" / "config").is_dir()
|
|
assert not (home / ".astrbot").exists()
|