fixes configuration flow #22706 (#22754)

This commit is contained in:
Fredrik Erlandsson 2019-04-05 08:48:41 +02:00 committed by Paulus Schoutsen
parent 5252c92670
commit e9d55bf1c0
2 changed files with 11 additions and 3 deletions

View File

@ -38,9 +38,12 @@ class FlowHandler(config_entries.ConfigFlow):
"""Create device."""
from pydaikin.appliance import Appliance
try:
device = Appliance(
host,
self.hass.helpers.aiohttp_client.async_get_clientsession(),
)
with async_timeout.timeout(10):
device = await self.hass.async_add_executor_job(
Appliance, host)
await device.init()
except asyncio.TimeoutError:
return self.async_abort(reason='device_timeout')
except Exception: # pylint: disable=broad-except

View File

@ -24,9 +24,14 @@ def init_config_flow(hass):
@pytest.fixture
def mock_daikin():
"""Mock tellduslive."""
"""Mock pydaikin."""
async def mock_daikin_init():
"""Mock the init function in pydaikin."""
pass
with MockDependency('pydaikin.appliance') as mock_daikin_:
mock_daikin_.Appliance().values.get.return_value = 'AABBCCDDEEFF'
mock_daikin_.Appliance().init = mock_daikin_init
yield mock_daikin_