fix: add null check in get_recommend_app_detail before accessing result['id'] (#36153)

This commit is contained in:
EvanYao
2026-05-14 14:42:22 +08:00
committed by GitHub
parent 9355d36718
commit aeb7687e2c
3 changed files with 75 additions and 1 deletions

View File

@@ -47,7 +47,9 @@ class RecommendedAppService:
"""
mode = dify_config.HOSTED_FETCH_APP_TEMPLATES_MODE
retrieval_instance = RecommendAppRetrievalFactory.get_recommend_app_factory(mode)()
result: dict[str, Any] = retrieval_instance.get_recommend_app_detail(app_id)
result: dict[str, Any] | None = retrieval_instance.get_recommend_app_detail(app_id)
if result is None:
return None
if FeatureService.get_system_features().enable_trial_app:
app_id = result["id"]
trial_app_model = db.session.scalar(select(TrialApp).where(TrialApp.app_id == app_id).limit(1))

View File

@@ -352,6 +352,32 @@ class TestRecommendedAppServiceTrialFeatures:
assert result["id"] == app_id
assert result["can_trial"] is has_trial_app
def test_get_detail_returns_none_when_not_found_and_trial_enabled(
self,
db_session_with_containers: Session,
monkeypatch: pytest.MonkeyPatch,
):
"""Regression: accessing result['id'] when result is None must not crash."""
retrieval_instance = MagicMock()
retrieval_instance.get_recommend_app_detail.return_value = None
retrieval_factory = MagicMock(return_value=retrieval_instance)
monkeypatch.setattr(service_module.dify_config, "HOSTED_FETCH_APP_TEMPLATES_MODE", "remote", raising=False)
monkeypatch.setattr(
service_module.RecommendAppRetrievalFactory,
"get_recommend_app_factory",
MagicMock(return_value=retrieval_factory),
)
monkeypatch.setattr(
service_module.FeatureService,
"get_system_features",
MagicMock(return_value=SimpleNamespace(enable_trial_app=True)),
)
result = RecommendedAppService.get_recommend_app_detail("nonexistent")
assert result is None
retrieval_instance.get_recommend_app_detail.assert_called_once_with("nonexistent")
def test_add_trial_app_record_increments_count_for_existing(self, db_session_with_containers: Session):
app_id = str(uuid.uuid4())
account_id = str(uuid.uuid4())

View File

@@ -0,0 +1,46 @@
"""Unit tests for RecommendedAppService.get_recommend_app_detail null handling.
Regression tests for #36096: accessing result['id'] when the retrieval
returns None causes a TypeError / KeyError in self-hosted mode.
"""
from types import SimpleNamespace
from unittest.mock import MagicMock, patch
from services.recommended_app_service import RecommendedAppService
class TestGetRecommendAppDetailNullCheck:
@patch("services.recommended_app_service.FeatureService", autospec=True)
@patch("services.recommended_app_service.RecommendAppRetrievalFactory", autospec=True)
@patch("services.recommended_app_service.dify_config", autospec=True)
def test_returns_none_when_retrieval_returns_none_and_trial_disabled(
self, mock_config, mock_factory_class, mock_feature_service
):
mock_config.HOSTED_FETCH_APP_TEMPLATES_MODE = "remote"
mock_instance = MagicMock()
mock_instance.get_recommend_app_detail.return_value = None
mock_factory_class.get_recommend_app_factory.return_value = MagicMock(return_value=mock_instance)
mock_feature_service.get_system_features.return_value = SimpleNamespace(enable_trial_app=False)
result = RecommendedAppService.get_recommend_app_detail("nonexistent")
assert result is None
@patch("services.recommended_app_service.FeatureService", autospec=True)
@patch("services.recommended_app_service.RecommendAppRetrievalFactory", autospec=True)
@patch("services.recommended_app_service.dify_config", autospec=True)
def test_returns_none_when_retrieval_returns_none_and_trial_enabled(
self, mock_config, mock_factory_class, mock_feature_service
):
"""Regression for #36096: must not crash when result is None and enable_trial_app is True."""
mock_config.HOSTED_FETCH_APP_TEMPLATES_MODE = "remote"
mock_instance = MagicMock()
mock_instance.get_recommend_app_detail.return_value = None
mock_factory_class.get_recommend_app_factory.return_value = MagicMock(return_value=mock_instance)
mock_feature_service.get_system_features.return_value = SimpleNamespace(enable_trial_app=True)
result = RecommendedAppService.get_recommend_app_detail("nonexistent")
assert result is None
mock_instance.get_recommend_app_detail.assert_called_once_with("nonexistent")