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

Use mock_config_flow helper in config_entries tests (#117241)

Browse Source
This commit is contained in:
Erik Montnemery 2024-05-11 20:11:18 +02:00 committed by GitHub
parent 5c1f6aeb60
commit 021b057a87
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 25 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

51
tests/test_config_entries.py
View File

@ -1407,7 +1407,7 @@ async def test_entry_options(
entry = MockConfigEntry(domain="test", data={"first": True}, options=None) entry = MockConfigEntry(domain="test", data={"first": True}, options=None)
entry.add_to_manager(manager) entry.add_to_manager(manager)
class TestFlow: class TestFlow(config_entries.ConfigFlow):
"""Test flow.""" """Test flow."""
@staticmethod @staticmethod
@ -1420,25 +1420,24 @@ async def test_entry_options(
return OptionsFlowHandler() return OptionsFlowHandler()
def async_supports_options_flow(self, entry: MockConfigEntry) -> bool: with mock_config_flow("test", TestFlow):
"""Test options flow.""" flow = await manager.options.async_create_flow(
return True entry.entry_id, context={"source": "test"}, data=None
)
config_entries.HANDLERS["test"] = TestFlow() flow.handler = entry.entry_id # Used to keep reference to config entry
flow = await manager.options.async_create_flow(
entry.entry_id, context={"source": "test"}, data=None
)
flow.handler = entry.entry_id # Used to keep reference to config entry await manager.options.async_finish_flow(
flow,
{
"data": {"second": True},
"type": data_entry_flow.FlowResultType.CREATE_ENTRY,
},
)
await manager.options.async_finish_flow( assert entry.data == {"first": True}
flow, assert entry.options == {"second": True}
{"data": {"second": True}, "type": data_entry_flow.FlowResultType.CREATE_ENTRY}, assert entry.supports_options is True
)
assert entry.data == {"first": True}
assert entry.options == {"second": True}
assert entry.supports_options is True
async def test_entry_options_abort( async def test_entry_options_abort(
@ -1450,7 +1449,7 @@ async def test_entry_options_abort(
entry = MockConfigEntry(domain="test", data={"first": True}, options=None) entry = MockConfigEntry(domain="test", data={"first": True}, options=None)
entry.add_to_manager(manager) entry.add_to_manager(manager)
class TestFlow: class TestFlow(config_entries.ConfigFlow):
"""Test flow.""" """Test flow."""
@staticmethod @staticmethod
@ -1463,16 +1462,16 @@ async def test_entry_options_abort(
return OptionsFlowHandler() return OptionsFlowHandler()
config_entries.HANDLERS["test"] = TestFlow() with mock_config_flow("test", TestFlow):
flow = await manager.options.async_create_flow( flow = await manager.options.async_create_flow(
entry.entry_id, context={"source": "test"}, data=None entry.entry_id, context={"source": "test"}, data=None
) )
flow.handler = entry.entry_id # Used to keep reference to config entry flow.handler = entry.entry_id # Used to keep reference to config entry
assert await manager.options.async_finish_flow( assert await manager.options.async_finish_flow(
flow, {"type": data_entry_flow.FlowResultType.ABORT, "reason": "test"} flow, {"type": data_entry_flow.FlowResultType.ABORT, "reason": "test"}
) )
async def test_entry_options_unknown_config_entry( async def test_entry_options_unknown_config_entry(