Use HassioServiceInfo in deconz tests (#60868)

This commit is contained in:
epenet 2021-12-02 21:33:23 +01:00 committed by GitHub
parent d7c4a669f1
commit 9e96f3e227
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -18,6 +18,7 @@ from homeassistant.components.deconz.const import (
CONF_MASTER_GATEWAY, CONF_MASTER_GATEWAY,
DOMAIN as DECONZ_DOMAIN, DOMAIN as DECONZ_DOMAIN,
) )
from homeassistant.components.hassio.discovery import HassioServiceInfo
from homeassistant.components.ssdp import ATTR_UPNP_MANUFACTURER_URL, ATTR_UPNP_SERIAL from homeassistant.components.ssdp import ATTR_UPNP_MANUFACTURER_URL, ATTR_UPNP_SERIAL
from homeassistant.config_entries import ( from homeassistant.config_entries import (
SOURCE_HASSIO, SOURCE_HASSIO,
@ -542,13 +543,15 @@ async def test_flow_hassio_discovery(hass):
"""Test hassio discovery flow works.""" """Test hassio discovery flow works."""
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
DECONZ_DOMAIN, DECONZ_DOMAIN,
data={ data=HassioServiceInfo(
"addon": "Mock Addon", config={
CONF_HOST: "mock-deconz", "addon": "Mock Addon",
CONF_PORT: 80, CONF_HOST: "mock-deconz",
CONF_SERIAL: BRIDGEID, CONF_PORT: 80,
CONF_API_KEY: API_KEY, CONF_SERIAL: BRIDGEID,
}, CONF_API_KEY: API_KEY,
}
),
context={"source": SOURCE_HASSIO}, context={"source": SOURCE_HASSIO},
) )
assert result["type"] == RESULT_TYPE_FORM assert result["type"] == RESULT_TYPE_FORM
@ -583,12 +586,14 @@ async def test_hassio_discovery_update_configuration(hass, aioclient_mock):
) as mock_setup_entry: ) as mock_setup_entry:
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
DECONZ_DOMAIN, DECONZ_DOMAIN,
data={ data=HassioServiceInfo(
CONF_HOST: "2.3.4.5", config={
CONF_PORT: 8080, CONF_HOST: "2.3.4.5",
CONF_API_KEY: "updated", CONF_PORT: 8080,
CONF_SERIAL: BRIDGEID, CONF_API_KEY: "updated",
}, CONF_SERIAL: BRIDGEID,
}
),
context={"source": SOURCE_HASSIO}, context={"source": SOURCE_HASSIO},
) )
await hass.async_block_till_done() await hass.async_block_till_done()
@ -607,12 +612,14 @@ async def test_hassio_discovery_dont_update_configuration(hass, aioclient_mock):
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
DECONZ_DOMAIN, DECONZ_DOMAIN,
data={ data=HassioServiceInfo(
CONF_HOST: "1.2.3.4", config={
CONF_PORT: 80, CONF_HOST: "1.2.3.4",
CONF_API_KEY: API_KEY, CONF_PORT: 80,
CONF_SERIAL: BRIDGEID, CONF_API_KEY: API_KEY,
}, CONF_SERIAL: BRIDGEID,
}
),
context={"source": SOURCE_HASSIO}, context={"source": SOURCE_HASSIO},
) )