diff --git a/autogpt_platform/backend/backend/api/features/store/content_handlers_test.py b/autogpt_platform/backend/backend/api/features/store/content_handlers_test.py index fee879fae0..28a3db0c28 100644 --- a/autogpt_platform/backend/backend/api/features/store/content_handlers_test.py +++ b/autogpt_platform/backend/backend/api/features/store/content_handlers_test.py @@ -85,6 +85,8 @@ async def test_block_handler_get_missing_items(mocker): mock_block_instance.input_schema.model_json_schema.return_value = { "properties": {"expression": {"description": "Math expression to evaluate"}} } + mock_block_instance.input_schema.get_credentials_fields_info.return_value = {} + mock_block_instance.input_schema.model_fields = {} mock_block_class.return_value = mock_block_instance mock_blocks = {"block-uuid-1": mock_block_class} @@ -309,19 +311,20 @@ async def test_content_handlers_registry(): @pytest.mark.asyncio(loop_scope="session") -async def test_block_handler_handles_missing_attributes(): - """Test BlockHandler gracefully handles blocks with missing attributes.""" +async def test_block_handler_handles_empty_attributes(): + """Test BlockHandler handles blocks with empty/falsy attribute values.""" handler = BlockHandler() - # Mock block with minimal attributes + # Mock block with empty values (all attributes exist but are falsy) mock_block_class = MagicMock() mock_block_instance = MagicMock() mock_block_instance.name = "Minimal Block" mock_block_instance.disabled = False - # No description, categories, or schema - del mock_block_instance.description - del mock_block_instance.categories - del mock_block_instance.input_schema + mock_block_instance.description = "" + mock_block_instance.categories = set() + mock_block_instance.input_schema.model_json_schema.return_value = {} + mock_block_instance.input_schema.get_credentials_fields_info.return_value = {} + mock_block_instance.input_schema.model_fields = {} mock_block_class.return_value = mock_block_instance mock_blocks = {"block-minimal": mock_block_class} @@ -352,6 +355,9 @@ async def test_block_handler_skips_failed_blocks(): good_instance.description = "Works fine" good_instance.categories = [] good_instance.disabled = False + good_instance.input_schema.model_json_schema.return_value = {} + good_instance.input_schema.get_credentials_fields_info.return_value = {} + good_instance.input_schema.model_fields = {} good_block.return_value = good_instance bad_block = MagicMock()