evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Use HassioServiceInfo in mqtt tests (#60865)

Browse source
This commit is contained in:
epenet 2021-12-02 21:10:50 +01:00 • committed by GitHub
parent 272b147550
commit 37200decf9
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 17 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

27
tests/components/mqtt/test_config_flow.py
View file

@ -7,6 +7,7 @@ import voluptuous as vol
from homeassistant import config_entries, data_entry_flow from homeassistant import config_entries, data_entry_flow
from homeassistant.components import mqtt from homeassistant.components import mqtt
from homeassistant.components.hassio.discovery import HassioServiceInfo
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.setup import async_setup_component from homeassistant.setup import async_setup_component
@ -110,7 +111,9 @@ async def test_hassio_already_configured(hass):
MockConfigEntry(domain="mqtt").add_to_hass(hass) MockConfigEntry(domain="mqtt").add_to_hass(hass)
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
"mqtt", context={"source": config_entries.SOURCE_HASSIO} "mqtt",
context={"source": config_entries.SOURCE_HASSIO},
data=HassioServiceInfo(config={}),
) )
assert result["type"] == "abort" assert result["type"] == "abort"
assert result["reason"] == "already_configured" assert result["reason"] == "already_configured"
@ -124,7 +127,9 @@ async def test_hassio_ignored(hass: HomeAssistant) -> None:
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
mqtt.DOMAIN, mqtt.DOMAIN,
data={"addon": "Mosquitto", "host": "mock-mosquitto", "port": "1883"}, data=HassioServiceInfo(
config={"addon": "Mosquitto", "host": "mock-mosquitto", "port": "1883"}
),
context={"source": config_entries.SOURCE_HASSIO}, context={"source": config_entries.SOURCE_HASSIO},
) )
assert result assert result
@ -140,14 +145,16 @@ async def test_hassio_confirm(
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
"mqtt", "mqtt",
data={ data=HassioServiceInfo(
"addon": "Mock Addon", config={
"host": "mock-broker", "addon": "Mock Addon",
"port": 1883, "host": "mock-broker",
"username": "mock-user", "port": 1883,
"password": "mock-pass", "username": "mock-user",
"protocol": "3.1.1", "password": "mock-pass",
}, "protocol": "3.1.1",
}
),
context={"source": config_entries.SOURCE_HASSIO}, context={"source": config_entries.SOURCE_HASSIO},
) )
assert result["type"] == "form" assert result["type"] == "form"