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

Mock setup in sonarr config flow tests (#37432)

Browse source
This commit is contained in:
J. Nick Koston 2020-07-03 16:20:17 -05:00 • committed by GitHub
parent 1aafa459e5
commit 76b956a969
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 21 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

31
tests/components/sonarr/test_config_flow.py
View file

@ -38,10 +38,13 @@ async def test_options(hass, aioclient_mock: AiohttpClientMocker):
assert result["type"] == RESULT_TYPE_FORM
assert result["step_id"] == "init"
result = await hass.config_entries.options.async_configure(
result["flow_id"],
user_input={CONF_UPCOMING_DAYS: 2, CONF_WANTED_MAX_ITEMS: 100},
)
with patch(
"homeassistant.components.sonarr.async_setup_entry", return_value=True
), patch("homeassistant.components.sonarr.async_setup", return_value=True):
result = await hass.config_entries.options.async_configure(
result["flow_id"],
user_input={CONF_UPCOMING_DAYS: 2, CONF_WANTED_MAX_ITEMS: 100},
)
assert result["type"] == RESULT_TYPE_CREATE_ENTRY
assert result["data"][CONF_UPCOMING_DAYS] == 2
@ -114,6 +117,7 @@ async def test_full_import_flow_implementation(
mock_connection(aioclient_mock)
user_input = MOCK_USER_INPUT.copy()
result = await hass.config_entries.flow.async_init(
DOMAIN, context={CONF_SOURCE: SOURCE_IMPORT}, data=user_input,
)
@ -143,9 +147,12 @@ async def test_full_user_flow_implementation(
assert result["step_id"] == "user"
user_input = MOCK_USER_INPUT.copy()
result = await hass.config_entries.flow.async_configure(
result["flow_id"], user_input=user_input,
)
with patch(
"homeassistant.components.sonarr.async_setup_entry", return_value=True
), patch("homeassistant.components.sonarr.async_setup", return_value=True):
result = await hass.config_entries.flow.async_configure(
result["flow_id"], user_input=user_input,
)
assert result["type"] == RESULT_TYPE_CREATE_ENTRY
assert result["title"] == HOST
@ -171,9 +178,13 @@ async def test_full_user_flow_advanced_options(
**MOCK_USER_INPUT,
CONF_VERIFY_SSL: True,
}
result = await hass.config_entries.flow.async_configure(
result["flow_id"], user_input=user_input,
)
with patch(
"homeassistant.components.sonarr.async_setup_entry", return_value=True
), patch("homeassistant.components.sonarr.async_setup", return_value=True):
result = await hass.config_entries.flow.async_configure(
result["flow_id"], user_input=user_input,
)
assert result["type"] == RESULT_TYPE_CREATE_ENTRY
assert result["title"] == HOST