mirror of
https://github.com/langgenius/dify.git
synced 2026-11-11 13:51:25 +08:00
41 lines
1.2 KiB
Python
41 lines
1.2 KiB
Python
"""Shared fixtures for services.plugin test suite."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from collections.abc import Iterator
|
|
from unittest.mock import MagicMock
|
|
|
|
import pytest
|
|
|
|
from services.entities.feature_entities import PluginInstallationScope
|
|
|
|
|
|
def make_features(
|
|
restrict_to_marketplace: bool = False,
|
|
scope: PluginInstallationScope = PluginInstallationScope.ALL,
|
|
) -> MagicMock:
|
|
"""Create a mock plugin installation permission."""
|
|
permission = MagicMock()
|
|
permission.restrict_to_marketplace_only = restrict_to_marketplace
|
|
permission.plugin_installation_scope = scope
|
|
return permission
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_installer(monkeypatch: pytest.MonkeyPatch) -> MagicMock:
|
|
"""Patch PluginInstaller at the service import site."""
|
|
mock = MagicMock()
|
|
monkeypatch.setattr("core.plugin.plugin_service.PluginInstaller", lambda: mock)
|
|
return mock
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_features() -> Iterator[MagicMock]:
|
|
"""Patch SystemFeatureService to return permissive defaults."""
|
|
from unittest.mock import patch
|
|
|
|
features = make_features()
|
|
with patch("core.plugin.plugin_service.SystemFeatureService") as mock_fs:
|
|
mock_fs.get_plugin_installation_permission.return_value = features
|
|
yield features
|