jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-23 13:17:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use FlowResultType enum in Plugwise tests (#74638)

Browse Source
This commit is contained in:
Franck Nijhof 2022-07-07 17:25:23 +02:00 committed by GitHub
parent c01f7d75d5
commit 323d4a0e1b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
tests/components/plugwise/test_config_flow.py
View File

@ -21,11 +21,7 @@ from homeassistant.const import (
CONF_USERNAME, CONF_USERNAME,
) )
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.data_entry_flow import ( from homeassistant.data_entry_flow import FlowResultType
RESULT_TYPE_ABORT,
RESULT_TYPE_CREATE_ENTRY,
RESULT_TYPE_FORM,
)
from tests.common import MockConfigEntry from tests.common import MockConfigEntry
@ -88,7 +84,7 @@ async def test_form(
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
DOMAIN, context={CONF_SOURCE: SOURCE_USER} DOMAIN, context={CONF_SOURCE: SOURCE_USER}
) )
assert result.get("type") == RESULT_TYPE_FORM assert result.get("type") == FlowResultType.FORM
assert result.get("errors") == {} assert result.get("errors") == {}
assert result.get("step_id") == "user" assert result.get("step_id") == "user"
assert "flow_id" in result assert "flow_id" in result
@ -102,7 +98,7 @@ async def test_form(
) )
await hass.async_block_till_done() await hass.async_block_till_done()
assert result2.get("type") == RESULT_TYPE_CREATE_ENTRY assert result2.get("type") == FlowResultType.CREATE_ENTRY
assert result2.get("title") == "Test Smile Name" assert result2.get("title") == "Test Smile Name"
assert result2.get("data") == { assert result2.get("data") == {
CONF_HOST: TEST_HOST, CONF_HOST: TEST_HOST,
@ -136,7 +132,7 @@ async def test_zeroconf_flow(
context={CONF_SOURCE: SOURCE_ZEROCONF}, context={CONF_SOURCE: SOURCE_ZEROCONF},
data=discovery, data=discovery,
) )
assert result.get("type") == RESULT_TYPE_FORM assert result.get("type") == FlowResultType.FORM
assert result.get("errors") == {} assert result.get("errors") == {}
assert result.get("step_id") == "user" assert result.get("step_id") == "user"
assert "flow_id" in result assert "flow_id" in result
@ -147,7 +143,7 @@ async def test_zeroconf_flow(
) )
await hass.async_block_till_done() await hass.async_block_till_done()
assert result2.get("type") == RESULT_TYPE_CREATE_ENTRY assert result2.get("type") == FlowResultType.CREATE_ENTRY
assert result2.get("title") == "Test Smile Name" assert result2.get("title") == "Test Smile Name"
assert result2.get("data") == { assert result2.get("data") == {
CONF_HOST: TEST_HOST, CONF_HOST: TEST_HOST,
@ -172,7 +168,7 @@ async def test_zeroconf_flow_stretch(
context={CONF_SOURCE: SOURCE_ZEROCONF}, context={CONF_SOURCE: SOURCE_ZEROCONF},
data=TEST_DISCOVERY2, data=TEST_DISCOVERY2,
) )
assert result.get("type") == RESULT_TYPE_FORM assert result.get("type") == FlowResultType.FORM
assert result.get("errors") == {} assert result.get("errors") == {}
assert result.get("step_id") == "user" assert result.get("step_id") == "user"
assert "flow_id" in result assert "flow_id" in result
@ -183,7 +179,7 @@ async def test_zeroconf_flow_stretch(
) )
await hass.async_block_till_done() await hass.async_block_till_done()
assert result2.get("type") == RESULT_TYPE_CREATE_ENTRY assert result2.get("type") == FlowResultType.CREATE_ENTRY
assert result2.get("title") == "Test Smile Name" assert result2.get("title") == "Test Smile Name"
assert result2.get("data") == { assert result2.get("data") == {
CONF_HOST: TEST_HOST, CONF_HOST: TEST_HOST,
@ -224,7 +220,7 @@ async def test_zercoconf_discovery_update_configuration(
context={CONF_SOURCE: SOURCE_ZEROCONF}, context={CONF_SOURCE: SOURCE_ZEROCONF},
data=TEST_DISCOVERY, data=TEST_DISCOVERY,
) )
assert result.get("type") == RESULT_TYPE_ABORT assert result.get("type") == FlowResultType.ABORT
assert result.get("reason") == "already_configured" assert result.get("reason") == "already_configured"
assert entry.data[CONF_HOST] == "0.0.0.0" assert entry.data[CONF_HOST] == "0.0.0.0"
@ -235,7 +231,7 @@ async def test_zercoconf_discovery_update_configuration(
data=TEST_DISCOVERY, data=TEST_DISCOVERY,
) )
assert result.get("type") == RESULT_TYPE_ABORT assert result.get("type") == FlowResultType.ABORT
assert result.get("reason") == "already_configured" assert result.get("reason") == "already_configured"
assert entry.data[CONF_HOST] == "1.1.1.1" assert entry.data[CONF_HOST] == "1.1.1.1"
@ -262,7 +258,7 @@ async def test_flow_errors(
DOMAIN, DOMAIN,
context={CONF_SOURCE: SOURCE_USER}, context={CONF_SOURCE: SOURCE_USER},
) )
assert result.get("type") == RESULT_TYPE_FORM assert result.get("type") == FlowResultType.FORM
assert result.get("errors") == {} assert result.get("errors") == {}
assert result.get("step_id") == "user" assert result.get("step_id") == "user"
assert "flow_id" in result assert "flow_id" in result
@ -273,7 +269,7 @@ async def test_flow_errors(
user_input={CONF_HOST: TEST_HOST, CONF_PASSWORD: TEST_PASSWORD}, user_input={CONF_HOST: TEST_HOST, CONF_PASSWORD: TEST_PASSWORD},
) )
assert result2.get("type") == RESULT_TYPE_FORM assert result2.get("type") == FlowResultType.FORM
assert result2.get("errors") == {"base": reason} assert result2.get("errors") == {"base": reason}
assert result2.get("step_id") == "user" assert result2.get("step_id") == "user"
@ -286,7 +282,7 @@ async def test_flow_errors(
user_input={CONF_HOST: TEST_HOST, CONF_PASSWORD: TEST_PASSWORD}, user_input={CONF_HOST: TEST_HOST, CONF_PASSWORD: TEST_PASSWORD},
) )
assert result3.get("type") == RESULT_TYPE_CREATE_ENTRY assert result3.get("type") == FlowResultType.CREATE_ENTRY
assert result3.get("title") == "Test Smile Name" assert result3.get("title") == "Test Smile Name"
assert result3.get("data") == { assert result3.get("data") == {
CONF_HOST: TEST_HOST, CONF_HOST: TEST_HOST,