evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix flaky ONVIF test (#41848)

Browse source
This commit is contained in:
Paulus Schoutsen 2020-10-15 09:19:08 +02:00 • committed by GitHub
parent 6ccdf62e6b
commit 4899d7628a
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 37 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

54
tests/components/onvif/test_config_flow.py
View file

@ -210,15 +210,22 @@ async def test_flow_discovered_devices(hass):
assert result["type"] == data_entry_flow.RESULT_TYPE_FORM assert result["type"] == data_entry_flow.RESULT_TYPE_FORM
assert result["step_id"] == "auth" assert result["step_id"] == "auth"
result = await hass.config_entries.flow.async_configure( with patch(
result["flow_id"], "homeassistant.components.onvif.async_setup", return_value=True
user_input={ ) as mock_setup, patch(
config_flow.CONF_USERNAME: USERNAME, "homeassistant.components.onvif.async_setup_entry", return_value=True
config_flow.CONF_PASSWORD: PASSWORD, ) as mock_setup_entry:
}, result = await hass.config_entries.flow.async_configure(
) result["flow_id"],
user_input={
config_flow.CONF_USERNAME: USERNAME,
config_flow.CONF_PASSWORD: PASSWORD,
},
)
await hass.async_block_till_done() await hass.async_block_till_done()
assert len(mock_setup.mock_calls) == 1
assert len(mock_setup_entry.mock_calls) == 1
assert result["type"] == data_entry_flow.RESULT_TYPE_CREATE_ENTRY assert result["type"] == data_entry_flow.RESULT_TYPE_CREATE_ENTRY
assert result["title"] == f"{URN} - {MAC}" assert result["title"] == f"{URN} - {MAC}"
@ -377,15 +384,22 @@ async def test_flow_manual_entry(hass):
assert result["type"] == data_entry_flow.RESULT_TYPE_FORM assert result["type"] == data_entry_flow.RESULT_TYPE_FORM
assert result["step_id"] == "auth" assert result["step_id"] == "auth"
result = await hass.config_entries.flow.async_configure( with patch(
result["flow_id"], "homeassistant.components.onvif.async_setup", return_value=True
user_input={ ) as mock_setup, patch(
config_flow.CONF_USERNAME: USERNAME, "homeassistant.components.onvif.async_setup_entry", return_value=True
config_flow.CONF_PASSWORD: PASSWORD, ) as mock_setup_entry:
}, result = await hass.config_entries.flow.async_configure(
) result["flow_id"],
user_input={
config_flow.CONF_USERNAME: USERNAME,
config_flow.CONF_PASSWORD: PASSWORD,
},
)
await hass.async_block_till_done() await hass.async_block_till_done()
assert len(mock_setup.mock_calls) == 1
assert len(mock_setup_entry.mock_calls) == 1
assert result["type"] == data_entry_flow.RESULT_TYPE_CREATE_ENTRY assert result["type"] == data_entry_flow.RESULT_TYPE_CREATE_ENTRY
assert result["title"] == f"{NAME} - {MAC}" assert result["title"] == f"{NAME} - {MAC}"
@ -404,7 +418,11 @@ async def test_flow_import_no_mac(hass):
"homeassistant.components.onvif.config_flow.get_device" "homeassistant.components.onvif.config_flow.get_device"
) as mock_onvif_camera, patch( ) as mock_onvif_camera, patch(
"homeassistant.components.onvif.ONVIFDevice" "homeassistant.components.onvif.ONVIFDevice"
) as mock_device: ) as mock_device, patch(
"homeassistant.components.onvif.async_setup", return_value=True
) as mock_setup, patch(
"homeassistant.components.onvif.async_setup_entry", return_value=True
) as mock_setup_entry:
setup_mock_onvif_camera(mock_onvif_camera, with_interfaces=False) setup_mock_onvif_camera(mock_onvif_camera, with_interfaces=False)
setup_mock_device(mock_device) setup_mock_device(mock_device)
@ -421,6 +439,8 @@ async def test_flow_import_no_mac(hass):
) )
await hass.async_block_till_done() await hass.async_block_till_done()
assert len(mock_setup.mock_calls) == 1
assert len(mock_setup_entry.mock_calls) == 1
assert result["type"] == data_entry_flow.RESULT_TYPE_CREATE_ENTRY assert result["type"] == data_entry_flow.RESULT_TYPE_CREATE_ENTRY
assert result["title"] == f"{NAME} - {SERIAL_NUMBER}" assert result["title"] == f"{NAME} - {SERIAL_NUMBER}"