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

View File

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