From 01fea8bbdd1667e6280794fbf0873c2e8e428e7a Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Fri, 19 Nov 2021 12:18:32 +0100 Subject: [PATCH] Use DhcpServiceInfo in guardian tests (#59970) Co-authored-by: epenet --- tests/components/guardian/test_config_flow.py | 23 +++++++++---------- 1 file changed, 11 insertions(+), 12 deletions(-) diff --git a/tests/components/guardian/test_config_flow.py b/tests/components/guardian/test_config_flow.py index b8d23508c77..f3e0adeec0f 100644 --- a/tests/components/guardian/test_config_flow.py +++ b/tests/components/guardian/test_config_flow.py @@ -4,8 +4,7 @@ from unittest.mock import patch from aioguardian.errors import GuardianError from homeassistant import data_entry_flow -from homeassistant.components import zeroconf -from homeassistant.components.dhcp import HOSTNAME, IP_ADDRESS, MAC_ADDRESS +from homeassistant.components import dhcp, zeroconf from homeassistant.components.guardian import CONF_UID, DOMAIN from homeassistant.components.guardian.config_flow import ( async_get_pin_from_discovery_hostname, @@ -137,11 +136,11 @@ async def test_step_zeroconf_already_in_progress(hass): async def test_step_dhcp(hass, ping_client): """Test the dhcp step.""" - dhcp_data = { - IP_ADDRESS: "192.168.1.100", - HOSTNAME: "GVC1-ABCD.local.", - MAC_ADDRESS: "aa:bb:cc:dd:ee:ff", - } + dhcp_data = dhcp.DhcpServiceInfo( + ip="192.168.1.100", + hostname="GVC1-ABCD.local.", + macaddress="aa:bb:cc:dd:ee:ff", + ) result = await hass.config_entries.flow.async_init( DOMAIN, context={"source": SOURCE_DHCP}, data=dhcp_data @@ -163,11 +162,11 @@ async def test_step_dhcp(hass, ping_client): async def test_step_dhcp_already_in_progress(hass): """Test the zeroconf step aborting because it's already in progress.""" - dhcp_data = { - IP_ADDRESS: "192.168.1.100", - HOSTNAME: "GVC1-ABCD.local.", - MAC_ADDRESS: "aa:bb:cc:dd:ee:ff", - } + dhcp_data = dhcp.DhcpServiceInfo( + ip="192.168.1.100", + hostname="GVC1-ABCD.local.", + macaddress="aa:bb:cc:dd:ee:ff", + ) result = await hass.config_entries.flow.async_init( DOMAIN, context={"source": SOURCE_DHCP}, data=dhcp_data