Use HassioServiceInfo in mqtt tests (#60865)
parent
272b147550
commit
37200decf9
|
@ -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"
|
||||
|
|
Loading…
Reference in New Issue