From 272b14755091ab85abd5c1aeb00b7ce52928c75c Mon Sep 17 00:00:00 2001
From: epenet <6771947+epenet@users.noreply.github.com>
Date: Thu, 2 Dec 2021 21:10:15 +0100
Subject: [PATCH] Use HassioServiceInfo in adguard tests (#60866)

---
 tests/components/adguard/test_config_flow.py | 25 ++++++++++++++++----
 1 file changed, 21 insertions(+), 4 deletions(-)

diff --git a/tests/components/adguard/test_config_flow.py b/tests/components/adguard/test_config_flow.py
index 75d138f400c..b87d0796a54 100644
--- a/tests/components/adguard/test_config_flow.py
+++ b/tests/components/adguard/test_config_flow.py
@@ -3,6 +3,7 @@ import aiohttp
 
 from homeassistant import config_entries, data_entry_flow
 from homeassistant.components.adguard.const import DOMAIN
+from homeassistant.components.hassio.discovery import HassioServiceInfo
 from homeassistant.config_entries import SOURCE_USER
 from homeassistant.const import (
     CONF_HOST,
@@ -121,7 +122,13 @@ async def test_hassio_already_configured(hass: HomeAssistant) -> None:
 
     result = await hass.config_entries.flow.async_init(
         DOMAIN,
-        data={"addon": "AdGuard Home Addon", "host": "mock-adguard", "port": "3000"},
+        data=HassioServiceInfo(
+            config={
+                "addon": "AdGuard Home Addon",
+                "host": "mock-adguard",
+                "port": "3000",
+            }
+        ),
         context={"source": config_entries.SOURCE_HASSIO},
     )
     assert result
@@ -137,7 +144,13 @@ async def test_hassio_ignored(hass: HomeAssistant) -> None:
 
     result = await hass.config_entries.flow.async_init(
         DOMAIN,
-        data={"addon": "AdGuard Home Addon", "host": "mock-adguard", "port": "3000"},
+        data=HassioServiceInfo(
+            config={
+                "addon": "AdGuard Home Addon",
+                "host": "mock-adguard",
+                "port": "3000",
+            }
+        ),
         context={"source": config_entries.SOURCE_HASSIO},
     )
     assert result
@@ -157,7 +170,9 @@ async def test_hassio_confirm(
 
     result = await hass.config_entries.flow.async_init(
         DOMAIN,
-        data={"addon": "AdGuard Home Addon", "host": "mock-adguard", "port": 3000},
+        data=HassioServiceInfo(
+            config={"addon": "AdGuard Home Addon", "host": "mock-adguard", "port": 3000}
+        ),
         context={"source": config_entries.SOURCE_HASSIO},
     )
     assert result
@@ -191,7 +206,9 @@ async def test_hassio_connection_error(
 
     result = await hass.config_entries.flow.async_init(
         DOMAIN,
-        data={"addon": "AdGuard Home Addon", "host": "mock-adguard", "port": 3000},
+        data=HassioServiceInfo(
+            config={"addon": "AdGuard Home Addon", "host": "mock-adguard", "port": 3000}
+        ),
         context={"source": config_entries.SOURCE_HASSIO},
     )