test(open_responses): cover endpoint review paths

Author: Komzpa

tests/components/open_responses/test_ai_task.py

@@ -4,10 +4,15 @@
 from copy import deepcopy
 from typing import Any
 
+import pytest
 import voluptuous as vol
 
 from homeassistant.components import ai_task
+from homeassistant.components.open_responses.client import (
+    OpenResponsesInvalidModelError,
+)
 from homeassistant.core import HomeAssistant
+from homeassistant.exceptions import HomeAssistantError
 from homeassistant.helpers import selector
 
 from tests.common import MockConfigEntry
@@ -91,3 +96,26 @@ async def test_generate_structured_data(
 
     assert result.data == {"characters": ["Mario", "Luigi"]}
     assert calls[0]["text"]["format"]["type"] == "json_schema"
+
+
+async def test_generate_data_handles_invalid_model(
+    hass: HomeAssistant,
+    mock_config_entry: MockConfigEntry,
+    mock_init_component: None,
+) -> None:
+    """Test runtime model errors are surfaced as Home Assistant errors."""
+
+    async def stream_response(**params: Any) -> AsyncGenerator[dict[str, Any]]:
+        if params["model"] == "open-responses-model":
+            raise OpenResponsesInvalidModelError("missing model")
+        yield {}
+
+    mock_config_entry.runtime_data.stream_response = stream_response
+
+    with pytest.raises(HomeAssistantError, match="Invalid Open Responses model"):
+        await ai_task.async_generate_data(
+            hass,
+            task_name="Test Task",
+            entity_id="ai_task.open_responses_ai_task",
+            instructions="Generate test data",
+        )

tests/components/open_responses/test_client.py

@@ -78,6 +78,32 @@ async def test_create_response_requests_json() -> None:
     assert http_client.post.await_args.kwargs["headers"]["accept"] == "application/json"
 
 
+async def test_stream_response_requests_event_stream() -> None:
+    """Test streaming responses request server-sent events."""
+    requests: list[httpx.Request] = []
+
+    async def handler(request: httpx.Request) -> httpx.Response:
+        requests.append(request)
+        return httpx.Response(
+            200,
+            content=b'data: {"type": "response.created"}\n\ndata: [DONE]\n\n',
+            request=request,
+        )
+
+    async with httpx.AsyncClient(transport=httpx.MockTransport(handler)) as http_client:
+        client = OpenResponsesClient(http_client, "api-key", "https://example.local/v1")
+
+        assert [
+            event
+            async for event in client.stream_response(
+                model="model",
+                input=[{"type": "message", "role": "user", "content": "ping"}],
+            )
+        ] == [{"type": "response.created"}]
+
+    assert requests[0].headers["accept"] == "text/event-stream"
+
+
 async def test_iter_sse_events_accumulates_multiline_data() -> None:
     """Test SSE data lines are joined until the event delimiter."""