Fix unload race in unifiprotect tests (#81361)

This commit is contained in:
J. Nick Koston 2022-11-01 13:21:20 -05:00 committed by GitHub
parent 5c99e2e5d3
commit 914ccdbc4f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -254,23 +254,28 @@ async def test_form_options(hass: HomeAssistant, ufp_client: ProtectApiClient) -
await hass.async_block_till_done() await hass.async_block_till_done()
assert mock_config.state == config_entries.ConfigEntryState.LOADED assert mock_config.state == config_entries.ConfigEntryState.LOADED
result = await hass.config_entries.options.async_init(mock_config.entry_id) result = await hass.config_entries.options.async_init(mock_config.entry_id)
assert result["type"] == FlowResultType.FORM assert result["type"] == FlowResultType.FORM
assert not result["errors"] assert not result["errors"]
assert result["step_id"] == "init" assert result["step_id"] == "init"
result2 = await hass.config_entries.options.async_configure( result2 = await hass.config_entries.options.async_configure(
result["flow_id"], result["flow_id"],
{CONF_DISABLE_RTSP: True, CONF_ALL_UPDATES: True, CONF_OVERRIDE_CHOST: True}, {
) CONF_DISABLE_RTSP: True,
CONF_ALL_UPDATES: True,
CONF_OVERRIDE_CHOST: True,
},
)
assert result2["type"] == FlowResultType.CREATE_ENTRY assert result2["type"] == FlowResultType.CREATE_ENTRY
assert result2["data"] == { assert result2["data"] == {
"all_updates": True, "all_updates": True,
"disable_rtsp": True, "disable_rtsp": True,
"override_connection_host": True, "override_connection_host": True,
"max_media": 1000, "max_media": 1000,
} }
await hass.config_entries.async_unload(mock_config.entry_id)
@pytest.mark.parametrize( @pytest.mark.parametrize(