Fix call to API in airnow option flow tests (#101457)

This commit is contained in:
Erik Montnemery 2023-10-05 13:23:59 +02:00 committed by GitHub
parent 5975974a37
commit 2464232f24
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -1,5 +1,5 @@
"""Test the AirNow config flow.""" """Test the AirNow config flow."""
from unittest.mock import AsyncMock from unittest.mock import AsyncMock, patch
from pyairnow.errors import AirNowError, EmptyResponseError, InvalidKeyError from pyairnow.errors import AirNowError, EmptyResponseError, InvalidKeyError
import pytest import pytest
@ -142,12 +142,18 @@ async def test_options_flow(hass: HomeAssistant, setup_airnow) -> None:
assert result["type"] == data_entry_flow.FlowResultType.FORM assert result["type"] == data_entry_flow.FlowResultType.FORM
assert result["step_id"] == "init" assert result["step_id"] == "init"
result = await hass.config_entries.options.async_configure( with patch(
result["flow_id"], "homeassistant.components.airnow.async_setup_entry",
user_input={CONF_RADIUS: 25}, return_value=True,
) ) as mock_setup_entry:
result = await hass.config_entries.options.async_configure(
result["flow_id"],
user_input={CONF_RADIUS: 25},
)
await hass.async_block_till_done()
assert result["type"] == data_entry_flow.FlowResultType.CREATE_ENTRY assert result["type"] == data_entry_flow.FlowResultType.CREATE_ENTRY
assert config_entry.options == { assert config_entry.options == {
CONF_RADIUS: 25, CONF_RADIUS: 25,
} }
assert len(mock_setup_entry.mock_calls) == 1