mirror of
https://github.com/langgenius/dify.git
synced 2026-09-06 01:04:49 +08:00
42 lines
1.7 KiB
Python
42 lines
1.7 KiB
Python
from pytest_mock import MockerFixture
|
|
|
|
from enums import DeploymentEdition
|
|
from services.entities.feature_entities import FeatureModel, SystemFeatureModel
|
|
from services.feature_service import FeatureService
|
|
from services.feature_service_gateway import FeatureServiceGateway
|
|
from services.system_feature_service import SystemFeatureService
|
|
|
|
|
|
def test_public_system_features_delegate_to_existing_service(mocker: MockerFixture) -> None:
|
|
system_features = SystemFeatureModel(deployment_edition=DeploymentEdition.COMMUNITY)
|
|
get_system_features = mocker.patch.object(
|
|
SystemFeatureService,
|
|
"get_public_system_features",
|
|
return_value=system_features,
|
|
)
|
|
|
|
result = FeatureServiceGateway().get_public_system_features()
|
|
|
|
assert result is system_features
|
|
get_system_features.assert_called_once_with()
|
|
|
|
|
|
def test_workspace_features_exclude_independently_queried_vector_space(mocker: MockerFixture) -> None:
|
|
features = FeatureModel(vector_space=None)
|
|
get_features = mocker.patch.object(FeatureService, "get_features", return_value=features)
|
|
|
|
result = FeatureServiceGateway().get_workspace_features("workspace_123")
|
|
|
|
assert result is features
|
|
get_features.assert_called_once_with("workspace_123", exclude_vector_space=True)
|
|
|
|
|
|
def test_trial_models_delegate_to_workspace_aware_feature_service(mocker: MockerFixture) -> None:
|
|
trial_models = ["langgenius/openai/openai"]
|
|
get_trial_models = mocker.patch.object(FeatureService, "get_trial_models", return_value=trial_models)
|
|
|
|
result = FeatureServiceGateway().get_trial_models("workspace_123")
|
|
|
|
assert result == trial_models
|
|
get_trial_models.assert_called_once_with("workspace_123")
|