Remove deprecated hass.components from vilfo config flow tests (#113821)

This commit is contained in:
Jan-Philipp Benecke 2024-03-20 01:13:05 +01:00 committed by GitHub
parent 685468d845
commit 9add4aea70
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -5,6 +5,7 @@ from unittest.mock import Mock, patch
import vilfo
from homeassistant import config_entries, data_entry_flow
from homeassistant.components.vilfo import config_flow
from homeassistant.components.vilfo.const import DOMAIN
from homeassistant.const import CONF_ACCESS_TOKEN, CONF_HOST, CONF_ID, CONF_MAC
from homeassistant.core import HomeAssistant
@ -173,9 +174,7 @@ async def test_validate_input_returns_data(hass: HomeAssistant) -> None:
), patch(
"vilfo.Client.resolve_firmware_version", return_value=firmware_version
), patch("vilfo.Client.resolve_mac_address", return_value=None):
result = await hass.components.vilfo.config_flow.validate_input(
hass, data=mock_data
)
result = await config_flow.validate_input(hass, data=mock_data)
assert result["title"] == mock_data["host"]
assert result[CONF_HOST] == mock_data["host"]
@ -187,15 +186,9 @@ async def test_validate_input_returns_data(hass: HomeAssistant) -> None:
), patch(
"vilfo.Client.resolve_firmware_version", return_value=firmware_version
), patch("vilfo.Client.resolve_mac_address", return_value=mock_mac):
result2 = await hass.components.vilfo.config_flow.validate_input(
hass, data=mock_data
)
result3 = await hass.components.vilfo.config_flow.validate_input(
hass, data=mock_data_with_ip
)
result4 = await hass.components.vilfo.config_flow.validate_input(
hass, data=mock_data_with_ipv6
)
result2 = await config_flow.validate_input(hass, data=mock_data)
result3 = await config_flow.validate_input(hass, data=mock_data_with_ip)
result4 = await config_flow.validate_input(hass, data=mock_data_with_ipv6)
assert result2["title"] == mock_data["host"]
assert result2[CONF_HOST] == mock_data["host"]