mirror of
https://github.com/home-assistant/core
synced 2024-09-18 19:55:20 +02:00
93 lines
3.0 KiB
Python
93 lines
3.0 KiB
Python
"""Test Mikrotik setup process."""
|
|
from unittest.mock import AsyncMock, Mock, patch
|
|
|
|
from homeassistant.components import mikrotik
|
|
from homeassistant.setup import async_setup_component
|
|
|
|
from . import MOCK_DATA
|
|
|
|
from tests.common import MockConfigEntry
|
|
|
|
|
|
async def test_setup_with_no_config(hass):
|
|
"""Test that we do not discover anything or try to set up a hub."""
|
|
assert await async_setup_component(hass, mikrotik.DOMAIN, {}) is True
|
|
assert mikrotik.DOMAIN not in hass.data
|
|
|
|
|
|
async def test_successful_config_entry(hass):
|
|
"""Test config entry successful setup."""
|
|
entry = MockConfigEntry(
|
|
domain=mikrotik.DOMAIN,
|
|
data=MOCK_DATA,
|
|
)
|
|
entry.add_to_hass(hass)
|
|
mock_registry = Mock()
|
|
|
|
with patch.object(mikrotik, "MikrotikHub") as mock_hub, patch(
|
|
"homeassistant.helpers.device_registry.async_get_registry",
|
|
return_value=mock_registry,
|
|
):
|
|
mock_hub.return_value.async_setup = AsyncMock(return_value=True)
|
|
mock_hub.return_value.serial_num = "12345678"
|
|
mock_hub.return_value.model = "RB750"
|
|
mock_hub.return_value.hostname = "mikrotik"
|
|
mock_hub.return_value.firmware = "3.65"
|
|
assert await mikrotik.async_setup_entry(hass, entry) is True
|
|
|
|
assert len(mock_hub.mock_calls) == 2
|
|
p_hass, p_entry = mock_hub.mock_calls[0][1]
|
|
|
|
assert p_hass is hass
|
|
assert p_entry is entry
|
|
|
|
assert len(mock_registry.mock_calls) == 1
|
|
assert mock_registry.mock_calls[0][2] == {
|
|
"config_entry_id": entry.entry_id,
|
|
"connections": {("mikrotik", "12345678")},
|
|
"manufacturer": mikrotik.ATTR_MANUFACTURER,
|
|
"model": "RB750",
|
|
"name": "mikrotik",
|
|
"sw_version": "3.65",
|
|
}
|
|
|
|
|
|
async def test_hub_fail_setup(hass):
|
|
"""Test that a failed setup will not store the hub."""
|
|
entry = MockConfigEntry(
|
|
domain=mikrotik.DOMAIN,
|
|
data=MOCK_DATA,
|
|
)
|
|
entry.add_to_hass(hass)
|
|
|
|
with patch.object(mikrotik, "MikrotikHub") as mock_hub:
|
|
mock_hub.return_value.async_setup = AsyncMock(return_value=False)
|
|
assert await mikrotik.async_setup_entry(hass, entry) is False
|
|
|
|
assert mikrotik.DOMAIN not in hass.data
|
|
|
|
|
|
async def test_unload_entry(hass):
|
|
"""Test being able to unload an entry."""
|
|
entry = MockConfigEntry(
|
|
domain=mikrotik.DOMAIN,
|
|
data=MOCK_DATA,
|
|
)
|
|
entry.add_to_hass(hass)
|
|
|
|
with patch.object(mikrotik, "MikrotikHub") as mock_hub, patch(
|
|
"homeassistant.helpers.device_registry.async_get_registry",
|
|
return_value=Mock(),
|
|
):
|
|
mock_hub.return_value.async_setup = AsyncMock(return_value=True)
|
|
mock_hub.return_value.serial_num = "12345678"
|
|
mock_hub.return_value.model = "RB750"
|
|
mock_hub.return_value.hostname = "mikrotik"
|
|
mock_hub.return_value.firmware = "3.65"
|
|
assert await mikrotik.async_setup_entry(hass, entry) is True
|
|
|
|
assert len(mock_hub.return_value.mock_calls) == 1
|
|
|
|
assert await mikrotik.async_unload_entry(hass, entry)
|
|
assert entry.entry_id not in hass.data[mikrotik.DOMAIN]
|