home-assistant/core
1
0
Fork 1
mirror of https://github.com/home-assistant/core.git synced 2025-08-07 22:55:10 +02:00
Code Issues Packages Projects Releases Wiki Activity

Deduplicate tests

Browse Source
This commit is contained in:
Erik
2025-04-07 14:23:19 +02:00
parent 1aa2fb28e7
commit d535c688fd
1 changed files with 78 additions and 132 deletions
Download Patch File Download Diff File Expand all files Collapse all files

154
tests/components/config/test_config_entries.py
View File

@@ -947,81 +947,55 @@ async def test_get_progress_subscribe(
assert response == {"id": ANY, "result": None, "success": True, "type": "result"}
subscription = response["id"]
flow_context = {
"bluetooth": {"source": core_ce.SOURCE_BLUETOOTH},
"hassio": {"source": core_ce.SOURCE_HASSIO},
"user": {"source": core_ce.SOURCE_USER},
"reauth": {"source": core_ce.SOURCE_REAUTH, "entry_id": "1234"},
"reconfigure": {"source": core_ce.SOURCE_RECONFIGURE, "entry_id": "1234"},
}
forms = {}
with mock_config_flow("test", TestFlow):
form_bluetooth = await hass.config_entries.flow.async_init(
"test", context={"source": core_ce.SOURCE_BLUETOOTH}
)
form_hassio = await hass.config_entries.flow.async_init(
"test", context={"source": core_ce.SOURCE_HASSIO}
)
form_user = await hass.config_entries.flow.async_init(
"test", context={"source": core_ce.SOURCE_USER}
)
form_reauth = await hass.config_entries.flow.async_init(
"test", context={"source": core_ce.SOURCE_REAUTH, "entry_id": "1234"}
)
form_reconfigure = await hass.config_entries.flow.async_init(
"test", context={"source": core_ce.SOURCE_RECONFIGURE, "entry_id": "1234"}
for key, context in flow_context.items():
forms[key] = await hass.config_entries.flow.async_init(
"test", context=context
)
assert form_bluetooth["type"] == data_entry_flow.FlowResultType.ABORT
for form in (form_hassio, form_user, form_reauth, form_reconfigure):
assert form["type"] == data_entry_flow.FlowResultType.FORM
assert form["step_id"] == "account"
assert forms["bluetooth"]["type"] == data_entry_flow.FlowResultType.ABORT
for key in ("hassio", "user", "reauth", "reconfigure"):
assert forms[key]["type"] == data_entry_flow.FlowResultType.FORM
assert forms[key]["step_id"] == "account"
hass.config_entries.flow.async_abort(form_hassio["flow_id"])
hass.config_entries.flow.async_abort(form_user["flow_id"])
hass.config_entries.flow.async_abort(form_reauth["flow_id"])
hass.config_entries.flow.async_abort(form_reconfigure["flow_id"])
for key in ("hassio", "user", "reauth", "reconfigure"):
hass.config_entries.flow.async_abort(forms[key]["flow_id"])
# Uninitialized flows and flows with SOURCE_USER and SOURCE_RECONFIGURE
# should be filtered out
for key in ("hassio", "reauth"):
response = await ws_client.receive_json()
assert response == {
"event": {
"flow": {
"flow_id": form_hassio["flow_id"],
"flow_id": forms[key]["flow_id"],
"handler": "test",
"step_id": "account",
"context": {"source": core_ce.SOURCE_HASSIO},
"context": flow_context[key],
},
"flow_id": form_hassio["flow_id"],
"flow_id": forms[key]["flow_id"],
"type": "init",
},
"id": subscription,
"type": "event",
}
for key in ("hassio", "reauth"):
response = await ws_client.receive_json()
assert response == {
"event": {
"flow": {
"flow_id": form_reauth["flow_id"],
"handler": "test",
"step_id": "account",
"context": {"entry_id": "1234", "source": core_ce.SOURCE_REAUTH},
},
"flow_id": form_reauth["flow_id"],
"type": "init",
},
"id": subscription,
"type": "event",
}
response = await ws_client.receive_json()
assert response == {
"event": {
"flow_id": form_hassio["flow_id"],
"flow_id": forms[key]["flow_id"],
"type": "remove",
},
"id": subscription,
"type": "event",
}
response = await ws_client.receive_json()
assert response == {
"event": {
"flow_id": form_reauth["flow_id"],
"type": "remove",
},
"id": subscription,
"id": 1,
"type": "event",
}
@@ -1078,27 +1052,25 @@ async def test_get_progress_subscribe_in_progress(
assert self._get_reconfigure_entry() is entry
return await self.async_step_account()
flow_context = {
"bluetooth": {"source": core_ce.SOURCE_BLUETOOTH},
"hassio": {"source": core_ce.SOURCE_HASSIO},
"user": {"source": core_ce.SOURCE_USER},
"reauth": {"source": core_ce.SOURCE_REAUTH, "entry_id": "1234"},
"reconfigure": {"source": core_ce.SOURCE_RECONFIGURE, "entry_id": "1234"},
}
forms = {}
with mock_config_flow("test", TestFlow):
form_bluetooth = await hass.config_entries.flow.async_init(
"test", context={"source": core_ce.SOURCE_BLUETOOTH}
)
form_hassio = await hass.config_entries.flow.async_init(
"test", context={"source": core_ce.SOURCE_HASSIO}
)
form_user = await hass.config_entries.flow.async_init(
"test", context={"source": core_ce.SOURCE_USER}
)
form_reauth = await hass.config_entries.flow.async_init(
"test", context={"source": core_ce.SOURCE_REAUTH, "entry_id": "1234"}
)
form_reconfigure = await hass.config_entries.flow.async_init(
"test", context={"source": core_ce.SOURCE_RECONFIGURE, "entry_id": "1234"}
for key, context in flow_context.items():
forms[key] = await hass.config_entries.flow.async_init(
"test", context=context
)
assert form_bluetooth["type"] == data_entry_flow.FlowResultType.ABORT
for form in (form_hassio, form_user, form_reauth, form_reconfigure):
assert form["type"] == data_entry_flow.FlowResultType.FORM
assert form["step_id"] == "account"
assert forms["bluetooth"]["type"] == data_entry_flow.FlowResultType.ABORT
for key in ("hassio", "user", "reauth", "reconfigure"):
assert forms[key]["type"] == data_entry_flow.FlowResultType.FORM
assert forms[key]["step_id"] == "account"
await ws_client.send_json({"id": 1, "type": "config_entries/flow/subscribe"})
@@ -1112,58 +1084,32 @@ async def test_get_progress_subscribe_in_progress(
{
"event": {
"flow": {
"flow_id": form_hassio["flow_id"],
"flow_id": forms[key]["flow_id"],
"handler": "test",
"step_id": "account",
"context": {"source": core_ce.SOURCE_HASSIO},
"context": flow_context[key],
},
"flow_id": form_hassio["flow_id"],
"flow_id": forms[key]["flow_id"],
"type": "init",
},
"id": 1,
"type": "event",
},
{
"event": {
"flow": {
"flow_id": form_reauth["flow_id"],
"handler": "test",
"step_id": "account",
"context": {
"entry_id": "1234",
"source": core_ce.SOURCE_REAUTH,
},
},
"flow_id": form_reauth["flow_id"],
"type": "init",
},
"id": 1,
"type": "event",
},
}
for key in ("hassio", "reauth")
]
)
response = await ws_client.receive_json()
assert response == {"id": ANY, "result": None, "success": True, "type": "result"}
hass.config_entries.flow.async_abort(form_hassio["flow_id"])
hass.config_entries.flow.async_abort(form_user["flow_id"])
hass.config_entries.flow.async_abort(form_reauth["flow_id"])
hass.config_entries.flow.async_abort(form_reconfigure["flow_id"])
for key in ("hassio", "user", "reauth", "reconfigure"):
hass.config_entries.flow.async_abort(forms[key]["flow_id"])
for key in ("hassio", "reauth"):
response = await ws_client.receive_json()
assert response == {
"event": {
"flow_id": form_hassio["flow_id"],
"type": "remove",
},
"id": 1,
"type": "event",
}
response = await ws_client.receive_json()
assert response == {
"event": {
"flow_id": form_reauth["flow_id"],
"flow_id": forms[key]["flow_id"],
"type": "remove",
},
"id": 1,