mirror of
https://github.com/AstrBotDevs/AstrBot
synced 2026-07-15 17:30:13 +08:00
e218620a37
* feat: add one-line deploy script (deploy-cli.sh) and update cli.md - Add docs/scripts/deploy-cli.sh for one-command deployment (Linux/macOS/WSL) - Update docs/zh/deploy/astrbot/cli.md to reference local script - Replace non-existent https://astrbot.app/deploy.sh with raw.githubusercontent.com URL - Add local run tip and WSL-based Windows support * fix: address PR review feedback for deploy-cli.sh - Replace sort -V with Python-native version check (macOS BSD compat) - Bump Python requirement from >=3.10 to >=3.12 (match pyproject.toml) - Detect current directory as project root (avoid nested clone) - Handle existing non-git directory explicitly - Add curl dependency check - Support ASTRBOT_REPO and ASTRBOT_DIR env vars for forks/mirrors - Update cli.md version description to match * feat: add Windows PowerShell deploy script and update docs - Add deploy-cli.ps1 for Windows native environment (PowerShell 7+) - Update cli.md to document Windows one-liner deployment - Add local script execution instructions for both bash and ps1 * refactor: standardize log messages and improve clarity in various modules Co-authored-by: Copilot <copilot@github.com> * docs: 移除一行命令快速部署部分，简化安装说明 * feat: 添加脚本以复制部署 CLI 文件并更新构建命令 * refactor: 更新日志消息以提高可读性，统一英文提示信息 --------- Co-authored-by: Soulter <905617992@qq.com> Co-authored-by: Copilot <copilot@github.com> Co-authored-by: Weilong Liao <37870767+Soulter@users.noreply.github.com>
200 lines
7.2 KiB
Python
200 lines
7.2 KiB
Python
import os
|
|
import sys
|
|
|
|
# 将项目根目录添加到 sys.path
|
|
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
|
|
|
|
from unittest import mock
|
|
|
|
import pytest
|
|
|
|
from main import check_dashboard_files, check_env
|
|
|
|
|
|
class _version_info:
|
|
def __init__(self, major, minor):
|
|
self.major = major
|
|
self.minor = minor
|
|
|
|
def __eq__(self, other):
|
|
if isinstance(other, tuple):
|
|
return (self.major, self.minor) == other[:2]
|
|
return (self.major, self.minor) == (other.major, other.minor)
|
|
|
|
def __ge__(self, other):
|
|
if isinstance(other, tuple):
|
|
return (self.major, self.minor) >= other[:2]
|
|
return (self.major, self.minor) >= (other.major, other.minor)
|
|
|
|
def __le__(self, other):
|
|
if isinstance(other, tuple):
|
|
return (self.major, self.minor) <= other[:2]
|
|
return (self.major, self.minor) <= (other.major, other.minor)
|
|
|
|
def __gt__(self, other):
|
|
if isinstance(other, tuple):
|
|
return (self.major, self.minor) > other[:2]
|
|
return (self.major, self.minor) > (other.major, other.minor)
|
|
|
|
def __lt__(self, other):
|
|
if isinstance(other, tuple):
|
|
return (self.major, self.minor) < other[:2]
|
|
return (self.major, self.minor) < (other.major, other.minor)
|
|
|
|
|
|
def test_check_env(monkeypatch):
|
|
version_info_correct = _version_info(3, 10)
|
|
version_info_wrong = _version_info(3, 9)
|
|
monkeypatch.setattr(sys, "version_info", version_info_correct)
|
|
with mock.patch("os.makedirs") as mock_makedirs:
|
|
check_env()
|
|
# check_env uses get_astrbot_*_path() which returns absolute paths,
|
|
# so just verify makedirs was called the expected number of times
|
|
assert mock_makedirs.call_count >= 4
|
|
# Verify all calls used exist_ok=True
|
|
for call_args in mock_makedirs.call_args_list:
|
|
assert call_args[1].get("exist_ok") is True
|
|
|
|
monkeypatch.setattr(sys, "version_info", version_info_wrong)
|
|
with pytest.raises(SystemExit):
|
|
check_env()
|
|
|
|
|
|
def test_check_env_appends_user_site_packages_after_runtime_paths(monkeypatch):
|
|
astrbot_root = "/tmp/astrbot-root"
|
|
site_packages_path = "/tmp/astrbot-site-packages"
|
|
original_sys_path = list(sys.path)
|
|
|
|
monkeypatch.setattr(sys, "version_info", _version_info(3, 12))
|
|
monkeypatch.setattr("main.get_astrbot_root", lambda: astrbot_root)
|
|
monkeypatch.setattr(
|
|
"main.get_astrbot_site_packages_path", lambda: site_packages_path
|
|
)
|
|
monkeypatch.setattr("main.get_astrbot_config_path", lambda: "/tmp/config")
|
|
monkeypatch.setattr("main.get_astrbot_plugin_path", lambda: "/tmp/plugins")
|
|
monkeypatch.setattr("main.get_astrbot_temp_path", lambda: "/tmp/temp")
|
|
monkeypatch.setattr("main.get_astrbot_knowledge_base_path", lambda: "/tmp/kb")
|
|
monkeypatch.setattr(sys, "path", ["/runtime/lib", *original_sys_path])
|
|
|
|
with mock.patch("os.makedirs"):
|
|
check_env()
|
|
|
|
assert sys.path[0] == astrbot_root
|
|
assert sys.path[-1] == site_packages_path
|
|
assert sys.path.index(site_packages_path) > sys.path.index("/runtime/lib")
|
|
|
|
|
|
def test_check_env_does_not_append_duplicate_user_site_packages(monkeypatch):
|
|
astrbot_root = "/tmp/astrbot-root"
|
|
site_packages_path = "/tmp/astrbot-site-packages"
|
|
original_sys_path = list(sys.path)
|
|
|
|
monkeypatch.setattr(sys, "version_info", _version_info(3, 12))
|
|
monkeypatch.setattr("main.get_astrbot_root", lambda: astrbot_root)
|
|
monkeypatch.setattr(
|
|
"main.get_astrbot_site_packages_path", lambda: site_packages_path
|
|
)
|
|
monkeypatch.setattr("main.get_astrbot_config_path", lambda: "/tmp/config")
|
|
monkeypatch.setattr("main.get_astrbot_plugin_path", lambda: "/tmp/plugins")
|
|
monkeypatch.setattr("main.get_astrbot_temp_path", lambda: "/tmp/temp")
|
|
monkeypatch.setattr("main.get_astrbot_knowledge_base_path", lambda: "/tmp/kb")
|
|
monkeypatch.setattr(
|
|
sys, "path", [astrbot_root, *original_sys_path, site_packages_path]
|
|
)
|
|
|
|
with mock.patch("os.makedirs"):
|
|
check_env()
|
|
|
|
assert sys.path.count(site_packages_path) == 1
|
|
|
|
|
|
def test_version_info_comparisons():
|
|
"""Test _version_info comparison operators with tuples and other instances."""
|
|
v3_10 = _version_info(3, 10)
|
|
v3_9 = _version_info(3, 9)
|
|
v3_11 = _version_info(3, 11)
|
|
|
|
# Test __eq__ with tuples
|
|
assert v3_10 == (3, 10)
|
|
assert v3_10 != (3, 9)
|
|
assert v3_9 == (3, 9)
|
|
|
|
# Test __ge__ with tuples
|
|
assert v3_10 >= (3, 10)
|
|
assert v3_10 >= (3, 9)
|
|
assert not (v3_9 >= (3, 10))
|
|
assert v3_11 >= (3, 10)
|
|
|
|
# Test __eq__ with other _version_info instances
|
|
assert v3_10 == _version_info(3, 10)
|
|
assert v3_10 != v3_9
|
|
assert v3_10 == v3_10 # Same instance
|
|
|
|
assert v3_10 != v3_11
|
|
|
|
# Test __ge__ with other _version_info instances
|
|
assert v3_10 >= v3_10
|
|
assert v3_10 >= v3_9
|
|
assert not (v3_9 >= v3_10)
|
|
assert v3_11 >= v3_10
|
|
|
|
assert v3_11 >= v3_11 # Same instance
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_check_dashboard_files_not_exists(monkeypatch):
|
|
"""Tests dashboard download when files do not exist."""
|
|
monkeypatch.setattr(os.path, "exists", lambda x: False)
|
|
|
|
with mock.patch("main.download_dashboard") as mock_download:
|
|
await check_dashboard_files()
|
|
mock_download.assert_called_once()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_check_dashboard_files_exists_and_version_match(monkeypatch):
|
|
"""Tests that dashboard is not downloaded when it exists and version matches."""
|
|
# Mock os.path.exists to return True
|
|
monkeypatch.setattr(os.path, "exists", lambda x: True)
|
|
|
|
# Mock get_dashboard_version to return the current version
|
|
with mock.patch("main.get_dashboard_version") as mock_get_version:
|
|
# We need to import VERSION from main's context
|
|
from main import VERSION
|
|
|
|
mock_get_version.return_value = f"v{VERSION}"
|
|
|
|
with mock.patch("main.download_dashboard") as mock_download:
|
|
await check_dashboard_files()
|
|
# Assert that download_dashboard was NOT called
|
|
mock_download.assert_not_called()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_check_dashboard_files_exists_but_version_mismatch(monkeypatch):
|
|
"""Tests that a warning is logged when dashboard version mismatches."""
|
|
monkeypatch.setattr(os.path, "exists", lambda x: True)
|
|
|
|
with mock.patch("main.get_dashboard_version") as mock_get_version:
|
|
mock_get_version.return_value = "v0.0.1" # A different version
|
|
|
|
with mock.patch("main.logger.warning") as mock_logger_warning:
|
|
await check_dashboard_files()
|
|
mock_logger_warning.assert_called_once()
|
|
call_args, _ = mock_logger_warning.call_args
|
|
assert "WebUI version mismatch" in call_args[0]
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_check_dashboard_files_with_webui_dir_arg(monkeypatch):
|
|
"""Tests that providing a valid webui_dir skips all checks."""
|
|
valid_dir = "/tmp/my-custom-webui"
|
|
monkeypatch.setattr(os.path, "exists", lambda path: path == valid_dir)
|
|
|
|
with mock.patch("main.download_dashboard") as mock_download:
|
|
with mock.patch("main.get_dashboard_version") as mock_get_version:
|
|
result = await check_dashboard_files(webui_dir=valid_dir)
|
|
assert result == valid_dir
|
|
mock_download.assert_not_called()
|
|
mock_get_version.assert_not_called()
|