From 4276ce96eab22c8a31f97d5627f940374261a27b Mon Sep 17 00:00:00 2001
From: Erik Montnemery <erik@montnemery.com>
Date: Thu, 6 Apr 2023 02:07:42 +0200
Subject: [PATCH] Use send_json_auto_id in voice_assistant tests (#90857)
---
.../voice_assistant/test_websocket.py | 27 +++++++------------
1 file changed, 9 insertions(+), 18 deletions(-)
diff --git a/tests/components/voice_assistant/test_websocket.py b/tests/components/voice_assistant/test_websocket.py
index 2acb954c87d..08dadcdd99d 100644
--- a/tests/components/voice_assistant/test_websocket.py
+++ b/tests/components/voice_assistant/test_websocket.py
@@ -17,9 +17,8 @@ async def test_text_only_pipeline(
"""Test events from a pipeline run with text input (no STT/TTS)."""
client = await hass_ws_client(hass)
- await client.send_json(
+ await client.send_json_auto_id(
{
- "id": 5,
"type": "voice_assistant/run",
"start_stage": "intent",
"end_stage": "intent",
@@ -57,9 +56,8 @@ async def test_audio_pipeline(
"""Test events from a pipeline run with audio input/output."""
client = await hass_ws_client(hass)
- await client.send_json(
+ await client.send_json_auto_id(
{
- "id": 5,
"type": "voice_assistant/run",
"start_stage": "stt",
"end_stage": "tts",
@@ -127,9 +125,8 @@ async def test_intent_timeout(
"homeassistant.components.conversation.async_converse",
new=sleepy_converse,
):
- await client.send_json(
+ await client.send_json_auto_id(
{
- "id": 5,
"type": "voice_assistant/run",
"start_stage": "intent",
"end_stage": "intent",
@@ -174,9 +171,8 @@ async def test_text_pipeline_timeout(
"homeassistant.components.voice_assistant.pipeline.PipelineInput.execute",
new=sleepy_run,
):
- await client.send_json(
+ await client.send_json_auto_id(
{
- "id": 5,
"type": "voice_assistant/run",
"start_stage": "intent",
"end_stage": "intent",
@@ -208,9 +204,8 @@ async def test_intent_failed(
"homeassistant.components.conversation.async_converse",
new=MagicMock(return_value=RuntimeError),
):
- await client.send_json(
+ await client.send_json_auto_id(
{
- "id": 5,
"type": "voice_assistant/run",
"start_stage": "intent",
"end_stage": "intent",
@@ -254,9 +249,8 @@ async def test_audio_pipeline_timeout(
"homeassistant.components.voice_assistant.pipeline.PipelineInput.execute",
new=sleepy_run,
):
- await client.send_json(
+ await client.send_json_auto_id(
{
- "id": 5,
"type": "voice_assistant/run",
"start_stage": "stt",
"end_stage": "tts",
@@ -286,9 +280,8 @@ async def test_stt_provider_missing(
):
client = await hass_ws_client(hass)
- await client.send_json(
+ await client.send_json_auto_id(
{
- "id": 5,
"type": "voice_assistant/run",
"start_stage": "stt",
"end_stage": "tts",
@@ -313,9 +306,8 @@ async def test_stt_stream_failed(
):
client = await hass_ws_client(hass)
- await client.send_json(
+ await client.send_json_auto_id(
{
- "id": 5,
"type": "voice_assistant/run",
"start_stage": "stt",
"end_stage": "tts",
@@ -394,9 +386,8 @@ async def test_invalid_stage_order(
"""Test pipeline run with invalid stage order."""
client = await hass_ws_client(hass)
- await client.send_json(
+ await client.send_json_auto_id(
{
- "id": 5,
"type": "voice_assistant/run",
"start_stage": "tts",
"end_stage": "stt",
--
GitLab