mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-08-29 03:56:43 +08:00
240 lines
8.0 KiB
Python
240 lines
8.0 KiB
Python
"""插件市场同步服务用例。"""
|
|
|
|
import asyncio
|
|
from datetime import datetime, timezone
|
|
from types import SimpleNamespace
|
|
from unittest.mock import AsyncMock, Mock
|
|
|
|
import pytest
|
|
|
|
from app.application.plugin.declaration import PluginDeclaredMetadata
|
|
from app.application.plugin.gateway import PluginInstallGateway
|
|
from app.application.plugin.identity import (
|
|
PluginBindingBasis,
|
|
PluginIdentity,
|
|
PluginPayloadSourceType,
|
|
TrustedPluginSourceType,
|
|
)
|
|
from app.application.plugin.install import PluginInstallResult
|
|
from app.application.plugin.lifecycle import plugin_lifecycle
|
|
from app.application.plugin.source import (
|
|
CandidateInventory,
|
|
MarketRead,
|
|
PluginMarketCandidate,
|
|
)
|
|
from app.runtime.config import global_vars
|
|
from app.runtime.extensions.plugin.sync import PluginSyncService
|
|
from app.startup.initializers import plugins as plugins_initializer
|
|
|
|
REPO_URL = "https://github.com/jxxghp/MoviePilot-Plugins"
|
|
|
|
|
|
def test_market_sync_keeps_install_rollback_enabled() -> None:
|
|
"""自动更新插件时保留旧版本，失败后可由安装器恢复。"""
|
|
plugin = SimpleNamespace(
|
|
id="DemoPlugin",
|
|
repo_url="https://example.com/plugins",
|
|
plugin_name="Demo",
|
|
plugin_version="1.0.0",
|
|
system_version_compatible=True,
|
|
)
|
|
install = Mock(return_value=(True, ""))
|
|
service = PluginSyncService(
|
|
frozen=lambda: False,
|
|
installed_plugins=lambda: [plugin.id],
|
|
online_plugins=lambda: [plugin],
|
|
local_plugins=lambda: [],
|
|
merge_plugins=lambda items, *_args: items,
|
|
plugin_exists=lambda *_args: False,
|
|
install=install,
|
|
log=Mock(),
|
|
)
|
|
|
|
assert service.sync() == [plugin.id]
|
|
install.assert_called_once_with(plugin.id, None, False, None)
|
|
|
|
|
|
def test_market_sync_restores_trusted_online_payload_after_local_source_removed() -> None:
|
|
"""本地高版本来源消失后，启动同步仍恢复已绑定的在线载荷。"""
|
|
plugin = SimpleNamespace(
|
|
id="DemoPlugin",
|
|
repo_url=REPO_URL,
|
|
plugin_name="Demo",
|
|
plugin_version="1.2.0",
|
|
system_version_compatible=False,
|
|
)
|
|
install = Mock(return_value=(True, ""))
|
|
service = PluginSyncService(
|
|
frozen=lambda: False,
|
|
installed_plugins=lambda: [plugin.id],
|
|
online_plugins=lambda: [plugin],
|
|
local_plugins=lambda: [],
|
|
merge_plugins=lambda items, *_args: items,
|
|
plugin_exists=lambda *_args: True,
|
|
install=install,
|
|
log=Mock(),
|
|
)
|
|
|
|
assert service.sync(
|
|
online_restore_plugins={"demoplugin"},
|
|
) == [plugin.id]
|
|
install.assert_called_once_with(plugin.id, None, False, None)
|
|
|
|
|
|
def test_market_sync_keeps_active_local_payload_when_candidate_still_exists() -> None:
|
|
"""本地候选仍存在时，不应被启动在线恢复覆盖。"""
|
|
online = SimpleNamespace(
|
|
id="DemoPlugin",
|
|
repo_url=REPO_URL,
|
|
plugin_name="Demo",
|
|
plugin_version="1.2.0",
|
|
system_version_compatible=True,
|
|
)
|
|
local = SimpleNamespace(
|
|
id="DemoPlugin",
|
|
repo_url="local://DemoPlugin?package_version=v3",
|
|
plugin_name="Demo Local",
|
|
plugin_version="9.9.10",
|
|
system_version_compatible=True,
|
|
)
|
|
install = Mock(return_value=(True, ""))
|
|
service = PluginSyncService(
|
|
frozen=lambda: False,
|
|
installed_plugins=lambda: [online.id],
|
|
online_plugins=lambda: [online],
|
|
local_plugins=lambda: [local],
|
|
merge_plugins=lambda items, *_args: [online],
|
|
plugin_exists=lambda *_args: True,
|
|
install=install,
|
|
log=Mock(),
|
|
)
|
|
|
|
assert service.sync(online_restore_plugins={"demoplugin"}) == []
|
|
install.assert_not_called()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_market_sync_reuses_startup_lease_through_real_gateway(
|
|
monkeypatch,
|
|
) -> None:
|
|
"""启动自动安装跨线程进入 Gateway 时必须复用同一个 startup lease。"""
|
|
competing_repo_url = "https://github.com/example/MoviePilot-Plugins"
|
|
plugin = SimpleNamespace(
|
|
id="DemoPlugin",
|
|
repo_url=competing_repo_url,
|
|
plugin_name="Demo",
|
|
plugin_version="9.0.0",
|
|
system_version_compatible=True,
|
|
)
|
|
official_candidate = PluginMarketCandidate(
|
|
plugin_id=plugin.id,
|
|
source_key="github:jxxghp/moviepilot-plugins",
|
|
source_type=TrustedPluginSourceType.OFFICIAL,
|
|
repo_url=REPO_URL,
|
|
package_generation="v3",
|
|
plugin_version="1.1.0",
|
|
dto={"v3": True},
|
|
)
|
|
competing_candidate = PluginMarketCandidate(
|
|
plugin_id=plugin.id,
|
|
source_key="github:example/moviepilot-plugins",
|
|
source_type=TrustedPluginSourceType.THIRD_PARTY,
|
|
repo_url=competing_repo_url,
|
|
package_generation="v3",
|
|
plugin_version=plugin.plugin_version,
|
|
dto={"v3": True},
|
|
)
|
|
inventory = CandidateInventory((
|
|
MarketRead.present(
|
|
REPO_URL,
|
|
(official_candidate,),
|
|
package_generation="v3",
|
|
),
|
|
MarketRead.present(
|
|
competing_repo_url,
|
|
(competing_candidate,),
|
|
package_generation="v3",
|
|
),
|
|
))
|
|
identity = PluginIdentity(
|
|
plugin_id=plugin.id,
|
|
normalized_plugin_id="demoplugin",
|
|
trusted_source_type=TrustedPluginSourceType.OFFICIAL,
|
|
trusted_source_key="github:jxxghp/moviepilot-plugins",
|
|
binding_basis=PluginBindingBasis.OFFICIAL_DEFAULT,
|
|
payload_source_type=PluginPayloadSourceType.LOCAL,
|
|
payload_source_key=None,
|
|
declared_version="9.9.10",
|
|
package_generation="v3",
|
|
declared_metadata=PluginDeclaredMetadata.from_package(
|
|
{"name": "Demo local", "v3": True},
|
|
declaration_version="9.9.10",
|
|
manifest_matches_payload=True,
|
|
),
|
|
payload_receipt="sha256:" + "0" * 64,
|
|
revision=1,
|
|
created_at=datetime(2026, 8, 25, 12, 0, tzinfo=timezone.utc),
|
|
updated_at=datetime(2026, 8, 25, 12, 0, tzinfo=timezone.utc),
|
|
bound_at=datetime(2026, 8, 25, 12, 0, tzinfo=timezone.utc),
|
|
payload_applied_at=datetime(2026, 8, 25, 12, 0, tzinfo=timezone.utc),
|
|
)
|
|
executor = AsyncMock()
|
|
executor.execute.return_value = PluginInstallResult(success=True)
|
|
gateway = PluginInstallGateway(
|
|
inventory=AsyncMock(return_value=inventory),
|
|
identity=AsyncMock(return_value=identity),
|
|
candidate_compatibility=lambda _candidate: (True, ""),
|
|
executor=executor,
|
|
clock=lambda: datetime(2026, 8, 25, 12, 0, tzinfo=timezone.utc),
|
|
)
|
|
monkeypatch.setattr(
|
|
global_vars,
|
|
"CURRENT_EVENT_LOOP",
|
|
asyncio.get_running_loop(),
|
|
)
|
|
|
|
def install(
|
|
plugin_id: str,
|
|
repo_url: str | None,
|
|
force: bool,
|
|
startup_token: object | None,
|
|
) -> tuple[bool, str]:
|
|
"""复用生产同步包装层，把线程池安装提交回宿主事件循环。"""
|
|
return plugins_initializer._run_plugin_install_sync(
|
|
gateway,
|
|
plugin_id=plugin_id,
|
|
repo_url=repo_url,
|
|
package_version="v3",
|
|
release_version=None,
|
|
force=force,
|
|
local_sync=False,
|
|
explicit_source=False,
|
|
startup_token=startup_token,
|
|
)
|
|
|
|
service = PluginSyncService(
|
|
frozen=lambda: False,
|
|
installed_plugins=lambda: [plugin.id],
|
|
online_plugins=lambda: [plugin],
|
|
local_plugins=lambda: [],
|
|
merge_plugins=lambda items, *_args: items,
|
|
plugin_exists=lambda *_args: True,
|
|
install=install,
|
|
log=Mock(),
|
|
)
|
|
|
|
async with plugin_lifecycle.hold_startup() as startup_token:
|
|
synced = await asyncio.wait_for(
|
|
asyncio.to_thread(
|
|
service.sync,
|
|
startup_token,
|
|
online_restore_plugins={"demoplugin"},
|
|
),
|
|
timeout=2,
|
|
)
|
|
|
|
assert synced == [plugin.id]
|
|
executor.execute.assert_awaited_once()
|
|
admission = executor.execute.await_args.kwargs["admission"]
|
|
assert admission.candidate.repo_url == REPO_URL
|