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

Clean twentemilieu config flow tests (#50460)

Browse source
This commit is contained in:
Martin Hjelmare 2021-05-11 15:53:36 +02:00 • committed by GitHub
parent 1538271555
commit 48b5ef0bac
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 25 additions and 21 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

46
tests/components/twentemilieu/test_config_flow.py
View file

@ -2,13 +2,13 @@
import aiohttp import aiohttp
from homeassistant import data_entry_flow from homeassistant import data_entry_flow
from homeassistant.components.twentemilieu import config_flow
from homeassistant.components.twentemilieu.const import ( from homeassistant.components.twentemilieu.const import (
CONF_HOUSE_LETTER, CONF_HOUSE_LETTER,
CONF_HOUSE_NUMBER, CONF_HOUSE_NUMBER,
CONF_POST_CODE, CONF_POST_CODE,
DOMAIN, DOMAIN,
) )
from homeassistant.config_entries import SOURCE_USER
from homeassistant.const import CONF_ID, CONTENT_TYPE_JSON from homeassistant.const import CONF_ID, CONTENT_TYPE_JSON
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
@ -16,7 +16,6 @@ from tests.common import MockConfigEntry
from tests.test_util.aiohttp import AiohttpClientMocker from tests.test_util.aiohttp import AiohttpClientMocker
FIXTURE_USER_INPUT = { FIXTURE_USER_INPUT = {
CONF_ID: "12345",
CONF_POST_CODE: "1234AB", CONF_POST_CODE: "1234AB",
CONF_HOUSE_NUMBER: "1", CONF_HOUSE_NUMBER: "1",
CONF_HOUSE_LETTER: "A", CONF_HOUSE_LETTER: "A",
@ -25,9 +24,9 @@ FIXTURE_USER_INPUT = {
async def test_show_set_form(hass: HomeAssistant) -> None: async def test_show_set_form(hass: HomeAssistant) -> None:
"""Test that the setup form is served.""" """Test that the setup form is served."""
flow = config_flow.TwenteMilieuFlowHandler() result = await hass.config_entries.flow.async_init(
flow.hass = hass DOMAIN, context={"source": SOURCE_USER}
result = await flow.async_step_user(user_input=None) )
assert result["type"] == data_entry_flow.RESULT_TYPE_FORM assert result["type"] == data_entry_flow.RESULT_TYPE_FORM
assert result["step_id"] == "user" assert result["step_id"] == "user"
@ -41,9 +40,9 @@ async def test_connection_error(
"https://twentemilieuapi.ximmio.com/api/FetchAdress", exc=aiohttp.ClientError "https://twentemilieuapi.ximmio.com/api/FetchAdress", exc=aiohttp.ClientError
) )
flow = config_flow.TwenteMilieuFlowHandler() result = await hass.config_entries.flow.async_init(
flow.hass = hass DOMAIN, context={"source": SOURCE_USER}, data=FIXTURE_USER_INPUT
result = await flow.async_step_user(user_input=FIXTURE_USER_INPUT) )
assert result["type"] == data_entry_flow.RESULT_TYPE_FORM assert result["type"] == data_entry_flow.RESULT_TYPE_FORM
assert result["step_id"] == "user" assert result["step_id"] == "user"
@ -60,9 +59,9 @@ async def test_invalid_address(
headers={"Content-Type": CONTENT_TYPE_JSON}, headers={"Content-Type": CONTENT_TYPE_JSON},
) )
flow = config_flow.TwenteMilieuFlowHandler() result = await hass.config_entries.flow.async_init(
flow.hass = hass DOMAIN, context={"source": SOURCE_USER}, data=FIXTURE_USER_INPUT
result = await flow.async_step_user(user_input=FIXTURE_USER_INPUT) )
assert result["type"] == data_entry_flow.RESULT_TYPE_FORM assert result["type"] == data_entry_flow.RESULT_TYPE_FORM
assert result["step_id"] == "user" assert result["step_id"] == "user"
@ -73,9 +72,9 @@ async def test_address_already_set_up(
hass: HomeAssistant, aioclient_mock: AiohttpClientMocker hass: HomeAssistant, aioclient_mock: AiohttpClientMocker
) -> None: ) -> None:
"""Test we abort if address has already been set up.""" """Test we abort if address has already been set up."""
MockConfigEntry(domain=DOMAIN, data=FIXTURE_USER_INPUT, title="12345").add_to_hass( MockConfigEntry(
hass domain=DOMAIN, data={**FIXTURE_USER_INPUT, CONF_ID: "12345"}, title="12345"
) ).add_to_hass(hass)
aioclient_mock.post( aioclient_mock.post(
"https://twentemilieuapi.ximmio.com/api/FetchAdress", "https://twentemilieuapi.ximmio.com/api/FetchAdress",
@ -83,9 +82,9 @@ async def test_address_already_set_up(
headers={"Content-Type": CONTENT_TYPE_JSON}, headers={"Content-Type": CONTENT_TYPE_JSON},
) )
flow = config_flow.TwenteMilieuFlowHandler() result = await hass.config_entries.flow.async_init(
flow.hass = hass DOMAIN, context={"source": SOURCE_USER}, data=FIXTURE_USER_INPUT
result = await flow.async_step_user(user_input=FIXTURE_USER_INPUT) )
assert result["type"] == data_entry_flow.RESULT_TYPE_ABORT assert result["type"] == data_entry_flow.RESULT_TYPE_ABORT
assert result["reason"] == "already_configured" assert result["reason"] == "already_configured"
@ -101,13 +100,18 @@ async def test_full_flow_implementation(
headers={"Content-Type": CONTENT_TYPE_JSON}, headers={"Content-Type": CONTENT_TYPE_JSON},
) )
flow = config_flow.TwenteMilieuFlowHandler() result = await hass.config_entries.flow.async_init(
flow.hass = hass DOMAIN, context={"source": SOURCE_USER}
result = await flow.async_step_user(user_input=None) )
assert result["type"] == data_entry_flow.RESULT_TYPE_FORM assert result["type"] == data_entry_flow.RESULT_TYPE_FORM
assert result["step_id"] == "user" assert result["step_id"] == "user"
result = await flow.async_step_user(user_input=FIXTURE_USER_INPUT) result = await hass.config_entries.flow.async_configure(
result["flow_id"],
FIXTURE_USER_INPUT,
)
assert result["type"] == data_entry_flow.RESULT_TYPE_CREATE_ENTRY assert result["type"] == data_entry_flow.RESULT_TYPE_CREATE_ENTRY
assert result["title"] == "12345" assert result["title"] == "12345"
assert result["data"][CONF_POST_CODE] == FIXTURE_USER_INPUT[CONF_POST_CODE] assert result["data"][CONF_POST_CODE] == FIXTURE_USER_INPUT[CONF_POST_CODE]