146 lines
4.7 KiB
Python
146 lines
4.7 KiB
Python
"""Tests for the OpenAI integration."""
|
|
from unittest.mock import patch
|
|
|
|
from openai import error
|
|
|
|
from homeassistant.components import conversation
|
|
from homeassistant.core import Context, HomeAssistant
|
|
from homeassistant.helpers import area_registry, device_registry, intent
|
|
|
|
from tests.common import MockConfigEntry
|
|
|
|
|
|
async def test_default_prompt(hass: HomeAssistant, mock_init_component) -> None:
|
|
"""Test that the default prompt works."""
|
|
device_reg = device_registry.async_get(hass)
|
|
area_reg = area_registry.async_get(hass)
|
|
|
|
for i in range(3):
|
|
area_reg.async_create(f"{i}Empty Area")
|
|
|
|
device_reg.async_get_or_create(
|
|
config_entry_id="1234",
|
|
connections={("test", "1234")},
|
|
name="Test Device",
|
|
manufacturer="Test Manufacturer",
|
|
model="Test Model",
|
|
suggested_area="Test Area",
|
|
)
|
|
for i in range(3):
|
|
device_reg.async_get_or_create(
|
|
config_entry_id="1234",
|
|
connections={("test", f"{i}abcd")},
|
|
name="Test Service",
|
|
manufacturer="Test Manufacturer",
|
|
model="Test Model",
|
|
suggested_area="Test Area",
|
|
entry_type=device_registry.DeviceEntryType.SERVICE,
|
|
)
|
|
device_reg.async_get_or_create(
|
|
config_entry_id="1234",
|
|
connections={("test", "5678")},
|
|
name="Test Device 2",
|
|
manufacturer="Test Manufacturer 2",
|
|
model="Device 2",
|
|
suggested_area="Test Area 2",
|
|
)
|
|
device_reg.async_get_or_create(
|
|
config_entry_id="1234",
|
|
connections={("test", "9876")},
|
|
name="Test Device 3",
|
|
manufacturer="Test Manufacturer 3",
|
|
model="Test Model 3A",
|
|
suggested_area="Test Area 2",
|
|
)
|
|
device_reg.async_get_or_create(
|
|
config_entry_id="1234",
|
|
connections={("test", "qwer")},
|
|
name="Test Device 4",
|
|
suggested_area="Test Area 2",
|
|
)
|
|
device = device_reg.async_get_or_create(
|
|
config_entry_id="1234",
|
|
connections={("test", "9876-disabled")},
|
|
name="Test Device 3",
|
|
manufacturer="Test Manufacturer 3",
|
|
model="Test Model 3A",
|
|
suggested_area="Test Area 2",
|
|
)
|
|
device_reg.async_update_device(
|
|
device.id, disabled_by=device_registry.DeviceEntryDisabler.USER
|
|
)
|
|
device_reg.async_get_or_create(
|
|
config_entry_id="1234",
|
|
connections={("test", "9876-no-name")},
|
|
manufacturer="Test Manufacturer NoName",
|
|
model="Test Model NoName",
|
|
suggested_area="Test Area 2",
|
|
)
|
|
device_reg.async_get_or_create(
|
|
config_entry_id="1234",
|
|
connections={("test", "9876-integer-values")},
|
|
name=1,
|
|
manufacturer=2,
|
|
model=3,
|
|
suggested_area="Test Area 2",
|
|
)
|
|
with patch("openai.Completion.acreate") as mock_create:
|
|
result = await conversation.async_converse(hass, "hello", None, Context())
|
|
|
|
assert result.response.response_type == intent.IntentResponseType.ACTION_DONE
|
|
assert (
|
|
mock_create.mock_calls[0][2]["prompt"]
|
|
== """This smart home is controlled by Home Assistant.
|
|
|
|
An overview of the areas and the devices in this smart home:
|
|
|
|
Test Area:
|
|
- Test Device (Test Model)
|
|
|
|
Test Area 2:
|
|
- Test Device 2
|
|
- Test Device 3 (Test Model 3A)
|
|
- Test Device 4
|
|
- 1 (3)
|
|
|
|
Answer the user's questions about the world truthfully.
|
|
|
|
If the user wants to control a device, reject the request and suggest using the Home Assistant app.
|
|
|
|
Now finish this conversation:
|
|
|
|
Smart home: How can I assist?
|
|
User: hello
|
|
Smart home: """
|
|
)
|
|
|
|
|
|
async def test_error_handling(hass: HomeAssistant, mock_init_component) -> None:
|
|
"""Test that the default prompt works."""
|
|
with patch("openai.Completion.acreate", side_effect=error.ServiceUnavailableError):
|
|
result = await conversation.async_converse(hass, "hello", None, Context())
|
|
|
|
assert result.response.response_type == intent.IntentResponseType.ERROR, result
|
|
assert result.response.error_code == "unknown", result
|
|
|
|
|
|
async def test_template_error(
|
|
hass: HomeAssistant, mock_config_entry: MockConfigEntry
|
|
) -> None:
|
|
"""Test that template error handling works."""
|
|
hass.config_entries.async_update_entry(
|
|
mock_config_entry,
|
|
options={
|
|
"prompt": "talk like a {% if True %}smarthome{% else %}pirate please.",
|
|
},
|
|
)
|
|
with patch(
|
|
"openai.Engine.list",
|
|
), patch("openai.Completion.acreate"):
|
|
await hass.config_entries.async_setup(mock_config_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
result = await conversation.async_converse(hass, "hello", None, Context())
|
|
|
|
assert result.response.response_type == intent.IntentResponseType.ERROR, result
|
|
assert result.response.error_code == "unknown", result
|