Use DhcpServiceInfo in dhcp tests (#59962)

Co-authored-by: epenet <epenet@users.noreply.github.com>
This commit is contained in:
epenet 2021-11-19 12:49:28 +01:00 committed by GitHub
parent f17d58a049
commit 45d41e584f
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -171,11 +171,11 @@ async def test_dhcp_match_hostname_and_macaddress(hass):
assert mock_init.mock_calls[0][2]["context"] == { assert mock_init.mock_calls[0][2]["context"] == {
"source": config_entries.SOURCE_DHCP "source": config_entries.SOURCE_DHCP
} }
assert mock_init.mock_calls[0][2]["data"] == { assert mock_init.mock_calls[0][2]["data"] == dhcp.DhcpServiceInfo(
dhcp.IP_ADDRESS: "192.168.210.56", ip="192.168.210.56",
dhcp.HOSTNAME: "connect", hostname="connect",
dhcp.MAC_ADDRESS: "b8b7f16db533", macaddress="b8b7f16db533",
} )
async def test_dhcp_renewal_match_hostname_and_macaddress(hass): async def test_dhcp_renewal_match_hostname_and_macaddress(hass):
@ -199,11 +199,11 @@ async def test_dhcp_renewal_match_hostname_and_macaddress(hass):
assert mock_init.mock_calls[0][2]["context"] == { assert mock_init.mock_calls[0][2]["context"] == {
"source": config_entries.SOURCE_DHCP "source": config_entries.SOURCE_DHCP
} }
assert mock_init.mock_calls[0][2]["data"] == { assert mock_init.mock_calls[0][2]["data"] == dhcp.DhcpServiceInfo(
dhcp.IP_ADDRESS: "192.168.1.120", ip="192.168.1.120",
dhcp.HOSTNAME: "irobot-ae9ec12dd3b04885bcbfa36afb01e1cc", hostname="irobot-ae9ec12dd3b04885bcbfa36afb01e1cc",
dhcp.MAC_ADDRESS: "50147903852c", macaddress="50147903852c",
} )
async def test_dhcp_match_hostname(hass): async def test_dhcp_match_hostname(hass):
@ -223,11 +223,11 @@ async def test_dhcp_match_hostname(hass):
assert mock_init.mock_calls[0][2]["context"] == { assert mock_init.mock_calls[0][2]["context"] == {
"source": config_entries.SOURCE_DHCP "source": config_entries.SOURCE_DHCP
} }
assert mock_init.mock_calls[0][2]["data"] == { assert mock_init.mock_calls[0][2]["data"] == dhcp.DhcpServiceInfo(
dhcp.IP_ADDRESS: "192.168.210.56", ip="192.168.210.56",
dhcp.HOSTNAME: "connect", hostname="connect",
dhcp.MAC_ADDRESS: "b8b7f16db533", macaddress="b8b7f16db533",
} )
async def test_dhcp_match_macaddress(hass): async def test_dhcp_match_macaddress(hass):
@ -247,11 +247,11 @@ async def test_dhcp_match_macaddress(hass):
assert mock_init.mock_calls[0][2]["context"] == { assert mock_init.mock_calls[0][2]["context"] == {
"source": config_entries.SOURCE_DHCP "source": config_entries.SOURCE_DHCP
} }
assert mock_init.mock_calls[0][2]["data"] == { assert mock_init.mock_calls[0][2]["data"] == dhcp.DhcpServiceInfo(
dhcp.IP_ADDRESS: "192.168.210.56", ip="192.168.210.56",
dhcp.HOSTNAME: "connect", hostname="connect",
dhcp.MAC_ADDRESS: "b8b7f16db533", macaddress="b8b7f16db533",
} )
async def test_dhcp_match_macaddress_without_hostname(hass): async def test_dhcp_match_macaddress_without_hostname(hass):
@ -271,11 +271,11 @@ async def test_dhcp_match_macaddress_without_hostname(hass):
assert mock_init.mock_calls[0][2]["context"] == { assert mock_init.mock_calls[0][2]["context"] == {
"source": config_entries.SOURCE_DHCP "source": config_entries.SOURCE_DHCP
} }
assert mock_init.mock_calls[0][2]["data"] == { assert mock_init.mock_calls[0][2]["data"] == dhcp.DhcpServiceInfo(
dhcp.IP_ADDRESS: "192.168.107.151", ip="192.168.107.151",
dhcp.HOSTNAME: "", hostname="",
dhcp.MAC_ADDRESS: "606bbd59e4b4", macaddress="606bbd59e4b4",
} )
async def test_dhcp_nomatch(hass): async def test_dhcp_nomatch(hass):
@ -548,11 +548,11 @@ async def test_device_tracker_hostname_and_macaddress_exists_before_start(hass):
assert mock_init.mock_calls[0][2]["context"] == { assert mock_init.mock_calls[0][2]["context"] == {
"source": config_entries.SOURCE_DHCP "source": config_entries.SOURCE_DHCP
} }
assert mock_init.mock_calls[0][2]["data"] == { assert mock_init.mock_calls[0][2]["data"] == dhcp.DhcpServiceInfo(
dhcp.IP_ADDRESS: "192.168.210.56", ip="192.168.210.56",
dhcp.HOSTNAME: "connect", hostname="connect",
dhcp.MAC_ADDRESS: "b8b7f16db533", macaddress="b8b7f16db533",
} )
async def test_device_tracker_hostname_and_macaddress_after_start(hass): async def test_device_tracker_hostname_and_macaddress_after_start(hass):
@ -585,11 +585,11 @@ async def test_device_tracker_hostname_and_macaddress_after_start(hass):
assert mock_init.mock_calls[0][2]["context"] == { assert mock_init.mock_calls[0][2]["context"] == {
"source": config_entries.SOURCE_DHCP "source": config_entries.SOURCE_DHCP
} }
assert mock_init.mock_calls[0][2]["data"] == { assert mock_init.mock_calls[0][2]["data"] == dhcp.DhcpServiceInfo(
dhcp.IP_ADDRESS: "192.168.210.56", ip="192.168.210.56",
dhcp.HOSTNAME: "connect", hostname="connect",
dhcp.MAC_ADDRESS: "b8b7f16db533", macaddress="b8b7f16db533",
} )
async def test_device_tracker_hostname_and_macaddress_after_start_not_home(hass): async def test_device_tracker_hostname_and_macaddress_after_start_not_home(hass):
@ -731,11 +731,11 @@ async def test_aiodiscover_finds_new_hosts(hass):
assert mock_init.mock_calls[0][2]["context"] == { assert mock_init.mock_calls[0][2]["context"] == {
"source": config_entries.SOURCE_DHCP "source": config_entries.SOURCE_DHCP
} }
assert mock_init.mock_calls[0][2]["data"] == { assert mock_init.mock_calls[0][2]["data"] == dhcp.DhcpServiceInfo(
dhcp.IP_ADDRESS: "192.168.210.56", ip="192.168.210.56",
dhcp.HOSTNAME: "connect", hostname="connect",
dhcp.MAC_ADDRESS: "b8b7f16db533", macaddress="b8b7f16db533",
} )
async def test_aiodiscover_does_not_call_again_on_shorter_hostname(hass): async def test_aiodiscover_does_not_call_again_on_shorter_hostname(hass):
@ -786,20 +786,20 @@ async def test_aiodiscover_does_not_call_again_on_shorter_hostname(hass):
assert mock_init.mock_calls[0][2]["context"] == { assert mock_init.mock_calls[0][2]["context"] == {
"source": config_entries.SOURCE_DHCP "source": config_entries.SOURCE_DHCP
} }
assert mock_init.mock_calls[0][2]["data"] == { assert mock_init.mock_calls[0][2]["data"] == dhcp.DhcpServiceInfo(
dhcp.IP_ADDRESS: "192.168.210.56", ip="192.168.210.56",
dhcp.HOSTNAME: "irobot-abc", hostname="irobot-abc",
dhcp.MAC_ADDRESS: "b8b7f16db533", macaddress="b8b7f16db533",
} )
assert mock_init.mock_calls[1][1][0] == "mock-domain" assert mock_init.mock_calls[1][1][0] == "mock-domain"
assert mock_init.mock_calls[1][2]["context"] == { assert mock_init.mock_calls[1][2]["context"] == {
"source": config_entries.SOURCE_DHCP "source": config_entries.SOURCE_DHCP
} }
assert mock_init.mock_calls[1][2]["data"] == { assert mock_init.mock_calls[1][2]["data"] == dhcp.DhcpServiceInfo(
dhcp.IP_ADDRESS: "192.168.210.56", ip="192.168.210.56",
dhcp.HOSTNAME: "irobot-abcdef", hostname="irobot-abcdef",
dhcp.MAC_ADDRESS: "b8b7f16db533", macaddress="b8b7f16db533",
} )
async def test_aiodiscover_finds_new_hosts_after_interval(hass): async def test_aiodiscover_finds_new_hosts_after_interval(hass):
@ -838,8 +838,8 @@ async def test_aiodiscover_finds_new_hosts_after_interval(hass):
assert mock_init.mock_calls[0][2]["context"] == { assert mock_init.mock_calls[0][2]["context"] == {
"source": config_entries.SOURCE_DHCP "source": config_entries.SOURCE_DHCP
} }
assert mock_init.mock_calls[0][2]["data"] == { assert mock_init.mock_calls[0][2]["data"] == dhcp.DhcpServiceInfo(
dhcp.IP_ADDRESS: "192.168.210.56", ip="192.168.210.56",
dhcp.HOSTNAME: "connect", hostname="connect",
dhcp.MAC_ADDRESS: "b8b7f16db533", macaddress="b8b7f16db533",
} )