42 lines
1.5 KiB
Python
42 lines
1.5 KiB
Python
|
"""Test IGD setup process."""
|
||
|
|
|
||
|
|
from unittest.mock import patch, MagicMock
|
||
|
|
|
||
|
|
from homeassistant.setup import async_setup_component
|
||
|
|
from homeassistant.components import igd
|
||
|
|
from homeassistant.const import EVENT_HOMEASSISTANT_STOP
|
||
|
|
|
||
|
|
from tests.common import MockConfigEntry
|
||
|
|
from tests.common import mock_coro
|
||
|
|
|
||
|
|
|
||
|
|
async def test_async_setup_entry_port_forward_created(hass):
|
||
|
|
"""Test async_setup_entry."""
|
||
|
|
|
||
|
|
udn = 'uuid:device_1'
|
||
|
|
entry = MockConfigEntry(domain=igd.DOMAIN, data={
|
||
|
|
'ssdp_description': 'http://192.168.1.1/desc.xml',
|
||
|
|
'udn': udn,
|
||
|
|
'sensors': False,
|
||
|
|
'port_forward': True,
|
||
|
|
})
|
||
|
|
|
||
|
|
# ensure hass.http is available
|
||
|
|
await async_setup_component(hass, 'igd')
|
||
|
|
|
||
|
|
mock_igd_device = MagicMock()
|
||
|
|
mock_igd_device.udn = udn
|
||
|
|
mock_igd_device.async_add_port_mapping.return_value = mock_coro()
|
||
|
|
mock_igd_device.async_remove_port_mapping.return_value = mock_coro()
|
||
|
|
with patch.object(igd, '_async_create_igd_device') as mock_create_device:
|
||
|
|
mock_create_device.return_value = mock_coro(return_value=mock_igd_device)
|
||
|
|
with patch('homeassistant.components.igd.get_local_ip', return_value='192.168.1.10'):
|
||
|
|
assert await igd.async_setup_entry(hass, entry) is True
|
||
|
|
|
||
|
|
hass.bus.async_fire(EVENT_HOMEASSISTANT_STOP)
|
||
|
|
await hass.async_block_till_done()
|
||
|
|
|
||
|
|
assert hass.data[igd.DOMAIN]['devices'][udn] == mock_igd_device
|
||
|
|
assert len(mock_igd_device.async_add_port_mapping.mock_calls) > 0
|
||
|
|
assert len(mock_igd_device.async_delete_port_mapping.mock_calls) > 0
|