Release v1.0.1b

This commit is contained in:
SMKRV
2024-11-19 00:56:33 +03:00
parent 34e93e0045
commit 9d54f88520
3 changed files with 0 additions and 73 deletions
BIN
View File
Binary file not shown.
-8
View File
@@ -1,8 +0,0 @@
"""Common fixtures for tests."""
import pytest
from homeassistant.core import HomeAssistant
@pytest.fixture
def hass() -> HomeAssistant:
"""Return a Home Assistant instance for testing."""
return HomeAssistant()
-65
View File
@@ -1,65 +0,0 @@
"""Tests for the HA text AI integration."""
from unittest.mock import AsyncMock, patch
import pytest
from custom_components.ha_text_ai.const import DOMAIN
from homeassistant.core import HomeAssistant
from homeassistant.setup import async_setup_component
@pytest.fixture
def mock_setup_entry() -> AsyncMock:
"""Override async_setup_entry."""
with patch(
"custom_components.ha_text_ai.async_setup_entry",
return_value=True,
) as mock_setup_entry:
yield mock_setup_entry
@pytest.fixture
def mock_coordinator() -> AsyncMock:
"""Override coordinator."""
with patch(
"custom_components.ha_text_ai.coordinator.HATextAICoordinator",
return_value=AsyncMock(),
) as mock_coordinator:
yield mock_coordinator
async def test_async_setup(hass: HomeAssistant, mock_setup_entry: AsyncMock) -> None:
"""Test the initial setup."""
assert await async_setup_component(hass, DOMAIN, {
DOMAIN: {
"api_key": "test_key",
"model": "gpt-3.5-turbo",
"temperature": 0.7,
"max_tokens": 1000,
"api_endpoint": "https://api.openai.com/v1",
"request_interval": 1.0
}
})
await hass.async_block_till_done()
assert DOMAIN in hass.data
async def test_async_setup_entry(
hass: HomeAssistant,
mock_coordinator: AsyncMock
) -> None:
"""Test setup entry."""
entry = MockConfigEntry(
domain=DOMAIN,
data={
"api_key": "test_key",
"model": "gpt-3.5-turbo",
"temperature": 0.7,
"max_tokens": 1000,
"api_endpoint": "https://api.openai.com/v1",
"request_interval": 1.0
},
)
entry.add_to_hass(hass)
assert await hass.config_entries.async_setup(entry.entry_id)
await hass.async_block_till_done()
assert len(mock_coordinator.mock_calls) == 1
assert DOMAIN in hass.data
assert entry.entry_id in hass.data[DOMAIN]