Fix assist_pipeline tests (#91757)

This commit is contained in:
Erik Montnemery 2023-04-20 18:19:31 +02:00 committed by GitHub
parent 62d38b49bc
commit b3d50e67cd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 20 additions and 0 deletions

View File

@ -79,10 +79,14 @@ async def test_pipeline_from_audio_stream_legacy(
{ {
"type": "assist_pipeline/pipeline/create", "type": "assist_pipeline/pipeline/create",
"conversation_engine": "homeassistant", "conversation_engine": "homeassistant",
"conversation_language": "test_language",
"language": "en-US", "language": "en-US",
"name": "test_name", "name": "test_name",
"stt_engine": "test", "stt_engine": "test",
"stt_language": "test_language",
"tts_engine": "test", "tts_engine": "test",
"tts_language": "test_language",
"tts_voice": "Arnold Schwarzenegger",
} }
) )
msg = await client.receive_json() msg = await client.receive_json()
@ -141,10 +145,14 @@ async def test_pipeline_from_audio_stream_entity(
{ {
"type": "assist_pipeline/pipeline/create", "type": "assist_pipeline/pipeline/create",
"conversation_engine": "homeassistant", "conversation_engine": "homeassistant",
"conversation_language": "test_language",
"language": "en-US", "language": "en-US",
"name": "test_name", "name": "test_name",
"stt_engine": mock_stt_provider_entity.entity_id, "stt_engine": mock_stt_provider_entity.entity_id,
"stt_language": "test_language",
"tts_engine": "test", "tts_engine": "test",
"tts_language": "test_language",
"tts_voice": "Arnold Schwarzenegger",
} }
) )
msg = await client.receive_json() msg = await client.receive_json()

View File

@ -801,10 +801,14 @@ async def test_get_pipeline(
{ {
"type": "assist_pipeline/pipeline/create", "type": "assist_pipeline/pipeline/create",
"conversation_engine": "test_conversation_engine", "conversation_engine": "test_conversation_engine",
"conversation_language": "test_language",
"language": "test_language", "language": "test_language",
"name": "test_name", "name": "test_name",
"stt_engine": "test_stt_engine", "stt_engine": "test_stt_engine",
"stt_language": "test_language",
"tts_engine": "test_tts_engine", "tts_engine": "test_tts_engine",
"tts_language": "test_language",
"tts_voice": "Arnold Schwarzenegger",
} }
) )
msg = await client.receive_json() msg = await client.receive_json()
@ -821,11 +825,15 @@ async def test_get_pipeline(
assert msg["success"] assert msg["success"]
assert msg["result"] == { assert msg["result"] == {
"conversation_engine": "test_conversation_engine", "conversation_engine": "test_conversation_engine",
"conversation_language": "test_language",
"id": pipeline_id, "id": pipeline_id,
"language": "test_language", "language": "test_language",
"name": "test_name", "name": "test_name",
"stt_engine": "test_stt_engine", "stt_engine": "test_stt_engine",
"stt_language": "test_language",
"tts_engine": "test_tts_engine", "tts_engine": "test_tts_engine",
"tts_language": "test_language",
"tts_voice": "Arnold Schwarzenegger",
} }
await client.send_json_auto_id( await client.send_json_auto_id(
@ -838,11 +846,15 @@ async def test_get_pipeline(
assert msg["success"] assert msg["success"]
assert msg["result"] == { assert msg["result"] == {
"conversation_engine": "test_conversation_engine", "conversation_engine": "test_conversation_engine",
"conversation_language": "test_language",
"id": pipeline_id, "id": pipeline_id,
"language": "test_language", "language": "test_language",
"name": "test_name", "name": "test_name",
"stt_engine": "test_stt_engine", "stt_engine": "test_stt_engine",
"stt_language": "test_language",
"tts_engine": "test_tts_engine", "tts_engine": "test_tts_engine",
"tts_language": "test_language",
"tts_voice": "Arnold Schwarzenegger",
} }