diff --git a/homeassistant/components/daikin/config_flow.py b/homeassistant/components/daikin/config_flow.py index a26e2e1c05b..eb0441b9740 100644 --- a/homeassistant/components/daikin/config_flow.py +++ b/homeassistant/components/daikin/config_flow.py @@ -129,7 +129,7 @@ class FlowHandler(config_entries.ConfigFlow): async def async_step_zeroconf(self, discovery_info): """Prepare configuration for a discovered Daikin device.""" _LOGGER.debug("Zeroconf user_input: %s", discovery_info) - devices = Discovery.poll(discovery_info[CONF_HOST]) + devices = Discovery().poll(ip=discovery_info[CONF_HOST]) await self.async_set_unique_id(next(iter(devices.values()))[KEY_MAC]) self._abort_if_unique_id_configured() self.host = discovery_info[CONF_HOST] diff --git a/tests/components/daikin/test_config_flow.py b/tests/components/daikin/test_config_flow.py index 9702e5ad2c7..f8d13bdd355 100644 --- a/tests/components/daikin/test_config_flow.py +++ b/tests/components/daikin/test_config_flow.py @@ -45,9 +45,9 @@ def mock_daikin(): def mock_daikin_discovery(): """Mock pydaikin Discovery.""" with patch("homeassistant.components.daikin.config_flow.Discovery") as Discovery: - Discovery.poll = PropertyMock( - return_value={"127.0.01": {"mac": "AABBCCDDEEFF", "id": "test"}} - ) + Discovery().poll.return_value = { + "127.0.01": {"mac": "AABBCCDDEEFF", "id": "test"} + } yield Discovery