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

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

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