jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-20 19:57:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use send_json_auto_id in voice_assistant tests (#90857)

Browse Source
This commit is contained in:
Erik Montnemery 2023-04-06 02:07:42 +02:00 committed by GitHub
parent 4104659986
commit 4276ce96ea
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
tests/components/voice_assistant/test_websocket.py
View File

@ -17,9 +17,8 @@ async def test_text_only_pipeline(
"""Test events from a pipeline run with text input (no STT/TTS).""" """Test events from a pipeline run with text input (no STT/TTS)."""
client = await hass_ws_client(hass) client = await hass_ws_client(hass)
await client.send_json( await client.send_json_auto_id(
{ {
"id": 5,
"type": "voice_assistant/run", "type": "voice_assistant/run",
"start_stage": "intent", "start_stage": "intent",
"end_stage": "intent", "end_stage": "intent",
@ -57,9 +56,8 @@ async def test_audio_pipeline(
"""Test events from a pipeline run with audio input/output.""" """Test events from a pipeline run with audio input/output."""
client = await hass_ws_client(hass) client = await hass_ws_client(hass)
await client.send_json( await client.send_json_auto_id(
{ {
"id": 5,
"type": "voice_assistant/run", "type": "voice_assistant/run",
"start_stage": "stt", "start_stage": "stt",
"end_stage": "tts", "end_stage": "tts",
@ -127,9 +125,8 @@ async def test_intent_timeout(
"homeassistant.components.conversation.async_converse", "homeassistant.components.conversation.async_converse",
new=sleepy_converse, new=sleepy_converse,
): ):
await client.send_json( await client.send_json_auto_id(
{ {
"id": 5,
"type": "voice_assistant/run", "type": "voice_assistant/run",
"start_stage": "intent", "start_stage": "intent",
"end_stage": "intent", "end_stage": "intent",
@ -174,9 +171,8 @@ async def test_text_pipeline_timeout(
"homeassistant.components.voice_assistant.pipeline.PipelineInput.execute", "homeassistant.components.voice_assistant.pipeline.PipelineInput.execute",
new=sleepy_run, new=sleepy_run,
): ):
await client.send_json( await client.send_json_auto_id(
{ {
"id": 5,
"type": "voice_assistant/run", "type": "voice_assistant/run",
"start_stage": "intent", "start_stage": "intent",
"end_stage": "intent", "end_stage": "intent",
@ -208,9 +204,8 @@ async def test_intent_failed(
"homeassistant.components.conversation.async_converse", "homeassistant.components.conversation.async_converse",
new=MagicMock(return_value=RuntimeError), new=MagicMock(return_value=RuntimeError),
): ):
await client.send_json( await client.send_json_auto_id(
{ {
"id": 5,
"type": "voice_assistant/run", "type": "voice_assistant/run",
"start_stage": "intent", "start_stage": "intent",
"end_stage": "intent", "end_stage": "intent",
@ -254,9 +249,8 @@ async def test_audio_pipeline_timeout(
"homeassistant.components.voice_assistant.pipeline.PipelineInput.execute", "homeassistant.components.voice_assistant.pipeline.PipelineInput.execute",
new=sleepy_run, new=sleepy_run,
): ):
await client.send_json( await client.send_json_auto_id(
{ {
"id": 5,
"type": "voice_assistant/run", "type": "voice_assistant/run",
"start_stage": "stt", "start_stage": "stt",
"end_stage": "tts", "end_stage": "tts",
@ -286,9 +280,8 @@ async def test_stt_provider_missing(
): ):
client = await hass_ws_client(hass) client = await hass_ws_client(hass)
await client.send_json( await client.send_json_auto_id(
{ {
"id": 5,
"type": "voice_assistant/run", "type": "voice_assistant/run",
"start_stage": "stt", "start_stage": "stt",
"end_stage": "tts", "end_stage": "tts",
@ -313,9 +306,8 @@ async def test_stt_stream_failed(
): ):
client = await hass_ws_client(hass) client = await hass_ws_client(hass)
await client.send_json( await client.send_json_auto_id(
{ {
"id": 5,
"type": "voice_assistant/run", "type": "voice_assistant/run",
"start_stage": "stt", "start_stage": "stt",
"end_stage": "tts", "end_stage": "tts",
@ -394,9 +386,8 @@ async def test_invalid_stage_order(
"""Test pipeline run with invalid stage order.""" """Test pipeline run with invalid stage order."""
client = await hass_ws_client(hass) client = await hass_ws_client(hass)
await client.send_json( await client.send_json_auto_id(
{ {
"id": 5,
"type": "voice_assistant/run", "type": "voice_assistant/run",
"start_stage": "tts", "start_stage": "tts",
"end_stage": "stt", "end_stage": "stt",