mirror of
https://github.com/home-assistant/core.git
synced 2025-05-07 07:29:17 +00:00
63 lines
2.1 KiB
Python
63 lines
2.1 KiB
Python
"""Tests for the Smarty component."""
|
|
|
|
from unittest.mock import AsyncMock
|
|
|
|
from homeassistant.components.smarty import DOMAIN
|
|
from homeassistant.const import CONF_HOST, CONF_NAME
|
|
from homeassistant.core import DOMAIN as HOMEASSISTANT_DOMAIN, HomeAssistant
|
|
from homeassistant.helpers import issue_registry as ir
|
|
from homeassistant.setup import async_setup_component
|
|
|
|
from tests.common import MockConfigEntry
|
|
|
|
|
|
async def test_import_flow(
|
|
hass: HomeAssistant,
|
|
mock_smarty: AsyncMock,
|
|
issue_registry: ir.IssueRegistry,
|
|
mock_setup_entry: AsyncMock,
|
|
) -> None:
|
|
"""Test import flow."""
|
|
assert await async_setup_component(
|
|
hass, DOMAIN, {DOMAIN: {CONF_HOST: "192.168.0.2", CONF_NAME: "smarty"}}
|
|
)
|
|
await hass.async_block_till_done()
|
|
assert len(hass.config_entries.async_entries(DOMAIN)) == 1
|
|
assert (HOMEASSISTANT_DOMAIN, "deprecated_yaml_smarty") in issue_registry.issues
|
|
|
|
|
|
async def test_import_flow_already_exists(
|
|
hass: HomeAssistant,
|
|
mock_smarty: AsyncMock,
|
|
issue_registry: ir.IssueRegistry,
|
|
mock_setup_entry: AsyncMock,
|
|
mock_config_entry: MockConfigEntry,
|
|
) -> None:
|
|
"""Test import flow when entry already exists."""
|
|
mock_config_entry.add_to_hass(hass)
|
|
assert await async_setup_component(
|
|
hass, DOMAIN, {DOMAIN: {CONF_HOST: "192.168.0.2", CONF_NAME: "smarty"}}
|
|
)
|
|
await hass.async_block_till_done()
|
|
assert len(hass.config_entries.async_entries(DOMAIN)) == 1
|
|
assert (HOMEASSISTANT_DOMAIN, "deprecated_yaml_smarty") in issue_registry.issues
|
|
|
|
|
|
async def test_import_flow_error(
|
|
hass: HomeAssistant,
|
|
mock_smarty: AsyncMock,
|
|
issue_registry: ir.IssueRegistry,
|
|
mock_setup_entry: AsyncMock,
|
|
) -> None:
|
|
"""Test import flow when error occurs."""
|
|
mock_smarty.update.return_value = False
|
|
assert await async_setup_component(
|
|
hass, DOMAIN, {DOMAIN: {CONF_HOST: "192.168.0.2", CONF_NAME: "smarty"}}
|
|
)
|
|
await hass.async_block_till_done()
|
|
assert len(hass.config_entries.async_entries(DOMAIN)) == 0
|
|
assert (
|
|
DOMAIN,
|
|
"deprecated_yaml_import_issue_cannot_connect",
|
|
) in issue_registry.issues
|