mirror of https://github.com/home-assistant/core
83 lines
2.6 KiB
Python
83 lines
2.6 KiB
Python
"""Tests for the Smarty component."""
|
|
|
|
from unittest.mock import AsyncMock
|
|
|
|
from syrupy import SnapshotAssertion
|
|
|
|
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 device_registry as dr, issue_registry as ir
|
|
from homeassistant.setup import async_setup_component
|
|
|
|
from . import setup_integration
|
|
|
|
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
|
|
|
|
|
|
async def test_device(
|
|
hass: HomeAssistant,
|
|
snapshot: SnapshotAssertion,
|
|
mock_smarty: AsyncMock,
|
|
mock_config_entry: MockConfigEntry,
|
|
device_registry: dr.DeviceRegistry,
|
|
) -> None:
|
|
"""Test device."""
|
|
await setup_integration(hass, mock_config_entry)
|
|
device = device_registry.async_get_device(
|
|
identifiers={(DOMAIN, mock_config_entry.entry_id)}
|
|
)
|
|
assert device
|
|
assert device == snapshot
|