Improve test coverage

Author: RemyRoux

tests/components/wyoming/test_satellite.py

@@ -1679,3 +1679,82 @@ async def async_pipeline_from_audio_stream(
         # Stop the satellite
         await hass.config_entries.async_unload(entry.entry_id)
         await hass.async_block_till_done()
+
+
+async def test_ask_question(
+    hass: HomeAssistant,
+    entity_registry: er.EntityRegistry,
+    tmp_path: Path,
+) -> None:
+    """Test the internal ask question flow."""
+    assert await async_setup_component(hass, assist_pipeline.DOMAIN, {})
+
+    wav_path = tmp_path / "test.wav"
+    with wave.open(str(wav_path), "wb") as wav_file:
+        wav_file.setframerate(SAMPLE_RATE)
+        wav_file.setsampwidth(SAMPLE_WIDTH)
+        wav_file.setnchannels(SAMPLE_CHANNELS)
+        wav_file.writeframes(bytes(SAMPLE_RATE * 2))
+
+    # This is the announcement object our mock will return.
+    # It bypasses TTS and points directly to our local file.
+    fake_announcement = assist_satellite.AssistSatelliteAnnouncement(
+        message="Are you sure?",
+        media_id=str(wav_path),
+        original_media_id="",
+        tts_token=None,
+        media_id_source="url",
+    )
+
+    with (
+        patch(
+            "homeassistant.components.wyoming.data.load_wyoming_info",
+            return_value=SATELLITE_INFO,
+        ),
+        patch(
+            "homeassistant.components.wyoming.assist_satellite.AsyncTcpClient",
+            SatelliteAsyncTcpClient(responses=[], block_until_inject=True),
+        ) as mock_client,
+        patch(
+            "homeassistant.components.wyoming.assist_satellite.WyomingAssistSatellite._resolve_announcement_media_id",
+            return_value=fake_announcement,
+        ) as mock_resolve,
+    ):
+        entry = await setup_config_entry(hass)
+        satellite_entity = hass.data[DOMAIN][entry.entry_id].device.assist_satellite
+        assert satellite_entity is not None
+
+        async with asyncio.timeout(1):
+            await mock_client.connect_event.wait()
+            await mock_client.run_satellite_event.wait()
+
+        ask_task = hass.async_create_background_task(
+            satellite_entity.async_internal_ask_question(
+                question="Are you sure?",
+                answers=[
+                    {"id": "yes", "sentences": ["yes"]},
+                    {"id": "no", "sentences": ["no"]},
+                ],
+            ),
+            "wyoming_satellite_ask_question",
+        )
+
+        # Announcement should play
+        async with asyncio.timeout(2):
+            await mock_client.tts_audio_start_event.wait()
+            await mock_client.tts_audio_chunk_event.wait()
+            await mock_client.tts_audio_stop_event.wait()
+            mock_client.inject_event(Played().event())
+
+        # Satellite should be waiting for a response
+        async with asyncio.timeout(2):
+            # Simulate user saying "yes"
+            mock_client.inject_event(Transcript(text="yes").event())
+            mock_client.inject_event(AudioStop().event())
+
+            result = await ask_task
+        assert result is not None
+        assert result.id == "yes"
+        assert result.sentence == "yes"
+
+        mock_resolve.assert_called_once()