Use HassioServiceInfo in mqtt tests (#60865)

pull/60867/head
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

View File

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