Fix tests

KapJI · KapJI · commit 58eb58a0b273 · 2021-03-29T00:01:38.000+01:00
diff --git a/tests/components/lutron_caseta/test_config_flow.py b/tests/components/lutron_caseta/test_config_flow.py
@@ -17,11 +17,12 @@
     ERROR_CANNOT_CONNECT,
     STEP_IMPORT_FAILED,
 )
-from homeassistant.components.zeroconf import ATTR_HOSTNAME
 from homeassistant.const import CONF_HOST
 
 from tests.common import MockConfigEntry
 
+ATTR_HOSTNAME = "hostname"
+
 EMPTY_MOCK_CONFIG_ENTRY = {
     CONF_HOST: "",
     CONF_KEYFILE: "",
diff --git a/tests/components/xiaomi_miio/test_config_flow.py b/tests/components/xiaomi_miio/test_config_flow.py
@@ -4,7 +4,6 @@
 from miio import DeviceException
 
 from homeassistant import config_entries
-from homeassistant.components import zeroconf
 from homeassistant.components.xiaomi_miio import const
 from homeassistant.components.xiaomi_miio.config_flow import DEFAULT_GATEWAY_NAME
 from homeassistant.const import CONF_HOST, CONF_NAME, CONF_TOKEN
@@ -106,7 +105,7 @@ async def test_zeroconf_gateway_success(hass):
         const.DOMAIN,
         context={"source": config_entries.SOURCE_ZEROCONF},
         data={
-            zeroconf.ATTR_HOST: TEST_HOST,
+            CONF_HOST: TEST_HOST,
             ZEROCONF_NAME: TEST_ZEROCONF_NAME,
             ZEROCONF_PROP: {ZEROCONF_MAC: TEST_MAC},
         },
@@ -146,7 +145,7 @@ async def test_zeroconf_unknown_device(hass):
         const.DOMAIN,
         context={"source": config_entries.SOURCE_ZEROCONF},
         data={
-            zeroconf.ATTR_HOST: TEST_HOST,
+            CONF_HOST: TEST_HOST,
             ZEROCONF_NAME: "not-a-xiaomi-miio-device",
             ZEROCONF_PROP: {ZEROCONF_MAC: TEST_MAC},
         },
@@ -171,7 +170,7 @@ async def test_zeroconf_missing_data(hass):
     result = await hass.config_entries.flow.async_init(
         const.DOMAIN,
         context={"source": config_entries.SOURCE_ZEROCONF},
-        data={zeroconf.ATTR_HOST: TEST_HOST, ZEROCONF_NAME: TEST_ZEROCONF_NAME},
+        data={CONF_HOST: TEST_HOST, ZEROCONF_NAME: TEST_ZEROCONF_NAME},
     )
 
     assert result["type"] == "abort"
@@ -342,7 +341,7 @@ async def zeroconf_device_success(hass, zeroconf_name_to_test, model_to_test):
         const.DOMAIN,
         context={"source": config_entries.SOURCE_ZEROCONF},
         data={
-            zeroconf.ATTR_HOST: TEST_HOST,
+            CONF_HOST: TEST_HOST,
             ZEROCONF_NAME: zeroconf_name_to_test,
             ZEROCONF_PROP: {"poch": f"0:mac={TEST_MAC_DEVICE}\x00"},
         },
