github/ha-core
1
Fork 0
mirror of https://github.com/home-assistant/core synced 2024-07-09 04:58:30 +02:00
Code Releases Activity

Address late review of config entry wait for states tests (#81801)

Browse Source 
* Late review on tests #81771

* Clean-up duplicate assignment
This commit is contained in:
Jan Bouwhuis 2022-11-08 22:14:17 +01:00 committed by GitHub
parent 6021cedb09
commit 12d76a8a4f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
tests/test_config_entries.py
View File

@ -3332,7 +3332,7 @@ async def test_reauth(hass):
assert len(hass.config_entries.flow.async_progress()) == 2
async def test_wait_for_loading_entry(hass):
async def test_wait_for_loading_entry(hass: HomeAssistant) -> None:
"""Test waiting for entry to be set up."""
entry = MockConfigEntry(title="test_title", domain="test")
@ -3346,15 +3346,13 @@ async def test_wait_for_loading_entry(hass):
flow = hass.config_entries.flow
async def _load_entry():
# Mock config entry
async def _load_entry() -> None:
assert await async_setup_component(hass, "test", {})
entry = MockConfigEntry(title="test_title", domain="test")
entry.add_to_hass(hass)
flow = hass.config_entries.flow
with patch.object(flow, "async_init", wraps=flow.async_init):
hass.async_add_job(_load_entry)
hass.async_create_task(_load_entry())
new_state = await hass.config_entries.async_wait_for_states(
entry,
{
@ -3367,7 +3365,7 @@ async def test_wait_for_loading_entry(hass):
assert entry.state is config_entries.ConfigEntryState.LOADED
async def test_wait_for_loading_failed_entry(hass):
async def test_wait_for_loading_failed_entry(hass: HomeAssistant) -> None:
"""Test waiting for entry to be set up that fails loading."""
entry = MockConfigEntry(title="test_title", domain="test")
@ -3376,20 +3374,17 @@ async def test_wait_for_loading_failed_entry(hass):
mock_integration(hass, MockModule("test", async_setup_entry=mock_setup_entry))
mock_entity_platform(hass, "config_flow.test", None)
await entry.async_setup(hass)
await hass.async_block_till_done()
flow = hass.config_entries.flow
async def _load_entry():
# Mock config entry
async def _load_entry() -> None:
assert await async_setup_component(hass, "test", {})
entry = MockConfigEntry(title="test_title", domain="test")
entry.add_to_hass(hass)
flow = hass.config_entries.flow
with patch.object(flow, "async_init", wraps=flow.async_init):
hass.async_add_job(_load_entry)
hass.async_create_task(_load_entry())
new_state = await hass.config_entries.async_wait_for_states(
entry,
{
@ -3402,7 +3397,7 @@ async def test_wait_for_loading_failed_entry(hass):
assert entry.state is config_entries.ConfigEntryState.SETUP_ERROR
async def test_wait_for_loading_timeout(hass):
async def test_wait_for_loading_timeout(hass: HomeAssistant) -> None:
"""Test waiting for entry to be set up that fails with a timeout."""
async def _async_setup_entry(hass, entry):
@ -3414,20 +3409,17 @@ async def test_wait_for_loading_timeout(hass):
mock_integration(hass, MockModule("test", async_setup_entry=_async_setup_entry))
mock_entity_platform(hass, "config_flow.test", None)
await entry.async_setup(hass)
await hass.async_block_till_done()
flow = hass.config_entries.flow
async def _load_entry():
# Mock config entry
async def _load_entry() -> None:
assert await async_setup_component(hass, "test", {})
entry = MockConfigEntry(title="test_title", domain="test")
entry.add_to_hass(hass)
flow = hass.config_entries.flow
with patch.object(flow, "async_init", wraps=flow.async_init):
hass.async_add_job(_load_entry)
hass.async_create_task(_load_entry())
with pytest.raises(asyncio.exceptions.TimeoutError):
await hass.config_entries.async_wait_for_states(
entry,