Add test coverage for send_voice_assistant_event

This commit is contained in:
J. Nick Koston 2023-11-26 16:42:22 -06:00
parent 32cbc989ff
commit 9866a0fa84
No known key found for this signature in database
2 changed files with 12 additions and 15 deletions

View File

@ -1304,18 +1304,14 @@ class APIClient:
self, event_type: VoiceAssistantEventType, data: dict[str, str] | None
) -> None:
req = VoiceAssistantEventResponse(event_type=event_type)
data_args = []
if data is not None:
for name, value in data.items():
arg = VoiceAssistantEventData()
arg.name = name
arg.value = value
data_args.append(arg)
# pylint: disable=no-member
req.data.extend(data_args)
# pylint: disable=no-member
req.data.extend(
[
VoiceAssistantEventData(name=name, value=value)
for name, value in data.items()
]
)
self._get_connection().send_message(req)
async def alarm_control_panel_command(

View File

@ -1677,13 +1677,14 @@ async def test_send_voice_assistant_event(auth_client: APIClient) -> None:
auth_client.send_voice_assistant_event(
VoiceAssistantEventModelType.VOICE_ASSISTANT_ERROR,
{
"error": "error",
},
{"error": "error", "ok": "ok"},
)
send.assert_called_once_with(
VoiceAssistantEventResponse(
event_type=VoiceAssistantEventModelType.VOICE_ASSISTANT_ERROR.value,
data=[VoiceAssistantEventData(name="error", value="error")],
data=[
VoiceAssistantEventData(name="error", value="error"),
VoiceAssistantEventData(name="ok", value="ok"),
],
)
)