Move Home Connect entry state assertion at tests (#144027)

This commit is contained in:
J. Diego Rodríguez Royo 2025-05-01 23:04:03 +02:00 committed by GitHub
parent abd17d9af9
commit 883ab44437
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
14 changed files with 2 additions and 94 deletions

View File

@ -36,6 +36,7 @@ from homeassistant.components.application_credentials import (
async_import_client_credential, async_import_client_credential,
) )
from homeassistant.components.home_connect.const import DOMAIN from homeassistant.components.home_connect.const import DOMAIN
from homeassistant.config_entries import ConfigEntryState
from homeassistant.const import Platform from homeassistant.const import Platform
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.setup import async_setup_component from homeassistant.setup import async_setup_component
@ -147,6 +148,7 @@ async def mock_integration_setup(
config_entry.add_to_hass(hass) config_entry.add_to_hass(hass)
async def run(client: MagicMock) -> bool: async def run(client: MagicMock) -> bool:
assert config_entry.state is ConfigEntryState.NOT_LOADED
with ( with (
patch("homeassistant.components.home_connect.PLATFORMS", platforms), patch("homeassistant.components.home_connect.PLATFORMS", platforms),
patch( patch(

View File

@ -51,7 +51,6 @@ async def test_paired_depaired_devices_flow(
appliance: HomeAppliance, appliance: HomeAppliance,
) -> None: ) -> None:
"""Test that removed devices are correctly removed from and added to hass on API events.""" """Test that removed devices are correctly removed from and added to hass on API events."""
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
@ -128,7 +127,6 @@ async def test_connected_devices(
return get_status_original_mock.return_value return get_status_original_mock.return_value
client.get_status = AsyncMock(side_effect=get_status_side_effect) client.get_status = AsyncMock(side_effect=get_status_side_effect)
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
client.get_status = get_status_original_mock client.get_status = get_status_original_mock
@ -179,7 +177,6 @@ async def test_binary_sensors_entity_availability(
entity_ids = [ entity_ids = [
"binary_sensor.washer_remote_control", "binary_sensor.washer_remote_control",
] ]
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
@ -279,7 +276,6 @@ async def test_binary_sensors_functionality(
expected: str, expected: str,
) -> None: ) -> None:
"""Tests for Home Connect Fridge appliance door states.""" """Tests for Home Connect Fridge appliance door states."""
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
await client.add_events( await client.add_events(
@ -316,7 +312,6 @@ async def test_connected_sensor_functionality(
) -> None: ) -> None:
"""Test if the connected binary sensor reports the right values.""" """Test if the connected binary sensor reports the right values."""
entity_id = "binary_sensor.washer_connectivity" entity_id = "binary_sensor.washer_connectivity"
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED

View File

@ -43,7 +43,6 @@ async def test_paired_depaired_devices_flow(
appliance: HomeAppliance, appliance: HomeAppliance,
) -> None: ) -> None:
"""Test that removed devices are correctly removed from and added to hass on API events.""" """Test that removed devices are correctly removed from and added to hass on API events."""
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
@ -131,7 +130,6 @@ async def test_connected_devices(
side_effect=get_available_commands_side_effect side_effect=get_available_commands_side_effect
) )
client.get_all_programs = AsyncMock(side_effect=get_all_programs_side_effect) client.get_all_programs = AsyncMock(side_effect=get_all_programs_side_effect)
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
client.get_available_commands = get_available_commands_original_mock client.get_available_commands = get_available_commands_original_mock
@ -183,7 +181,6 @@ async def test_button_entity_availability(
"button.washer_pause_program", "button.washer_pause_program",
"button.washer_stop_program", "button.washer_stop_program",
] ]
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
@ -246,7 +243,6 @@ async def test_button_functionality(
appliance: HomeAppliance, appliance: HomeAppliance,
) -> None: ) -> None:
"""Test if button entities availability are based on the appliance connection state.""" """Test if button entities availability are based on the appliance connection state."""
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
@ -282,7 +278,6 @@ async def test_command_button_exception(
] ]
) )
) )
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client_with_exception) assert await integration_setup(client_with_exception)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
@ -308,7 +303,6 @@ async def test_stop_program_button_exception(
"""Test if button entities availability are based on the appliance connection state.""" """Test if button entities availability are based on the appliance connection state."""
entity_id = "button.washer_stop_program" entity_id = "button.washer_stop_program"
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client_with_exception) assert await integration_setup(client_with_exception)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED

View File

@ -218,7 +218,6 @@ async def test_coordinator_not_fetching_on_disconnected_appliance(
"""Test that the coordinator does not fetch anything on disconnected appliance.""" """Test that the coordinator does not fetch anything on disconnected appliance."""
appliance.connected = False appliance.connected = False
assert config_entry.state == ConfigEntryState.NOT_LOADED
await integration_setup(client) await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
@ -242,7 +241,6 @@ async def test_coordinator_update_failing(
""" """
setattr(client, mock_method, AsyncMock(side_effect=HomeConnectError())) setattr(client, mock_method, AsyncMock(side_effect=HomeConnectError()))
assert config_entry.state == ConfigEntryState.NOT_LOADED
await integration_setup(client) await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
@ -286,7 +284,6 @@ async def test_event_listener(
entity_id: str, entity_id: str,
) -> None: ) -> None:
"""Test that the event listener works.""" """Test that the event listener works."""
assert config_entry.state == ConfigEntryState.NOT_LOADED
await integration_setup(client) await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
@ -353,7 +350,6 @@ async def tests_receive_setting_and_status_for_first_time_at_events(
client.get_setting = AsyncMock(return_value=ArrayOfSettings([])) client.get_setting = AsyncMock(return_value=ArrayOfSettings([]))
client.get_status = AsyncMock(return_value=ArrayOfStatus([])) client.get_status = AsyncMock(return_value=ArrayOfStatus([]))
assert config_entry.state == ConfigEntryState.NOT_LOADED
await integration_setup(client) await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
@ -468,7 +464,6 @@ async def test_event_listener_resilience(
side_effect=[stream_exception(), client.stream_all_events()] side_effect=[stream_exception(), client.stream_all_events()]
) )
assert config_entry.state == ConfigEntryState.NOT_LOADED
await integration_setup(client) await integration_setup(client)
await hass.async_block_till_done() await hass.async_block_till_done()
@ -531,7 +526,6 @@ async def test_devices_updated_on_refresh(
) )
await async_setup_component(hass, HA_DOMAIN, {}) await async_setup_component(hass, HA_DOMAIN, {})
assert config_entry.state == ConfigEntryState.NOT_LOADED
await integration_setup(client) await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
@ -564,7 +558,6 @@ async def test_paired_disconnected_devices_not_fetching(
) -> None: ) -> None:
"""Test that Home Connect API is not fetched after pairing a disconnected device.""" """Test that Home Connect API is not fetched after pairing a disconnected device."""
client.get_home_appliances = AsyncMock(return_value=ArrayOfHomeAppliances([])) client.get_home_appliances = AsyncMock(return_value=ArrayOfHomeAppliances([]))
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
@ -601,7 +594,6 @@ async def test_coordinator_disabling_updates_for_appliance(
appliance_ha_id = "SIEMENS-HCS02DWH1-6BE58C26DCC1" appliance_ha_id = "SIEMENS-HCS02DWH1-6BE58C26DCC1"
issue_id = f"home_connect_too_many_connected_paired_events_{appliance_ha_id}" issue_id = f"home_connect_too_many_connected_paired_events_{appliance_ha_id}"
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
@ -692,7 +684,6 @@ async def test_coordinator_disabling_updates_for_appliance_is_gone_after_entry_r
appliance_ha_id = "SIEMENS-HCS02DWH1-6BE58C26DCC1" appliance_ha_id = "SIEMENS-HCS02DWH1-6BE58C26DCC1"
issue_id = f"home_connect_too_many_connected_paired_events_{appliance_ha_id}" issue_id = f"home_connect_too_many_connected_paired_events_{appliance_ha_id}"
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
@ -718,7 +709,6 @@ async def test_coordinator_disabling_updates_for_appliance_is_gone_after_entry_r
assert not issue_registry.async_get_issue(DOMAIN, issue_id) assert not issue_registry.async_get_issue(DOMAIN, issue_id)
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED

View File

@ -25,7 +25,6 @@ async def test_async_get_config_entry_diagnostics(
snapshot: SnapshotAssertion, snapshot: SnapshotAssertion,
) -> None: ) -> None:
"""Test config entry diagnostics.""" """Test config entry diagnostics."""
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
@ -41,7 +40,6 @@ async def test_async_get_device_diagnostics(
snapshot: SnapshotAssertion, snapshot: SnapshotAssertion,
) -> None: ) -> None:
"""Test device config entry diagnostics.""" """Test device config entry diagnostics."""
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED

View File

@ -157,7 +157,6 @@ async def test_program_options_retrieval(
) )
) )
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
@ -276,7 +275,6 @@ async def test_no_options_retrieval_on_unknown_program(
client.get_all_programs = AsyncMock(side_effect=get_all_programs_with_options_mock) client.get_all_programs = AsyncMock(side_effect=get_all_programs_with_options_mock)
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
@ -357,7 +355,6 @@ async def test_program_options_retrieval_after_appliance_connection(
) )
) )
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
@ -469,7 +466,6 @@ async def test_option_entity_functionality_exception(
) )
) )
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED

View File

@ -46,7 +46,6 @@ async def test_entry_setup(
integration_setup: Callable[[MagicMock], Awaitable[bool]], integration_setup: Callable[[MagicMock], Awaitable[bool]],
) -> None: ) -> None:
"""Test setup and unload.""" """Test setup and unload."""
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
@ -182,7 +181,6 @@ async def test_client_error(
"""Test client errors during setup integration.""" """Test client errors during setup integration."""
client_with_exception.get_home_appliances.return_value = None client_with_exception.get_home_appliances.return_value = None
client_with_exception.get_home_appliances.side_effect = exception client_with_exception.get_home_appliances.side_effect = exception
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert not await integration_setup(client_with_exception) assert not await integration_setup(client_with_exception)
assert config_entry.state == expected_state assert config_entry.state == expected_state
assert client_with_exception.get_home_appliances.call_count == 1 assert client_with_exception.get_home_appliances.call_count == 1
@ -239,7 +237,6 @@ async def test_required_program_or_at_least_an_option(
) -> None: ) -> None:
"Test that the set_program_and_options does raise an exception if no program nor options are set." "Test that the set_program_and_options does raise an exception if no program nor options are set."
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED

View File

@ -64,7 +64,6 @@ async def test_paired_depaired_devices_flow(
appliance: HomeAppliance, appliance: HomeAppliance,
) -> None: ) -> None:
"""Test that removed devices are correctly removed from and added to hass on API events.""" """Test that removed devices are correctly removed from and added to hass on API events."""
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
@ -141,7 +140,6 @@ async def test_connected_devices(
return await get_settings_original_mock.side_effect(ha_id) return await get_settings_original_mock.side_effect(ha_id)
client.get_settings = AsyncMock(side_effect=get_settings_side_effect) client.get_settings = AsyncMock(side_effect=get_settings_side_effect)
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
client.get_settings = get_settings_original_mock client.get_settings = get_settings_original_mock
@ -186,7 +184,6 @@ async def test_light_availability(
entity_ids = [ entity_ids = [
"light.hood_functional_light", "light.hood_functional_light",
] ]
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
@ -353,7 +350,6 @@ async def test_light_functionality(
appliance: HomeAppliance, appliance: HomeAppliance,
) -> None: ) -> None:
"""Test light functionality.""" """Test light functionality."""
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
@ -405,7 +401,6 @@ async def test_light_color_different_than_custom(
appliance: HomeAppliance, appliance: HomeAppliance,
) -> None: ) -> None:
"""Test that light color attributes are not set if color is different than custom.""" """Test that light color attributes are not set if color is different than custom."""
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
await hass.services.async_call( await hass.services.async_call(
@ -586,7 +581,6 @@ async def test_light_exception_handling(
client_with_exception.set_setting.side_effect = [ client_with_exception.set_setting.side_effect = [
exception() if exception else None for exception in attr_side_effect exception() if exception else None for exception in attr_side_effect
] ]
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client_with_exception) assert await integration_setup(client_with_exception)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED

View File

@ -80,7 +80,6 @@ async def test_paired_depaired_devices_flow(
], ],
) )
) )
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
@ -159,7 +158,6 @@ async def test_connected_devices(
return get_settings_original_mock.return_value return get_settings_original_mock.return_value
client.get_settings = AsyncMock(side_effect=get_settings_side_effect) client.get_settings = AsyncMock(side_effect=get_settings_side_effect)
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
client.get_settings = get_settings_original_mock client.get_settings = get_settings_original_mock
@ -209,7 +207,6 @@ async def test_number_entity_availability(
# Setting constrains are not needed for this test # Setting constrains are not needed for this test
# so we rise an error to easily test the availability # so we rise an error to easily test the availability
client.get_setting = AsyncMock(side_effect=HomeConnectError()) client.get_setting = AsyncMock(side_effect=HomeConnectError())
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
@ -316,7 +313,6 @@ async def test_number_entity_functionality(
) )
) )
assert config_entry.state is ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state is ConfigEntryState.LOADED assert config_entry.state is ConfigEntryState.LOADED
entity_state = hass.states.get(entity_id) entity_state = hass.states.get(entity_id)
@ -420,7 +416,6 @@ async def test_fetch_constraints_after_rate_limit_error(
] ]
) )
assert config_entry.state is ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
async_fire_time_changed(hass) async_fire_time_changed(hass)
await hass.async_block_till_done() await hass.async_block_till_done()
@ -472,7 +467,6 @@ async def test_number_entity_error(
) )
] ]
) )
assert config_entry.state is ConfigEntryState.NOT_LOADED
assert await integration_setup(client_with_exception) assert await integration_setup(client_with_exception)
assert config_entry.state is ConfigEntryState.LOADED assert config_entry.state is ConfigEntryState.LOADED
@ -599,7 +593,6 @@ async def test_options_functionality(
) )
) )
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
entity_state = hass.states.get(entity_id) entity_state = hass.states.get(entity_id)

View File

@ -84,7 +84,6 @@ async def test_paired_depaired_devices_flow(
], ],
) )
) )
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
@ -174,7 +173,6 @@ async def test_connected_devices(
client.get_settings = AsyncMock(side_effect=get_settings_side_effect) client.get_settings = AsyncMock(side_effect=get_settings_side_effect)
client.get_all_programs = AsyncMock(side_effect=get_all_programs_side_effect) client.get_all_programs = AsyncMock(side_effect=get_all_programs_side_effect)
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
client.get_settings = get_settings_original_mock client.get_settings = get_settings_original_mock
@ -220,7 +218,6 @@ async def test_select_entity_availability(
entity_ids = [ entity_ids = [
"select.washer_active_program", "select.washer_active_program",
] ]
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
@ -298,7 +295,6 @@ async def test_filter_programs(
] ]
) )
assert config_entry.state is ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state is ConfigEntryState.LOADED assert config_entry.state is ConfigEntryState.LOADED
@ -365,7 +361,6 @@ async def test_select_program_functionality(
event_key: EventKey, event_key: EventKey,
) -> None: ) -> None:
"""Test select functionality.""" """Test select functionality."""
assert config_entry.state is ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state is ConfigEntryState.LOADED assert config_entry.state is ConfigEntryState.LOADED
@ -448,7 +443,6 @@ async def test_select_exception_handling(
] ]
) )
assert config_entry.state is ConfigEntryState.NOT_LOADED
assert await integration_setup(client_with_exception) assert await integration_setup(client_with_exception)
assert config_entry.state is ConfigEntryState.LOADED assert config_entry.state is ConfigEntryState.LOADED
@ -494,7 +488,6 @@ async def test_programs_updated_on_connect(
return await get_all_programs_mock.side_effect(ha_id) return await get_all_programs_mock.side_effect(ha_id)
client.get_all_programs = AsyncMock(side_effect=get_all_programs_side_effect) client.get_all_programs = AsyncMock(side_effect=get_all_programs_side_effect)
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
client.get_all_programs = get_all_programs_mock client.get_all_programs = get_all_programs_mock
@ -566,7 +559,6 @@ async def test_select_functionality(
expected_value_call_arg: str, expected_value_call_arg: str,
) -> None: ) -> None:
"""Test select functionality.""" """Test select functionality."""
assert config_entry.state is ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state is ConfigEntryState.LOADED assert config_entry.state is ConfigEntryState.LOADED
@ -646,7 +638,6 @@ async def test_fetch_allowed_values(
client.get_setting = AsyncMock(side_effect=get_setting_side_effect) client.get_setting = AsyncMock(side_effect=get_setting_side_effect)
assert config_entry.state is ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state is ConfigEntryState.LOADED assert config_entry.state is ConfigEntryState.LOADED
@ -713,7 +704,6 @@ async def test_fetch_allowed_values_after_rate_limit_error(
] ]
) )
assert config_entry.state is ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
async_fire_time_changed(hass) async_fire_time_changed(hass)
await hass.async_block_till_done() await hass.async_block_till_done()
@ -775,7 +765,6 @@ async def test_default_values_after_fetch_allowed_values_error(
client.get_settings = AsyncMock(side_effect=get_settings_side_effect) client.get_settings = AsyncMock(side_effect=get_settings_side_effect)
client.get_setting = AsyncMock(side_effect=exception) client.get_setting = AsyncMock(side_effect=exception)
assert config_entry.state is ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state is ConfigEntryState.LOADED assert config_entry.state is ConfigEntryState.LOADED
@ -821,7 +810,6 @@ async def test_select_entity_error(
) )
] ]
) )
assert config_entry.state is ConfigEntryState.NOT_LOADED
assert await integration_setup(client_with_exception) assert await integration_setup(client_with_exception)
assert config_entry.state is ConfigEntryState.LOADED assert config_entry.state is ConfigEntryState.LOADED
@ -952,7 +940,6 @@ async def test_options_functionality(
) )
) )
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
entity_state = hass.states.get(entity_id) entity_state = hass.states.get(entity_id)

View File

@ -100,7 +100,6 @@ async def test_paired_depaired_devices_flow(
appliance: HomeAppliance, appliance: HomeAppliance,
) -> None: ) -> None:
"""Test that removed devices are correctly removed from and added to hass on API events.""" """Test that removed devices are correctly removed from and added to hass on API events."""
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
@ -200,7 +199,6 @@ async def test_connected_devices(
return get_status_original_mock.return_value return get_status_original_mock.return_value
client.get_status = AsyncMock(side_effect=get_status_side_effect) client.get_status = AsyncMock(side_effect=get_status_side_effect)
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
client.get_status = get_status_original_mock client.get_status = get_status_original_mock
@ -246,7 +244,6 @@ async def test_sensor_entity_availability(
"sensor.dishwasher_operation_state", "sensor.dishwasher_operation_state",
"sensor.dishwasher_salt_nearly_empty", "sensor.dishwasher_salt_nearly_empty",
] ]
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
@ -452,7 +449,6 @@ async def test_program_sensor_edge_case(
) )
) )
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
@ -515,7 +511,6 @@ async def test_remaining_prog_time_edge_cases(
time_to_freeze = "2021-01-09 12:00:00+00:00" time_to_freeze = "2021-01-09 12:00:00+00:00"
freezer.move_to(time_to_freeze) freezer.move_to(time_to_freeze)
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
@ -591,7 +586,6 @@ async def test_sensors_states(
appliance: HomeAppliance, appliance: HomeAppliance,
) -> None: ) -> None:
"""Tests for appliance sensors.""" """Tests for appliance sensors."""
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
@ -653,7 +647,6 @@ async def test_event_sensors_states(
) -> None: ) -> None:
"""Tests for appliance event sensors.""" """Tests for appliance event sensors."""
caplog.set_level(logging.ERROR) caplog.set_level(logging.ERROR)
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
@ -763,7 +756,6 @@ async def test_sensor_unit_fetching(
) )
) )
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
@ -819,7 +811,6 @@ async def test_sensor_unit_fetching_error(
client.get_status = AsyncMock(side_effect=get_status_mock) client.get_status = AsyncMock(side_effect=get_status_mock)
client.get_status_value = AsyncMock(side_effect=HomeConnectError()) client.get_status_value = AsyncMock(side_effect=HomeConnectError())
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
@ -881,7 +872,6 @@ async def test_sensor_unit_fetching_after_rate_limit_error(
] ]
) )
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
async_fire_time_changed(hass) async_fire_time_changed(hass)
await hass.async_block_till_done() await hass.async_block_till_done()

View File

@ -185,7 +185,6 @@ async def test_key_value_services(
service_call: dict[str, Any], service_call: dict[str, Any],
) -> None: ) -> None:
"""Create and test services.""" """Create and test services."""
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
@ -236,7 +235,6 @@ async def test_programs_and_options_actions_deprecation(
issue_id: str, issue_id: str,
) -> None: ) -> None:
"""Test deprecated service keys.""" """Test deprecated service keys."""
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
@ -305,7 +303,6 @@ async def test_set_program_and_options(
snapshot: SnapshotAssertion, snapshot: SnapshotAssertion,
) -> None: ) -> None:
"""Test recognized options.""" """Test recognized options."""
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
@ -347,7 +344,6 @@ async def test_set_program_and_options_exceptions(
error_regex: str, error_regex: str,
) -> None: ) -> None:
"""Test recognized options.""" """Test recognized options."""
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client_with_exception) assert await integration_setup(client_with_exception)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
@ -376,7 +372,6 @@ async def test_services_exception_device_id(
service_call: dict[str, Any], service_call: dict[str, Any],
) -> None: ) -> None:
"""Raise a HomeAssistantError when there is an API error.""" """Raise a HomeAssistantError when there is an API error."""
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client_with_exception) assert await integration_setup(client_with_exception)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
@ -399,7 +394,6 @@ async def test_services_appliance_not_found(
integration_setup: Callable[[MagicMock], Awaitable[bool]], integration_setup: Callable[[MagicMock], Awaitable[bool]],
) -> None: ) -> None:
"""Raise a ServiceValidationError when device id does not match.""" """Raise a ServiceValidationError when device id does not match."""
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
@ -448,7 +442,6 @@ async def test_services_exception(
service_call: dict[str, Any], service_call: dict[str, Any],
) -> None: ) -> None:
"""Raise a ValueError when device id does not match.""" """Raise a ValueError when device id does not match."""
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client_with_exception) assert await integration_setup(client_with_exception)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED

View File

@ -91,7 +91,6 @@ async def test_paired_depaired_devices_flow(
], ],
) )
) )
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
@ -181,7 +180,6 @@ async def test_connected_devices(
client.get_settings = AsyncMock(side_effect=get_settings_side_effect) client.get_settings = AsyncMock(side_effect=get_settings_side_effect)
client.get_all_programs = AsyncMock(side_effect=get_all_programs_side_effect) client.get_all_programs = AsyncMock(side_effect=get_all_programs_side_effect)
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
client.get_settings = get_settings_original_mock client.get_settings = get_settings_original_mock
@ -230,7 +228,6 @@ async def test_switch_entity_availability(
"switch.dishwasher_child_lock", "switch.dishwasher_child_lock",
"switch.dishwasher_program_eco50", "switch.dishwasher_program_eco50",
] ]
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
@ -313,7 +310,6 @@ async def test_switch_functionality(
) -> None: ) -> None:
"""Test switch functionality.""" """Test switch functionality."""
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
@ -380,7 +376,6 @@ async def test_program_switch_functionality(
) )
client.stop_program = AsyncMock(side_effect=mock_stop_program) client.stop_program = AsyncMock(side_effect=mock_stop_program)
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
assert hass.states.is_state(entity_id, initial_state) assert hass.states.is_state(entity_id, initial_state)
@ -488,7 +483,6 @@ async def test_switch_exception_handling(
] ]
) )
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client_with_exception) assert await integration_setup(client_with_exception)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
@ -532,7 +526,6 @@ async def test_ent_desc_switch_functionality(
) -> None: ) -> None:
"""Test switch functionality - entity description setup.""" """Test switch functionality - entity description setup."""
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
@ -589,7 +582,6 @@ async def test_ent_desc_switch_exception_handling(
for key, value in status.items() for key, value in status.items()
] ]
) )
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client_with_exception) assert await integration_setup(client_with_exception)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
@ -675,7 +667,6 @@ async def test_power_switch(
] ]
) )
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
@ -715,7 +706,6 @@ async def test_power_switch_fetch_off_state_from_current_value(
] ]
) )
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
@ -781,7 +771,6 @@ async def test_power_switch_service_validation_errors(
client.get_settings.return_value = ArrayOfSettings([setting]) client.get_settings.return_value = ArrayOfSettings([setting])
client.get_setting = AsyncMock(return_value=setting) client.get_setting = AsyncMock(return_value=setting)
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
@ -842,7 +831,6 @@ async def test_create_program_switch_deprecation_issue(
}, },
) )
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
@ -923,7 +911,6 @@ async def test_program_switch_deprecation_issue_fix(
}, },
) )
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
@ -1018,7 +1005,6 @@ async def test_options_functionality(
) )
) )
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
assert hass.states.get(entity_id) assert hass.states.get(entity_id)

View File

@ -57,7 +57,6 @@ async def test_paired_depaired_devices_flow(
appliance: HomeAppliance, appliance: HomeAppliance,
) -> None: ) -> None:
"""Test that removed devices are correctly removed from and added to hass on API events.""" """Test that removed devices are correctly removed from and added to hass on API events."""
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
@ -135,7 +134,6 @@ async def test_connected_devices(
return await get_settings_original_mock.side_effect(ha_id) return await get_settings_original_mock.side_effect(ha_id)
client.get_settings = AsyncMock(side_effect=get_settings_side_effect) client.get_settings = AsyncMock(side_effect=get_settings_side_effect)
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
client.get_settings = get_settings_original_mock client.get_settings = get_settings_original_mock
@ -181,7 +179,6 @@ async def test_time_entity_availability(
entity_ids = [ entity_ids = [
"time.oven_alarm_clock", "time.oven_alarm_clock",
] ]
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
@ -242,7 +239,6 @@ async def test_time_entity_functionality(
setting_key: SettingKey, setting_key: SettingKey,
) -> None: ) -> None:
"""Test time entity functionality.""" """Test time entity functionality."""
assert config_entry.state is ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state is ConfigEntryState.LOADED assert config_entry.state is ConfigEntryState.LOADED
@ -296,7 +292,6 @@ async def test_time_entity_error(
) )
] ]
) )
assert config_entry.state is ConfigEntryState.NOT_LOADED
assert await integration_setup(client_with_exception) assert await integration_setup(client_with_exception)
assert config_entry.state is ConfigEntryState.LOADED assert config_entry.state is ConfigEntryState.LOADED
@ -367,7 +362,6 @@ async def test_create_alarm_clock_deprecation_issue(
}, },
) )
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED
@ -447,7 +441,6 @@ async def test_alarm_clock_deprecation_issue_fix(
}, },
) )
assert config_entry.state == ConfigEntryState.NOT_LOADED
assert await integration_setup(client) assert await integration_setup(client)
assert config_entry.state == ConfigEntryState.LOADED assert config_entry.state == ConfigEntryState.LOADED