mirror of
https://github.com/home-assistant/core.git
synced 2025-11-08 02:19:31 +00:00
40 lines
1.2 KiB
Python
40 lines
1.2 KiB
Python
"""Tests for the AI Task entity model."""
|
|
|
|
from freezegun import freeze_time
|
|
|
|
from homeassistant.components.ai_task import async_generate_text
|
|
from homeassistant.const import STATE_UNKNOWN
|
|
from homeassistant.core import HomeAssistant
|
|
|
|
from .conftest import TEST_ENTITY_ID, MockAITaskEntity
|
|
|
|
from tests.common import MockConfigEntry
|
|
|
|
|
|
@freeze_time("2025-06-08 16:28:13")
|
|
async def test_state_generate_text(
|
|
hass: HomeAssistant,
|
|
init_components: None,
|
|
mock_config_entry: MockConfigEntry,
|
|
mock_ai_task_entity: MockAITaskEntity,
|
|
) -> None:
|
|
"""Test the state of the AI Task entity is updated when generating text."""
|
|
entity = hass.states.get(TEST_ENTITY_ID)
|
|
assert entity is not None
|
|
assert entity.state == STATE_UNKNOWN
|
|
|
|
result = await async_generate_text(
|
|
hass,
|
|
task_name="Test task",
|
|
entity_id=TEST_ENTITY_ID,
|
|
instructions="Test prompt",
|
|
)
|
|
assert result.result == "Mock result"
|
|
|
|
entity = hass.states.get(TEST_ENTITY_ID)
|
|
assert entity.state == "2025-06-08T16:28:13+00:00"
|
|
|
|
assert mock_ai_task_entity.mock_generate_text_tasks
|
|
task = mock_ai_task_entity.mock_generate_text_tasks[0]
|
|
assert task.instructions == "Test prompt"
|