mirror of
https://github.com/home-assistant/core
synced 2024-09-06 10:29:55 +02:00
Improve tests for Broadlink config flow (#39894)
This commit is contained in:
parent
9f08955fef
commit
a6d3ee90f0
@ -12,13 +12,16 @@ from . import get_device
|
|||||||
|
|
||||||
from tests.async_mock import call, patch
|
from tests.async_mock import call, patch
|
||||||
|
|
||||||
|
DEVICE_DISCOVERY = "homeassistant.components.broadlink.config_flow.blk.discover"
|
||||||
|
DEVICE_FACTORY = "homeassistant.components.broadlink.config_flow.blk.gendevice"
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture(autouse=True)
|
@pytest.fixture(autouse=True)
|
||||||
def broadlink_setup_fixture():
|
def broadlink_setup_fixture():
|
||||||
"""Mock broadlink entry setup."""
|
"""Mock broadlink entry setup."""
|
||||||
with patch(
|
with patch(
|
||||||
"homeassistant.components.broadlink.async_setup_entry", return_value=True
|
"homeassistant.components.broadlink.async_setup", return_value=True
|
||||||
):
|
), patch("homeassistant.components.broadlink.async_setup_entry", return_value=True):
|
||||||
yield
|
yield
|
||||||
|
|
||||||
|
|
||||||
@ -38,7 +41,7 @@ async def test_flow_user_works(hass):
|
|||||||
assert result["step_id"] == "user"
|
assert result["step_id"] == "user"
|
||||||
assert result["errors"] == {}
|
assert result["errors"] == {}
|
||||||
|
|
||||||
with patch("broadlink.discover", return_value=[mock_api]) as mock_discover:
|
with patch(DEVICE_DISCOVERY, return_value=[mock_api]) as mock_discover:
|
||||||
result = await hass.config_entries.flow.async_configure(
|
result = await hass.config_entries.flow.async_configure(
|
||||||
result["flow_id"],
|
result["flow_id"],
|
||||||
{"host": device.host, "timeout": device.timeout},
|
{"host": device.host, "timeout": device.timeout},
|
||||||
@ -69,7 +72,7 @@ async def test_flow_user_already_in_progress(hass):
|
|||||||
DOMAIN, context={"source": config_entries.SOURCE_USER}
|
DOMAIN, context={"source": config_entries.SOURCE_USER}
|
||||||
)
|
)
|
||||||
|
|
||||||
with patch("broadlink.discover", return_value=[device.get_mock_api()]):
|
with patch(DEVICE_DISCOVERY, return_value=[device.get_mock_api()]):
|
||||||
result = await hass.config_entries.flow.async_configure(
|
result = await hass.config_entries.flow.async_configure(
|
||||||
result["flow_id"],
|
result["flow_id"],
|
||||||
{"host": device.host, "timeout": device.timeout},
|
{"host": device.host, "timeout": device.timeout},
|
||||||
@ -79,7 +82,7 @@ async def test_flow_user_already_in_progress(hass):
|
|||||||
DOMAIN, context={"source": config_entries.SOURCE_USER}
|
DOMAIN, context={"source": config_entries.SOURCE_USER}
|
||||||
)
|
)
|
||||||
|
|
||||||
with patch("broadlink.discover", return_value=[device.get_mock_api()]):
|
with patch(DEVICE_DISCOVERY, return_value=[device.get_mock_api()]):
|
||||||
result = await hass.config_entries.flow.async_configure(
|
result = await hass.config_entries.flow.async_configure(
|
||||||
result["flow_id"],
|
result["flow_id"],
|
||||||
{"host": device.host, "timeout": device.timeout},
|
{"host": device.host, "timeout": device.timeout},
|
||||||
@ -106,7 +109,7 @@ async def test_flow_user_mac_already_configured(hass):
|
|||||||
device.timeout = 20
|
device.timeout = 20
|
||||||
mock_api = device.get_mock_api()
|
mock_api = device.get_mock_api()
|
||||||
|
|
||||||
with patch("broadlink.discover", return_value=[mock_api]):
|
with patch(DEVICE_DISCOVERY, return_value=[mock_api]):
|
||||||
result = await hass.config_entries.flow.async_configure(
|
result = await hass.config_entries.flow.async_configure(
|
||||||
result["flow_id"],
|
result["flow_id"],
|
||||||
{"host": device.host, "timeout": device.timeout},
|
{"host": device.host, "timeout": device.timeout},
|
||||||
@ -125,7 +128,7 @@ async def test_flow_user_invalid_ip_address(hass):
|
|||||||
DOMAIN, context={"source": config_entries.SOURCE_USER}
|
DOMAIN, context={"source": config_entries.SOURCE_USER}
|
||||||
)
|
)
|
||||||
|
|
||||||
with patch("broadlink.discover", side_effect=OSError(errno.EINVAL, None)):
|
with patch(DEVICE_DISCOVERY, side_effect=OSError(errno.EINVAL, None)):
|
||||||
result = await hass.config_entries.flow.async_configure(
|
result = await hass.config_entries.flow.async_configure(
|
||||||
result["flow_id"],
|
result["flow_id"],
|
||||||
{"host": "0.0.0.1"},
|
{"host": "0.0.0.1"},
|
||||||
@ -142,7 +145,7 @@ async def test_flow_user_invalid_hostname(hass):
|
|||||||
DOMAIN, context={"source": config_entries.SOURCE_USER}
|
DOMAIN, context={"source": config_entries.SOURCE_USER}
|
||||||
)
|
)
|
||||||
|
|
||||||
with patch("broadlink.discover", side_effect=OSError(socket.EAI_NONAME, None)):
|
with patch(DEVICE_DISCOVERY, side_effect=OSError(socket.EAI_NONAME, None)):
|
||||||
result = await hass.config_entries.flow.async_configure(
|
result = await hass.config_entries.flow.async_configure(
|
||||||
result["flow_id"],
|
result["flow_id"],
|
||||||
{"host": "pancakemaster.local"},
|
{"host": "pancakemaster.local"},
|
||||||
@ -161,7 +164,7 @@ async def test_flow_user_device_not_found(hass):
|
|||||||
DOMAIN, context={"source": config_entries.SOURCE_USER}
|
DOMAIN, context={"source": config_entries.SOURCE_USER}
|
||||||
)
|
)
|
||||||
|
|
||||||
with patch("broadlink.discover", return_value=[]):
|
with patch(DEVICE_DISCOVERY, return_value=[]):
|
||||||
result = await hass.config_entries.flow.async_configure(
|
result = await hass.config_entries.flow.async_configure(
|
||||||
result["flow_id"],
|
result["flow_id"],
|
||||||
{"host": device.host},
|
{"host": device.host},
|
||||||
@ -178,7 +181,7 @@ async def test_flow_user_network_unreachable(hass):
|
|||||||
DOMAIN, context={"source": config_entries.SOURCE_USER}
|
DOMAIN, context={"source": config_entries.SOURCE_USER}
|
||||||
)
|
)
|
||||||
|
|
||||||
with patch("broadlink.discover", side_effect=OSError(errno.ENETUNREACH, None)):
|
with patch(DEVICE_DISCOVERY, side_effect=OSError(errno.ENETUNREACH, None)):
|
||||||
result = await hass.config_entries.flow.async_configure(
|
result = await hass.config_entries.flow.async_configure(
|
||||||
result["flow_id"],
|
result["flow_id"],
|
||||||
{"host": "192.168.1.32"},
|
{"host": "192.168.1.32"},
|
||||||
@ -195,7 +198,7 @@ async def test_flow_user_os_error(hass):
|
|||||||
DOMAIN, context={"source": config_entries.SOURCE_USER}
|
DOMAIN, context={"source": config_entries.SOURCE_USER}
|
||||||
)
|
)
|
||||||
|
|
||||||
with patch("broadlink.discover", side_effect=OSError()):
|
with patch(DEVICE_DISCOVERY, side_effect=OSError()):
|
||||||
result = await hass.config_entries.flow.async_configure(
|
result = await hass.config_entries.flow.async_configure(
|
||||||
result["flow_id"],
|
result["flow_id"],
|
||||||
{"host": "192.168.1.32"},
|
{"host": "192.168.1.32"},
|
||||||
@ -216,7 +219,7 @@ async def test_flow_auth_authentication_error(hass):
|
|||||||
DOMAIN, context={"source": config_entries.SOURCE_USER}
|
DOMAIN, context={"source": config_entries.SOURCE_USER}
|
||||||
)
|
)
|
||||||
|
|
||||||
with patch("broadlink.discover", return_value=[mock_api]):
|
with patch(DEVICE_DISCOVERY, return_value=[mock_api]):
|
||||||
result = await hass.config_entries.flow.async_configure(
|
result = await hass.config_entries.flow.async_configure(
|
||||||
result["flow_id"],
|
result["flow_id"],
|
||||||
{"host": device.host, "timeout": device.timeout},
|
{"host": device.host, "timeout": device.timeout},
|
||||||
@ -237,7 +240,7 @@ async def test_flow_auth_device_offline(hass):
|
|||||||
DOMAIN, context={"source": config_entries.SOURCE_USER}
|
DOMAIN, context={"source": config_entries.SOURCE_USER}
|
||||||
)
|
)
|
||||||
|
|
||||||
with patch("broadlink.discover", return_value=[mock_api]):
|
with patch(DEVICE_DISCOVERY, return_value=[mock_api]):
|
||||||
result = await hass.config_entries.flow.async_configure(
|
result = await hass.config_entries.flow.async_configure(
|
||||||
result["flow_id"],
|
result["flow_id"],
|
||||||
{"host": device.host},
|
{"host": device.host},
|
||||||
@ -258,7 +261,7 @@ async def test_flow_auth_firmware_error(hass):
|
|||||||
DOMAIN, context={"source": config_entries.SOURCE_USER}
|
DOMAIN, context={"source": config_entries.SOURCE_USER}
|
||||||
)
|
)
|
||||||
|
|
||||||
with patch("broadlink.discover", return_value=[mock_api]):
|
with patch(DEVICE_DISCOVERY, return_value=[mock_api]):
|
||||||
result = await hass.config_entries.flow.async_configure(
|
result = await hass.config_entries.flow.async_configure(
|
||||||
result["flow_id"],
|
result["flow_id"],
|
||||||
{"host": device.host},
|
{"host": device.host},
|
||||||
@ -279,7 +282,7 @@ async def test_flow_auth_network_unreachable(hass):
|
|||||||
DOMAIN, context={"source": config_entries.SOURCE_USER}
|
DOMAIN, context={"source": config_entries.SOURCE_USER}
|
||||||
)
|
)
|
||||||
|
|
||||||
with patch("broadlink.discover", return_value=[mock_api]):
|
with patch(DEVICE_DISCOVERY, return_value=[mock_api]):
|
||||||
result = await hass.config_entries.flow.async_configure(
|
result = await hass.config_entries.flow.async_configure(
|
||||||
result["flow_id"],
|
result["flow_id"],
|
||||||
{"host": device.host},
|
{"host": device.host},
|
||||||
@ -300,7 +303,7 @@ async def test_flow_auth_os_error(hass):
|
|||||||
DOMAIN, context={"source": config_entries.SOURCE_USER}
|
DOMAIN, context={"source": config_entries.SOURCE_USER}
|
||||||
)
|
)
|
||||||
|
|
||||||
with patch("broadlink.discover", return_value=[mock_api]):
|
with patch(DEVICE_DISCOVERY, return_value=[mock_api]):
|
||||||
result = await hass.config_entries.flow.async_configure(
|
result = await hass.config_entries.flow.async_configure(
|
||||||
result["flow_id"],
|
result["flow_id"],
|
||||||
{"host": device.host},
|
{"host": device.host},
|
||||||
@ -321,13 +324,13 @@ async def test_flow_reset_works(hass):
|
|||||||
DOMAIN, context={"source": config_entries.SOURCE_USER}
|
DOMAIN, context={"source": config_entries.SOURCE_USER}
|
||||||
)
|
)
|
||||||
|
|
||||||
with patch("broadlink.discover", return_value=[mock_api]):
|
with patch(DEVICE_DISCOVERY, return_value=[mock_api]):
|
||||||
result = await hass.config_entries.flow.async_configure(
|
result = await hass.config_entries.flow.async_configure(
|
||||||
result["flow_id"],
|
result["flow_id"],
|
||||||
{"host": device.host, "timeout": device.timeout},
|
{"host": device.host, "timeout": device.timeout},
|
||||||
)
|
)
|
||||||
|
|
||||||
with patch("broadlink.discover", return_value=[device.get_mock_api()]):
|
with patch(DEVICE_DISCOVERY, return_value=[device.get_mock_api()]):
|
||||||
result = await hass.config_entries.flow.async_configure(
|
result = await hass.config_entries.flow.async_configure(
|
||||||
result["flow_id"],
|
result["flow_id"],
|
||||||
{"host": device.host, "timeout": device.timeout},
|
{"host": device.host, "timeout": device.timeout},
|
||||||
@ -353,7 +356,7 @@ async def test_flow_unlock_works(hass):
|
|||||||
DOMAIN, context={"source": config_entries.SOURCE_USER}
|
DOMAIN, context={"source": config_entries.SOURCE_USER}
|
||||||
)
|
)
|
||||||
|
|
||||||
with patch("broadlink.discover", return_value=[mock_api]):
|
with patch(DEVICE_DISCOVERY, return_value=[mock_api]):
|
||||||
result = await hass.config_entries.flow.async_configure(
|
result = await hass.config_entries.flow.async_configure(
|
||||||
result["flow_id"],
|
result["flow_id"],
|
||||||
{"host": device.host, "timeout": device.timeout},
|
{"host": device.host, "timeout": device.timeout},
|
||||||
@ -392,7 +395,7 @@ async def test_flow_unlock_device_offline(hass):
|
|||||||
DOMAIN, context={"source": config_entries.SOURCE_USER}
|
DOMAIN, context={"source": config_entries.SOURCE_USER}
|
||||||
)
|
)
|
||||||
|
|
||||||
with patch("broadlink.discover", return_value=[mock_api]):
|
with patch(DEVICE_DISCOVERY, return_value=[mock_api]):
|
||||||
result = await hass.config_entries.flow.async_configure(
|
result = await hass.config_entries.flow.async_configure(
|
||||||
result["flow_id"],
|
result["flow_id"],
|
||||||
{"host": device.host, "timeout": device.timeout},
|
{"host": device.host, "timeout": device.timeout},
|
||||||
@ -419,7 +422,7 @@ async def test_flow_unlock_firmware_error(hass):
|
|||||||
DOMAIN, context={"source": config_entries.SOURCE_USER}
|
DOMAIN, context={"source": config_entries.SOURCE_USER}
|
||||||
)
|
)
|
||||||
|
|
||||||
with patch("broadlink.discover", return_value=[mock_api]):
|
with patch(DEVICE_DISCOVERY, return_value=[mock_api]):
|
||||||
result = await hass.config_entries.flow.async_configure(
|
result = await hass.config_entries.flow.async_configure(
|
||||||
result["flow_id"],
|
result["flow_id"],
|
||||||
{"host": device.host, "timeout": device.timeout},
|
{"host": device.host, "timeout": device.timeout},
|
||||||
@ -446,7 +449,7 @@ async def test_flow_unlock_network_unreachable(hass):
|
|||||||
DOMAIN, context={"source": config_entries.SOURCE_USER}
|
DOMAIN, context={"source": config_entries.SOURCE_USER}
|
||||||
)
|
)
|
||||||
|
|
||||||
with patch("broadlink.discover", return_value=[mock_api]):
|
with patch(DEVICE_DISCOVERY, return_value=[mock_api]):
|
||||||
result = await hass.config_entries.flow.async_configure(
|
result = await hass.config_entries.flow.async_configure(
|
||||||
result["flow_id"],
|
result["flow_id"],
|
||||||
{"host": device.host, "timeout": device.timeout},
|
{"host": device.host, "timeout": device.timeout},
|
||||||
@ -473,7 +476,7 @@ async def test_flow_unlock_os_error(hass):
|
|||||||
DOMAIN, context={"source": config_entries.SOURCE_USER}
|
DOMAIN, context={"source": config_entries.SOURCE_USER}
|
||||||
)
|
)
|
||||||
|
|
||||||
with patch("broadlink.discover", return_value=[mock_api]):
|
with patch(DEVICE_DISCOVERY, return_value=[mock_api]):
|
||||||
result = await hass.config_entries.flow.async_configure(
|
result = await hass.config_entries.flow.async_configure(
|
||||||
result["flow_id"],
|
result["flow_id"],
|
||||||
{"host": device.host, "timeout": device.timeout},
|
{"host": device.host, "timeout": device.timeout},
|
||||||
@ -499,7 +502,7 @@ async def test_flow_do_not_unlock(hass):
|
|||||||
DOMAIN, context={"source": config_entries.SOURCE_USER}
|
DOMAIN, context={"source": config_entries.SOURCE_USER}
|
||||||
)
|
)
|
||||||
|
|
||||||
with patch("broadlink.discover", return_value=[mock_api]):
|
with patch(DEVICE_DISCOVERY, return_value=[mock_api]):
|
||||||
result = await hass.config_entries.flow.async_configure(
|
result = await hass.config_entries.flow.async_configure(
|
||||||
result["flow_id"],
|
result["flow_id"],
|
||||||
{"host": device.host, "timeout": device.timeout},
|
{"host": device.host, "timeout": device.timeout},
|
||||||
@ -527,7 +530,7 @@ async def test_flow_import_works(hass):
|
|||||||
device = get_device("Living Room")
|
device = get_device("Living Room")
|
||||||
mock_api = device.get_mock_api()
|
mock_api = device.get_mock_api()
|
||||||
|
|
||||||
with patch("broadlink.discover", return_value=[mock_api]) as mock_discover:
|
with patch(DEVICE_DISCOVERY, return_value=[mock_api]) as mock_discover:
|
||||||
result = await hass.config_entries.flow.async_init(
|
result = await hass.config_entries.flow.async_init(
|
||||||
DOMAIN,
|
DOMAIN,
|
||||||
context={"source": config_entries.SOURCE_IMPORT},
|
context={"source": config_entries.SOURCE_IMPORT},
|
||||||
@ -558,12 +561,12 @@ async def test_flow_import_already_in_progress(hass):
|
|||||||
device = get_device("Living Room")
|
device = get_device("Living Room")
|
||||||
data = {"host": device.host}
|
data = {"host": device.host}
|
||||||
|
|
||||||
with patch("broadlink.discover", return_value=[device.get_mock_api()]):
|
with patch(DEVICE_DISCOVERY, return_value=[device.get_mock_api()]):
|
||||||
result = await hass.config_entries.flow.async_init(
|
result = await hass.config_entries.flow.async_init(
|
||||||
DOMAIN, context={"source": config_entries.SOURCE_IMPORT}, data=data
|
DOMAIN, context={"source": config_entries.SOURCE_IMPORT}, data=data
|
||||||
)
|
)
|
||||||
|
|
||||||
with patch("broadlink.discover", return_value=[device.get_mock_api()]):
|
with patch(DEVICE_DISCOVERY, return_value=[device.get_mock_api()]):
|
||||||
result = await hass.config_entries.flow.async_init(
|
result = await hass.config_entries.flow.async_init(
|
||||||
DOMAIN, context={"source": config_entries.SOURCE_IMPORT}, data=data
|
DOMAIN, context={"source": config_entries.SOURCE_IMPORT}, data=data
|
||||||
)
|
)
|
||||||
@ -579,7 +582,7 @@ async def test_flow_import_host_already_configured(hass):
|
|||||||
mock_entry.add_to_hass(hass)
|
mock_entry.add_to_hass(hass)
|
||||||
mock_api = device.get_mock_api()
|
mock_api = device.get_mock_api()
|
||||||
|
|
||||||
with patch("broadlink.discover", return_value=[mock_api]):
|
with patch(DEVICE_DISCOVERY, return_value=[mock_api]):
|
||||||
result = await hass.config_entries.flow.async_init(
|
result = await hass.config_entries.flow.async_init(
|
||||||
DOMAIN,
|
DOMAIN,
|
||||||
context={"source": config_entries.SOURCE_IMPORT},
|
context={"source": config_entries.SOURCE_IMPORT},
|
||||||
@ -602,7 +605,7 @@ async def test_flow_import_mac_already_configured(hass):
|
|||||||
device.host = "192.168.1.16"
|
device.host = "192.168.1.16"
|
||||||
mock_api = device.get_mock_api()
|
mock_api = device.get_mock_api()
|
||||||
|
|
||||||
with patch("broadlink.discover", return_value=[mock_api]):
|
with patch(DEVICE_DISCOVERY, return_value=[mock_api]):
|
||||||
result = await hass.config_entries.flow.async_init(
|
result = await hass.config_entries.flow.async_init(
|
||||||
DOMAIN,
|
DOMAIN,
|
||||||
context={"source": config_entries.SOURCE_IMPORT},
|
context={"source": config_entries.SOURCE_IMPORT},
|
||||||
@ -620,7 +623,7 @@ async def test_flow_import_mac_already_configured(hass):
|
|||||||
|
|
||||||
async def test_flow_import_device_not_found(hass):
|
async def test_flow_import_device_not_found(hass):
|
||||||
"""Test we handle a device not found in the import step."""
|
"""Test we handle a device not found in the import step."""
|
||||||
with patch("broadlink.discover", return_value=[]):
|
with patch(DEVICE_DISCOVERY, return_value=[]):
|
||||||
result = await hass.config_entries.flow.async_init(
|
result = await hass.config_entries.flow.async_init(
|
||||||
DOMAIN,
|
DOMAIN,
|
||||||
context={"source": config_entries.SOURCE_IMPORT},
|
context={"source": config_entries.SOURCE_IMPORT},
|
||||||
@ -633,7 +636,7 @@ async def test_flow_import_device_not_found(hass):
|
|||||||
|
|
||||||
async def test_flow_import_invalid_ip_address(hass):
|
async def test_flow_import_invalid_ip_address(hass):
|
||||||
"""Test we handle an invalid IP address in the import step."""
|
"""Test we handle an invalid IP address in the import step."""
|
||||||
with patch("broadlink.discover", side_effect=OSError(errno.EINVAL, None)):
|
with patch(DEVICE_DISCOVERY, side_effect=OSError(errno.EINVAL, None)):
|
||||||
result = await hass.config_entries.flow.async_init(
|
result = await hass.config_entries.flow.async_init(
|
||||||
DOMAIN,
|
DOMAIN,
|
||||||
context={"source": config_entries.SOURCE_IMPORT},
|
context={"source": config_entries.SOURCE_IMPORT},
|
||||||
@ -646,7 +649,7 @@ async def test_flow_import_invalid_ip_address(hass):
|
|||||||
|
|
||||||
async def test_flow_import_invalid_hostname(hass):
|
async def test_flow_import_invalid_hostname(hass):
|
||||||
"""Test we handle an invalid hostname in the import step."""
|
"""Test we handle an invalid hostname in the import step."""
|
||||||
with patch("broadlink.discover", side_effect=OSError(socket.EAI_NONAME, None)):
|
with patch(DEVICE_DISCOVERY, side_effect=OSError(socket.EAI_NONAME, None)):
|
||||||
result = await hass.config_entries.flow.async_init(
|
result = await hass.config_entries.flow.async_init(
|
||||||
DOMAIN,
|
DOMAIN,
|
||||||
context={"source": config_entries.SOURCE_IMPORT},
|
context={"source": config_entries.SOURCE_IMPORT},
|
||||||
@ -659,7 +662,7 @@ async def test_flow_import_invalid_hostname(hass):
|
|||||||
|
|
||||||
async def test_flow_import_network_unreachable(hass):
|
async def test_flow_import_network_unreachable(hass):
|
||||||
"""Test we handle a network unreachable in the import step."""
|
"""Test we handle a network unreachable in the import step."""
|
||||||
with patch("broadlink.discover", side_effect=OSError(errno.ENETUNREACH, None)):
|
with patch(DEVICE_DISCOVERY, side_effect=OSError(errno.ENETUNREACH, None)):
|
||||||
result = await hass.config_entries.flow.async_init(
|
result = await hass.config_entries.flow.async_init(
|
||||||
DOMAIN,
|
DOMAIN,
|
||||||
context={"source": config_entries.SOURCE_IMPORT},
|
context={"source": config_entries.SOURCE_IMPORT},
|
||||||
@ -672,7 +675,7 @@ async def test_flow_import_network_unreachable(hass):
|
|||||||
|
|
||||||
async def test_flow_import_os_error(hass):
|
async def test_flow_import_os_error(hass):
|
||||||
"""Test we handle an OS error in the import step."""
|
"""Test we handle an OS error in the import step."""
|
||||||
with patch("broadlink.discover", side_effect=OSError()):
|
with patch(DEVICE_DISCOVERY, side_effect=OSError()):
|
||||||
result = await hass.config_entries.flow.async_init(
|
result = await hass.config_entries.flow.async_init(
|
||||||
DOMAIN,
|
DOMAIN,
|
||||||
context={"source": config_entries.SOURCE_IMPORT},
|
context={"source": config_entries.SOURCE_IMPORT},
|
||||||
@ -692,7 +695,7 @@ async def test_flow_reauth_works(hass):
|
|||||||
mock_api.auth.side_effect = blke.AuthenticationError()
|
mock_api.auth.side_effect = blke.AuthenticationError()
|
||||||
data = {"name": device.name, **device.get_entry_data()}
|
data = {"name": device.name, **device.get_entry_data()}
|
||||||
|
|
||||||
with patch("broadlink.gendevice", return_value=mock_api):
|
with patch(DEVICE_FACTORY, return_value=mock_api):
|
||||||
result = await hass.config_entries.flow.async_init(
|
result = await hass.config_entries.flow.async_init(
|
||||||
DOMAIN, context={"source": "reauth"}, data=data
|
DOMAIN, context={"source": "reauth"}, data=data
|
||||||
)
|
)
|
||||||
@ -702,7 +705,7 @@ async def test_flow_reauth_works(hass):
|
|||||||
|
|
||||||
mock_api = device.get_mock_api()
|
mock_api = device.get_mock_api()
|
||||||
|
|
||||||
with patch("broadlink.discover", return_value=[mock_api]) as mock_discover:
|
with patch(DEVICE_DISCOVERY, return_value=[mock_api]) as mock_discover:
|
||||||
result = await hass.config_entries.flow.async_configure(
|
result = await hass.config_entries.flow.async_configure(
|
||||||
result["flow_id"],
|
result["flow_id"],
|
||||||
{"host": device.host, "timeout": device.timeout},
|
{"host": device.host, "timeout": device.timeout},
|
||||||
@ -728,7 +731,7 @@ async def test_flow_reauth_invalid_host(hass):
|
|||||||
mock_api.auth.side_effect = blke.AuthenticationError()
|
mock_api.auth.side_effect = blke.AuthenticationError()
|
||||||
data = {"name": device.name, **device.get_entry_data()}
|
data = {"name": device.name, **device.get_entry_data()}
|
||||||
|
|
||||||
with patch("broadlink.gendevice", return_value=mock_api):
|
with patch(DEVICE_FACTORY, return_value=mock_api):
|
||||||
result = await hass.config_entries.flow.async_init(
|
result = await hass.config_entries.flow.async_init(
|
||||||
DOMAIN, context={"source": "reauth"}, data=data
|
DOMAIN, context={"source": "reauth"}, data=data
|
||||||
)
|
)
|
||||||
@ -736,7 +739,7 @@ async def test_flow_reauth_invalid_host(hass):
|
|||||||
device.mac = get_device("Office").mac
|
device.mac = get_device("Office").mac
|
||||||
mock_api = device.get_mock_api()
|
mock_api = device.get_mock_api()
|
||||||
|
|
||||||
with patch("broadlink.discover", return_value=[mock_api]) as mock_discover:
|
with patch(DEVICE_DISCOVERY, return_value=[mock_api]) as mock_discover:
|
||||||
result = await hass.config_entries.flow.async_configure(
|
result = await hass.config_entries.flow.async_configure(
|
||||||
result["flow_id"],
|
result["flow_id"],
|
||||||
{"host": device.host, "timeout": device.timeout},
|
{"host": device.host, "timeout": device.timeout},
|
||||||
@ -762,7 +765,7 @@ async def test_flow_reauth_valid_host(hass):
|
|||||||
mock_api.auth.side_effect = blke.AuthenticationError()
|
mock_api.auth.side_effect = blke.AuthenticationError()
|
||||||
data = {"name": device.name, **device.get_entry_data()}
|
data = {"name": device.name, **device.get_entry_data()}
|
||||||
|
|
||||||
with patch("broadlink.gendevice", return_value=mock_api):
|
with patch(DEVICE_FACTORY, return_value=mock_api):
|
||||||
result = await hass.config_entries.flow.async_init(
|
result = await hass.config_entries.flow.async_init(
|
||||||
DOMAIN, context={"source": "reauth"}, data=data
|
DOMAIN, context={"source": "reauth"}, data=data
|
||||||
)
|
)
|
||||||
@ -770,7 +773,7 @@ async def test_flow_reauth_valid_host(hass):
|
|||||||
device.host = "192.168.1.128"
|
device.host = "192.168.1.128"
|
||||||
mock_api = device.get_mock_api()
|
mock_api = device.get_mock_api()
|
||||||
|
|
||||||
with patch("broadlink.discover", return_value=[mock_api]) as mock_discover:
|
with patch(DEVICE_DISCOVERY, return_value=[mock_api]) as mock_discover:
|
||||||
result = await hass.config_entries.flow.async_configure(
|
result = await hass.config_entries.flow.async_configure(
|
||||||
result["flow_id"],
|
result["flow_id"],
|
||||||
{"host": device.host, "timeout": device.timeout},
|
{"host": device.host, "timeout": device.timeout},
|
||||||
|
Loading…
Reference in New Issue
Block a user