mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-08-29 03:56:43 +08:00
689 lines
27 KiB
Python
689 lines
27 KiB
Python
import asyncio
|
|
from contextlib import nullcontext
|
|
from unittest.mock import ANY, AsyncMock, MagicMock, patch
|
|
|
|
from app.api.endpoints import plugin as plugin_endpoint
|
|
from app import schemas
|
|
from app.api.endpoints.plugin import plugin_history
|
|
from app.api.endpoints.plugin import plugin_releases
|
|
from app.api.endpoints.plugin import reset_plugin
|
|
from app.api.endpoints.plugin import reload_plugin
|
|
from app.api.endpoints.plugin import runtime_status
|
|
from app.api.endpoints.plugin import plugin_static_file
|
|
from app.api.endpoints.plugin import uninstall_plugin
|
|
from app.api.endpoints.system import sync_plugin_market_from_wiki
|
|
from app.application.plugin.config import PluginConfigCommand
|
|
from app.runtime.config import settings
|
|
from app.runtime.extensions.plugin.admission import PluginMutationAdmission
|
|
from app.runtime.extensions.plugin_manager import PluginManager
|
|
from app.runtime.tasks import TaskRegistry
|
|
from app.schemas.event import PluginDataResetEventData
|
|
from app.schemas.plugin import PluginInstance, PluginRuntimeStatus
|
|
from app.schemas.types import ChainEventType, SystemConfigKey
|
|
from app.foundation.singleton import Singleton
|
|
|
|
|
|
def test_plugin_history_merges_remote_metadata():
|
|
"""
|
|
已安装插件点击更新说明时，接口会按需合并远端仓库中的更新记录。
|
|
"""
|
|
installed_plugin = schemas.Plugin(
|
|
id="DemoPlugin",
|
|
plugin_name="Demo Plugin",
|
|
plugin_version="1.0.0",
|
|
installed=True,
|
|
history={},
|
|
)
|
|
market_plugin = schemas.Plugin(
|
|
id="DemoPlugin",
|
|
repo_url="https://github.com/demo/plugins",
|
|
history={"v1.1.0": "- 新增更新说明"},
|
|
system_version=">=2.0.0",
|
|
system_version_compatible=True,
|
|
has_update=True,
|
|
)
|
|
plugin_manager = MagicMock()
|
|
plugin_manager.get_local_plugins.return_value = [installed_plugin]
|
|
plugin_manager.get_local_repo_plugins.return_value = []
|
|
plugin_manager.async_get_online_plugins = AsyncMock(return_value=[market_plugin])
|
|
|
|
with patch("app.api.endpoints.plugin.PluginManager", return_value=plugin_manager):
|
|
result = asyncio.run(plugin_history("DemoPlugin", None, True))
|
|
|
|
assert result.repo_url == "https://github.com/demo/plugins"
|
|
assert result.history == {"v1.1.0": "- 新增更新说明"}
|
|
assert result.system_version == ">=2.0.0"
|
|
assert result.has_update
|
|
|
|
|
|
def test_runtime_status_reports_pending_and_terminal_counts():
|
|
"""插件页摘要区分后台收敛、准备态和终态失败。"""
|
|
plugin_manager = MagicMock()
|
|
plugin_manager.get_plugin_runtime_statuses.return_value = {
|
|
"SourcePending": PluginRuntimeStatus.SOURCE_MISSING,
|
|
"DependencyPending": PluginRuntimeStatus.DEPENDENCY_PENDING,
|
|
"ActivePlugin": PluginRuntimeStatus.ACTIVE,
|
|
"FailedPlugin": PluginRuntimeStatus.LOAD_FAILED,
|
|
}
|
|
plugin_manager.is_plugin_settling.return_value = True
|
|
plugin_manager.get_plugin_runtime_generation.return_value = 7
|
|
|
|
with patch("app.api.endpoints.plugin.PluginManager", return_value=plugin_manager):
|
|
result = asyncio.run(runtime_status(None))
|
|
|
|
assert result.ready is False
|
|
assert result.generation == 7
|
|
assert result.pending_count == 2
|
|
assert result.failed_count == 1
|
|
|
|
|
|
def test_reload_endpoint_reports_load_failure(monkeypatch):
|
|
"""插件重载失败时接口返回失败，同时仍刷新旧注册投影。"""
|
|
plugin_manager = MagicMock()
|
|
plugin_manager.reload_plugin.return_value = PluginRuntimeStatus.LOAD_FAILED
|
|
register = MagicMock()
|
|
monkeypatch.setattr(plugin_endpoint, "PluginManager", lambda: plugin_manager)
|
|
monkeypatch.setattr(plugin_endpoint, "register_plugin", register)
|
|
|
|
result = reload_plugin("DemoPlugin", None)
|
|
|
|
assert result.success is False
|
|
assert result.message == "插件加载失败，请查看插件日志"
|
|
register.assert_called_once_with("DemoPlugin")
|
|
|
|
|
|
def test_plugin_history_returns_installed_plugin_when_remote_missing():
|
|
"""
|
|
远端仓库不可用时，接口仍返回本地已安装插件信息，前端可继续展示兜底状态。
|
|
"""
|
|
installed_plugin = schemas.Plugin(
|
|
id="DemoPlugin",
|
|
plugin_name="Demo Plugin",
|
|
plugin_version="1.0.0",
|
|
installed=True,
|
|
)
|
|
plugin_manager = MagicMock()
|
|
plugin_manager.get_local_plugins.return_value = [installed_plugin]
|
|
plugin_manager.get_local_repo_plugins.return_value = []
|
|
plugin_manager.async_get_online_plugins = AsyncMock(return_value=[])
|
|
|
|
with patch("app.api.endpoints.plugin.PluginManager", return_value=plugin_manager):
|
|
result = asyncio.run(plugin_history("DemoPlugin", None, True))
|
|
|
|
assert result.id == "DemoPlugin"
|
|
assert result.history == {}
|
|
|
|
|
|
def test_plugin_history_uses_installed_repo_without_refreshing_all_markets():
|
|
"""
|
|
已安装插件记录了来源仓库时，更新说明只刷新该仓库，避免弹窗触发全市场慢刷新。
|
|
"""
|
|
installed_plugin = schemas.Plugin(
|
|
id="DemoPlugin",
|
|
plugin_name="Demo Plugin",
|
|
plugin_version="1.0.0",
|
|
repo_url="https://github.com/demo/plugins",
|
|
installed=True,
|
|
)
|
|
market_plugin = schemas.Plugin(
|
|
id="DemoPlugin",
|
|
repo_url="https://github.com/demo/plugins",
|
|
history={"v1.1.0": "- 新增更新说明"},
|
|
)
|
|
plugin_manager = MagicMock()
|
|
plugin_manager.get_local_plugins.return_value = [installed_plugin]
|
|
plugin_manager.get_local_repo_plugins.return_value = []
|
|
plugin_manager.async_get_plugins_from_market = AsyncMock(return_value=[market_plugin])
|
|
plugin_manager.async_get_online_plugins = AsyncMock(return_value=[])
|
|
|
|
with patch("app.api.endpoints.plugin.PluginManager", return_value=plugin_manager):
|
|
result = asyncio.run(plugin_history("DemoPlugin", None, True))
|
|
|
|
assert result.history == {"v1.1.0": "- 新增更新说明"}
|
|
plugin_manager.async_get_plugins_from_market.assert_awaited_once_with(
|
|
"https://github.com/demo/plugins", settings.VERSION_FLAG, True
|
|
)
|
|
plugin_manager.async_get_online_plugins.assert_not_awaited()
|
|
|
|
|
|
def test_plugin_releases_returns_supported_versions_with_latest_and_current(monkeypatch):
|
|
"""
|
|
release 列表接口返回可安装版本，并标记当前 package 最新版本与本地已安装版本。
|
|
"""
|
|
market_plugin = schemas.Plugin(
|
|
id="DemoPlugin",
|
|
plugin_version="1.2.3",
|
|
repo_url="https://github.com/demo/plugins",
|
|
release=True,
|
|
)
|
|
plugin_manager = MagicMock()
|
|
plugin_manager.async_get_online_plugins = AsyncMock(return_value=[market_plugin])
|
|
plugin_manager.async_get_plugins_from_market = AsyncMock(return_value=[market_plugin])
|
|
plugin_manager.get_local_plugin_version.return_value = "1.2.0"
|
|
plugin_helper = MagicMock()
|
|
plugin_helper.async_get_plugin_release_versions = AsyncMock(return_value=[
|
|
{"version": "1.2.3", "tag_name": "DemoPlugin_v1.2.3", "asset_name": "demoplugin_v1.2.3.zip"},
|
|
{"version": "1.2.0", "tag_name": "DemoPlugin_v1.2.0", "asset_name": "demoplugin_v1.2.0.zip"},
|
|
])
|
|
|
|
with (
|
|
patch("app.api.endpoints.plugin.PluginManager", return_value=plugin_manager),
|
|
patch("app.api.endpoints.plugin.PluginHelper", return_value=plugin_helper),
|
|
):
|
|
result = asyncio.run(plugin_releases("DemoPlugin", None, "https://github.com/demo/plugins", False))
|
|
|
|
assert result["release_supported"] is True
|
|
assert result["latest_version"] == "1.2.3"
|
|
assert result["current_version"] == "1.2.0"
|
|
assert result["items"][0]["is_latest"] is True
|
|
assert result["items"][0]["is_current"] is False
|
|
assert result["items"][1]["is_latest"] is False
|
|
assert result["items"][1]["is_current"] is True
|
|
plugin_manager.async_get_plugins_from_market.assert_awaited_once_with(
|
|
"https://github.com/demo/plugins", settings.VERSION_FLAG, False
|
|
)
|
|
plugin_manager.async_get_online_plugins.assert_not_awaited()
|
|
plugin_manager.get_local_plugins.assert_not_called()
|
|
|
|
|
|
def test_plugin_releases_does_not_mutate_cached_release_items(monkeypatch):
|
|
"""
|
|
接口标记当前/最新版本时不能修改 helper 返回对象，避免污染缓存中的 release 列表。
|
|
"""
|
|
market_plugin = schemas.Plugin(
|
|
id="DemoPlugin",
|
|
plugin_version="1.2.3",
|
|
repo_url="https://github.com/demo/plugins",
|
|
release=True,
|
|
)
|
|
release_items = [
|
|
{"version": "1.2.3", "tag_name": "DemoPlugin_v1.2.3", "asset_name": "demoplugin_v1.2.3.zip"},
|
|
]
|
|
plugin_manager = MagicMock()
|
|
plugin_manager.async_get_plugins_from_market = AsyncMock(return_value=[market_plugin])
|
|
plugin_manager.get_local_plugin_version.return_value = "1.2.0"
|
|
plugin_helper = MagicMock()
|
|
plugin_helper.async_get_plugin_release_versions = AsyncMock(return_value=release_items)
|
|
|
|
with (
|
|
patch("app.api.endpoints.plugin.PluginManager", return_value=plugin_manager),
|
|
patch("app.api.endpoints.plugin.PluginHelper", return_value=plugin_helper),
|
|
):
|
|
result = asyncio.run(plugin_releases("DemoPlugin", None, "https://github.com/demo/plugins", False))
|
|
|
|
assert result["items"][0]["is_latest"] is True
|
|
assert "is_latest" not in release_items[0]
|
|
assert "is_current" not in release_items[0]
|
|
|
|
|
|
def test_plugin_releases_falls_back_to_compatible_base_package(monkeypatch):
|
|
"""
|
|
当前版本 package 未包含插件时，再读取基础 package 兼容项，不扫描其他市场。
|
|
"""
|
|
market_plugin = schemas.Plugin(
|
|
id="DemoPlugin",
|
|
plugin_version="1.2.3",
|
|
repo_url="https://github.com/demo/plugins",
|
|
release=True,
|
|
)
|
|
plugin_manager = MagicMock()
|
|
plugin_manager.async_get_plugins_from_market = AsyncMock(
|
|
side_effect=[[], [market_plugin]]
|
|
)
|
|
plugin_manager.get_local_plugin_version.return_value = None
|
|
plugin_helper = MagicMock()
|
|
plugin_helper.async_has_plugin_release_cache = AsyncMock(return_value=False)
|
|
plugin_helper.async_get_plugin_release_versions = AsyncMock(return_value=[])
|
|
|
|
with (
|
|
patch("app.api.endpoints.plugin.PluginManager", return_value=plugin_manager),
|
|
patch("app.api.endpoints.plugin.PluginHelper", return_value=plugin_helper),
|
|
):
|
|
result = asyncio.run(
|
|
plugin_releases("DemoPlugin", None, "https://github.com/demo/plugins", False)
|
|
)
|
|
|
|
assert result["latest_version"] == "1.2.3"
|
|
assert plugin_manager.async_get_plugins_from_market.await_args_list == [
|
|
(("https://github.com/demo/plugins", settings.VERSION_FLAG, False), {}),
|
|
(("https://github.com/demo/plugins", None, False), {}),
|
|
]
|
|
|
|
|
|
def test_plugin_releases_uses_force_refresh_for_market_metadata(monkeypatch):
|
|
"""
|
|
release 列表接口沿用插件市场的 force 语义，供前端手动刷新时绕过缓存。
|
|
"""
|
|
market_plugin = schemas.Plugin(
|
|
id="DemoPlugin",
|
|
plugin_version="1.2.3",
|
|
repo_url="https://github.com/demo/plugins",
|
|
release=True,
|
|
)
|
|
plugin_manager = MagicMock()
|
|
plugin_manager.async_get_plugins_from_market = AsyncMock(return_value=[market_plugin])
|
|
plugin_manager.get_local_plugin_version.return_value = None
|
|
plugin_helper = MagicMock()
|
|
plugin_helper.async_has_plugin_release_cache = AsyncMock(return_value=False)
|
|
plugin_helper.async_get_plugin_release_versions = AsyncMock(return_value=[])
|
|
|
|
with (
|
|
patch("app.api.endpoints.plugin.PluginManager", return_value=plugin_manager),
|
|
patch("app.api.endpoints.plugin.PluginHelper", return_value=plugin_helper),
|
|
):
|
|
result = asyncio.run(plugin_releases("DemoPlugin", None, "https://github.com/demo/plugins", True))
|
|
|
|
assert result["release_supported"] is False
|
|
plugin_manager.async_get_plugins_from_market.assert_awaited_once_with(
|
|
"https://github.com/demo/plugins", settings.VERSION_FLAG, True
|
|
)
|
|
assert plugin_helper.async_get_plugin_release_versions.await_args.args == (
|
|
"DemoPlugin",
|
|
"https://github.com/demo/plugins",
|
|
)
|
|
|
|
|
|
def test_plugin_releases_force_uses_cached_release_response_and_schedules_refresh(monkeypatch):
|
|
"""
|
|
手动刷新时 package 元数据仍强刷，但 Release 明细先读缓存并后台刷新，避免弹窗阻塞。
|
|
"""
|
|
from app.runtime.cache import is_fresh
|
|
|
|
market_plugin = schemas.Plugin(
|
|
id="DemoPlugin",
|
|
plugin_version="1.2.3",
|
|
repo_url="https://github.com/demo/plugins",
|
|
release=True,
|
|
)
|
|
plugin_manager = MagicMock()
|
|
plugin_manager.async_get_plugins_from_market = AsyncMock(return_value=[market_plugin])
|
|
plugin_manager.get_local_plugin_version.return_value = None
|
|
fresh_states = []
|
|
plugin_helper = MagicMock()
|
|
plugin_helper.async_has_plugin_release_cache = AsyncMock(return_value=True)
|
|
|
|
async def fake_releases(*_args):
|
|
fresh_states.append(is_fresh())
|
|
return [
|
|
{
|
|
"version": "1.2.3",
|
|
"tag_name": "DemoPlugin_v1.2.3",
|
|
"asset_name": "demoplugin_v1.2.3.zip",
|
|
}
|
|
]
|
|
|
|
plugin_helper.async_get_plugin_release_versions = fake_releases
|
|
scheduled = []
|
|
|
|
def fake_schedule(plugin_id, repo_url, task_registry):
|
|
scheduled.append((plugin_id, repo_url, task_registry))
|
|
|
|
with (
|
|
patch("app.api.endpoints.plugin.PluginManager", return_value=plugin_manager),
|
|
patch("app.api.endpoints.plugin.PluginHelper", return_value=plugin_helper),
|
|
patch.object(plugin_endpoint, "_schedule_plugin_release_refresh", fake_schedule),
|
|
):
|
|
result = asyncio.run(plugin_releases("DemoPlugin", None, "https://github.com/demo/plugins", True))
|
|
|
|
assert result["release_supported"] is True
|
|
assert fresh_states == [False]
|
|
assert len(scheduled) == 1
|
|
assert scheduled[0][:2] == (
|
|
"DemoPlugin",
|
|
"https://github.com/demo/plugins",
|
|
)
|
|
assert isinstance(scheduled[0][2], TaskRegistry)
|
|
plugin_helper.async_has_plugin_release_cache.assert_awaited_once_with(
|
|
"https://github.com/demo/plugins"
|
|
)
|
|
plugin_manager.async_get_plugins_from_market.assert_awaited_once_with(
|
|
"https://github.com/demo/plugins", settings.VERSION_FLAG, True
|
|
)
|
|
|
|
|
|
def test_plugin_releases_force_skips_background_refresh_without_release_cache(monkeypatch):
|
|
"""
|
|
冷缓存 force 请求已在响应路径读取 Release，不能马上再启动一次重复强刷。
|
|
"""
|
|
market_plugin = schemas.Plugin(
|
|
id="DemoPlugin",
|
|
plugin_version="1.2.3",
|
|
repo_url="https://github.com/demo/plugins",
|
|
release=True,
|
|
)
|
|
plugin_manager = MagicMock()
|
|
plugin_manager.async_get_plugins_from_market = AsyncMock(return_value=[market_plugin])
|
|
plugin_manager.get_local_plugin_version.return_value = None
|
|
plugin_helper = MagicMock()
|
|
plugin_helper.async_has_plugin_release_cache = AsyncMock(return_value=False)
|
|
plugin_helper.async_get_plugin_release_versions = AsyncMock(return_value=[
|
|
{
|
|
"version": "1.2.3",
|
|
"tag_name": "DemoPlugin_v1.2.3",
|
|
"asset_name": "demoplugin_v1.2.3.zip",
|
|
}
|
|
])
|
|
scheduled = []
|
|
|
|
def fake_schedule(plugin_id, repo_url):
|
|
scheduled.append((plugin_id, repo_url))
|
|
|
|
with (
|
|
patch("app.api.endpoints.plugin.PluginManager", return_value=plugin_manager),
|
|
patch("app.api.endpoints.plugin.PluginHelper", return_value=plugin_helper),
|
|
patch.object(plugin_endpoint, "_schedule_plugin_release_refresh", fake_schedule),
|
|
):
|
|
result = asyncio.run(plugin_releases("DemoPlugin", None, "https://github.com/demo/plugins", True))
|
|
|
|
assert result["release_supported"] is True
|
|
assert scheduled == []
|
|
plugin_helper.async_has_plugin_release_cache.assert_awaited_once_with(
|
|
"https://github.com/demo/plugins"
|
|
)
|
|
|
|
|
|
def test_plugin_releases_hides_items_when_market_plugin_does_not_enable_release(monkeypatch):
|
|
"""
|
|
接口是否支持 Release 安装要与当前 package 的 release 声明保持一致。
|
|
"""
|
|
market_plugin = schemas.Plugin(
|
|
id="DemoPlugin",
|
|
plugin_version="1.2.3",
|
|
repo_url="https://github.com/demo/plugins",
|
|
release=False,
|
|
)
|
|
plugin_manager = MagicMock()
|
|
plugin_manager.async_get_plugins_from_market = AsyncMock(return_value=[market_plugin])
|
|
plugin_manager.get_local_plugin_version.return_value = None
|
|
plugin_helper = MagicMock()
|
|
plugin_helper.async_get_plugin_release_versions = AsyncMock(return_value=[
|
|
{"version": "1.2.3", "tag_name": "DemoPlugin_v1.2.3", "asset_name": "demoplugin_v1.2.3.zip"},
|
|
])
|
|
|
|
with (
|
|
patch("app.api.endpoints.plugin.PluginManager", return_value=plugin_manager),
|
|
patch("app.api.endpoints.plugin.PluginHelper", return_value=plugin_helper),
|
|
):
|
|
result = asyncio.run(plugin_releases("DemoPlugin", None, "https://github.com/demo/plugins", False))
|
|
|
|
assert result["release_supported"] is False
|
|
assert result["items"] == []
|
|
plugin_helper.async_get_plugin_release_versions.assert_not_awaited()
|
|
|
|
|
|
def test_sync_plugin_market_from_wiki_merges_and_deduplicates_repos():
|
|
"""
|
|
Wiki 同步会提取标记区域内的 GitHub 仓库地址，并与本地配置合并去重后写入。
|
|
"""
|
|
markdown = """
|
|
<!-- plugin-market-repos:start -->
|
|
- https://github.com/local/existing/
|
|
- https://github.com/wiki/new-repo/
|
|
- https://github.com/wiki/new-repo
|
|
<!-- plugin-market-repos:end -->
|
|
- https://github.com/wiki/ignored-outside-marker
|
|
"""
|
|
response = MagicMock(status_code=200, text=markdown)
|
|
request_utils = MagicMock()
|
|
request_utils.get_res = AsyncMock(return_value=response)
|
|
runtime_settings = MagicMock()
|
|
runtime_settings.get.return_value = "https://github.com/local/existing"
|
|
runtime_settings.update.return_value = (True, "")
|
|
with (
|
|
patch("app.api.endpoints.system.AsyncRequestUtils", return_value=request_utils),
|
|
patch(
|
|
"app.api.endpoints.system.get_runtime_settings",
|
|
return_value=runtime_settings,
|
|
),
|
|
patch("app.api.endpoints.system.eventmanager.async_send_event", new=AsyncMock()) as send_event,
|
|
):
|
|
result = asyncio.run(sync_plugin_market_from_wiki(None, None))
|
|
|
|
assert result.success
|
|
assert result.data["repos"] == [
|
|
"https://github.com/local/existing",
|
|
"https://github.com/wiki/new-repo",
|
|
]
|
|
assert result.data["added_count"] == 1
|
|
assert result.data["total_count"] == 2
|
|
runtime_settings.update.assert_called_once_with(
|
|
"PLUGIN_MARKET",
|
|
"https://github.com/local/existing,https://github.com/wiki/new-repo",
|
|
)
|
|
send_event.assert_awaited_once()
|
|
|
|
|
|
def test_reset_plugin_sends_pre_reset_chain_event_before_deleting_data():
|
|
"""
|
|
插件重置会先触发同步链式事件，让插件在数据被清空前完成自有状态补偿。
|
|
"""
|
|
plugin_manager = MagicMock()
|
|
calls = []
|
|
|
|
def delete_config(plugin_id, force=False):
|
|
calls.append(("delete_config", plugin_id, force))
|
|
return True
|
|
|
|
def delete_data(plugin_id, force=False):
|
|
calls.append(("delete_data", plugin_id, force))
|
|
return True
|
|
|
|
def stop_plugin(plugin_id):
|
|
calls.append(("stop", plugin_id))
|
|
return True
|
|
|
|
plugin_manager.stop.side_effect = stop_plugin
|
|
plugin_manager.delete_plugin_config.side_effect = delete_config
|
|
plugin_manager.delete_plugin_data.side_effect = delete_data
|
|
|
|
def publish_reset(plugin_id):
|
|
"""记录重置前事件，验证应用用例保留补偿时序。"""
|
|
calls.append((
|
|
"event",
|
|
ChainEventType.PluginDataReset,
|
|
PluginDataResetEventData(
|
|
plugin_id=plugin_id,
|
|
reset_config=True,
|
|
reset_data=True,
|
|
),
|
|
))
|
|
|
|
command = PluginConfigCommand(
|
|
save_config=plugin_manager.save_plugin_config,
|
|
initialize=plugin_manager.init_plugin,
|
|
stop=plugin_manager.stop,
|
|
delete_config=plugin_manager.delete_plugin_config,
|
|
delete_data=plugin_manager.delete_plugin_data,
|
|
reload_runtime=plugin_manager.reload_plugin,
|
|
publish_reset=publish_reset,
|
|
refresh_registrations=lambda _plugin_id: None,
|
|
mutation=lambda _operation: nullcontext(),
|
|
)
|
|
result = reset_plugin("SubscribeAssistantEnhanced", None, command)
|
|
|
|
assert result.success is True
|
|
assert len(calls) == 4
|
|
event_call = calls[0]
|
|
assert event_call[0] == "event"
|
|
assert event_call[1] is ChainEventType.PluginDataReset
|
|
assert isinstance(event_call[2], PluginDataResetEventData)
|
|
assert event_call[2].plugin_id == "SubscribeAssistantEnhanced"
|
|
assert event_call[2].reset_config is True
|
|
assert event_call[2].reset_data is True
|
|
assert calls[1:] == [
|
|
("stop", "SubscribeAssistantEnhanced"),
|
|
("delete_config", "SubscribeAssistantEnhanced", True),
|
|
("delete_data", "SubscribeAssistantEnhanced", True),
|
|
]
|
|
plugin_manager.reload_plugin.assert_called_once_with("SubscribeAssistantEnhanced")
|
|
|
|
|
|
def test_delete_plugin_config_can_force_delete_after_plugin_is_stopped():
|
|
"""
|
|
重置入口会先停止插件；配置删除需要能处理运行态注册已清理的插件 ID。
|
|
"""
|
|
Singleton._instances.pop((PluginManager, (), frozenset()), None)
|
|
manager = PluginManager()
|
|
|
|
storage = MagicMock()
|
|
storage.delete.return_value = True
|
|
with patch("app.runtime.extensions.plugin_manager.get_plugin_storage", return_value=storage):
|
|
assert manager.delete_plugin_config("DemoPlugin", force=True) is True
|
|
storage.delete.assert_called_once_with("plugin.DemoPlugin")
|
|
Singleton._instances.pop((PluginManager, (), frozenset()), None)
|
|
|
|
|
|
def test_virtual_instance_static_file_reads_from_source_directory(tmp_path, monkeypatch):
|
|
"""实例 URL 保持独立，但静态内容直接读取共享的源插件目录。"""
|
|
source_file = tmp_path / "app/plugins/demoplugin/dist/remoteEntry.js"
|
|
source_file.parent.mkdir(parents=True)
|
|
source_file.write_text("export default 'shared'", encoding="utf-8")
|
|
plugin_manager = MagicMock()
|
|
plugin_manager.get_plugin_source_id.return_value = "DemoPlugin"
|
|
monkeypatch.setattr(plugin_endpoint, "PluginManager", lambda: plugin_manager)
|
|
monkeypatch.setattr(
|
|
plugin_endpoint,
|
|
"get_api_runtime_config_snapshot",
|
|
lambda: MagicMock(root_path=tmp_path),
|
|
)
|
|
|
|
response = asyncio.run(
|
|
plugin_static_file("DemoPluginwork", "dist/remoteEntry.js", None)
|
|
)
|
|
|
|
async def read_body() -> bytes:
|
|
"""读取流式响应的全部测试内容。"""
|
|
return b"".join([chunk async for chunk in response.body_iterator])
|
|
|
|
assert asyncio.run(read_body()) == b"export default 'shared'"
|
|
assert response.media_type == "application/javascript"
|
|
plugin_manager.get_plugin_source_id.assert_called_once_with("DemoPluginwork")
|
|
|
|
|
|
def test_uninstall_virtual_instance_never_removes_source_package(monkeypatch):
|
|
"""卸载虚拟实例只清理实例状态，不触碰源插件安装清单或目录。"""
|
|
plugin_manager = MagicMock()
|
|
plugin_manager.get_plugin_instance.return_value = PluginInstance(
|
|
instance_id="DemoPluginwork",
|
|
source_plugin_id="DemoPlugin",
|
|
)
|
|
plugin_manager.get_plugin_source_instances.return_value = []
|
|
config = MagicMock()
|
|
config.get.return_value = ["DemoPlugin"]
|
|
monkeypatch.setattr(plugin_endpoint, "PluginManager", lambda: plugin_manager)
|
|
monkeypatch.setattr(plugin_endpoint, "get_configured_system_config", lambda: config)
|
|
monkeypatch.setattr(plugin_endpoint, "remove_plugin_api", MagicMock())
|
|
monkeypatch.setattr(plugin_endpoint, "remove_plugin_job", MagicMock())
|
|
monkeypatch.setattr(plugin_endpoint, "remove_plugin_from_folders", MagicMock())
|
|
|
|
result = uninstall_plugin("DemoPluginwork", None)
|
|
|
|
assert result.success is True
|
|
config.set.assert_called_once_with(
|
|
SystemConfigKey.UserInstalledPlugins,
|
|
["DemoPlugin"],
|
|
)
|
|
plugin_manager.delete_plugin_config.assert_called_once_with(
|
|
"DemoPluginwork",
|
|
force=True,
|
|
)
|
|
plugin_manager.delete_plugin_data.assert_called_once_with(
|
|
"DemoPluginwork",
|
|
force=True,
|
|
)
|
|
plugin_manager.delete_plugin_instance.assert_called_once_with("DemoPluginwork")
|
|
plugin_manager.remove_plugin.assert_called_once_with("DemoPluginwork")
|
|
|
|
|
|
def test_sealed_http_uninstall_rejects_before_first_side_effect(monkeypatch):
|
|
"""HTTP 卸载在封口后明确失败，且不读取或写入插件持久化状态。"""
|
|
admission = PluginMutationAdmission()
|
|
admission.seal()
|
|
plugin_manager = MagicMock()
|
|
plugin_manager.mutation.side_effect = admission.hold
|
|
config_provider = MagicMock()
|
|
remove_api = MagicMock()
|
|
remove_job = MagicMock()
|
|
monkeypatch.setattr(plugin_endpoint, "PluginManager", lambda: plugin_manager)
|
|
monkeypatch.setattr(
|
|
plugin_endpoint,
|
|
"get_configured_system_config",
|
|
config_provider,
|
|
)
|
|
monkeypatch.setattr(plugin_endpoint, "remove_plugin_api", remove_api)
|
|
monkeypatch.setattr(plugin_endpoint, "remove_plugin_job", remove_job)
|
|
|
|
result = uninstall_plugin("DemoPlugin", None)
|
|
|
|
assert result.success is False
|
|
assert "停机阶段" in result.message
|
|
plugin_manager.get_plugin_instance.assert_not_called()
|
|
config_provider.assert_not_called()
|
|
remove_api.assert_not_called()
|
|
remove_job.assert_not_called()
|
|
|
|
|
|
def test_sealed_http_clone_rejects_before_runtime_and_registration(monkeypatch):
|
|
"""HTTP 分身事务未获 admission 时不创建实例、不刷新注册或文件夹。"""
|
|
admission = PluginMutationAdmission()
|
|
admission.seal()
|
|
plugin_manager = MagicMock()
|
|
plugin_manager.mutation.side_effect = admission.hold
|
|
register = MagicMock()
|
|
add_to_folder = MagicMock()
|
|
monkeypatch.setattr(plugin_endpoint, "PluginManager", lambda: plugin_manager)
|
|
monkeypatch.setattr(plugin_endpoint, "register_plugin", register)
|
|
monkeypatch.setattr(plugin_endpoint, "_add_clone_to_plugin_folder", add_to_folder)
|
|
|
|
result = plugin_endpoint.clone_plugin(
|
|
"DemoPlugin",
|
|
schemas.PluginCloneRequest(suffix="Work"),
|
|
None,
|
|
)
|
|
|
|
assert result.success is False
|
|
assert "停机阶段" in result.message
|
|
plugin_manager.clone_plugin.assert_not_called()
|
|
register.assert_not_called()
|
|
add_to_folder.assert_not_called()
|
|
|
|
|
|
def test_sealed_http_folder_update_rejects_before_config_access(monkeypatch):
|
|
"""插件文件夹写入口在封口后不读取或改写持久化配置。"""
|
|
admission = PluginMutationAdmission()
|
|
admission.seal()
|
|
plugin_manager = MagicMock()
|
|
plugin_manager.mutation.side_effect = admission.hold
|
|
config_provider = MagicMock()
|
|
monkeypatch.setattr(plugin_endpoint, "PluginManager", lambda: plugin_manager)
|
|
monkeypatch.setattr(
|
|
plugin_endpoint,
|
|
"get_configured_system_config",
|
|
config_provider,
|
|
)
|
|
|
|
result = asyncio.run(
|
|
plugin_endpoint.update_folder_plugins("常用", ["DemoPlugin"], None)
|
|
)
|
|
|
|
assert result.success is False
|
|
assert "停机阶段" in result.message
|
|
config_provider.assert_not_called()
|
|
|
|
|
|
def test_delete_plugin_data_can_force_delete_after_plugin_is_stopped():
|
|
"""
|
|
重置入口会先停止插件；插件数据删除不能依赖运行态注册仍存在。
|
|
"""
|
|
Singleton._instances.pop((PluginManager, (), frozenset()), None)
|
|
manager = PluginManager()
|
|
calls = []
|
|
|
|
storage = MagicMock()
|
|
storage.delete_data.side_effect = lambda pid: calls.append(pid)
|
|
with patch("app.runtime.extensions.plugin_manager.get_plugin_storage", return_value=storage):
|
|
assert manager.delete_plugin_data("DemoPlugin", force=True) is True
|
|
|
|
assert calls == ["DemoPlugin"]
|
|
Singleton._instances.pop((PluginManager, (), frozenset()), None)
|