diff --git a/tests/components/rituals_perfume_genie/common.py b/tests/components/rituals_perfume_genie/common.py index f2a54ca5def..044582c5735 100644 --- a/tests/components/rituals_perfume_genie/common.py +++ b/tests/components/rituals_perfume_genie/common.py @@ -85,7 +85,7 @@ def mock_diffuser_v2_no_battery_no_cartridge() -> MagicMock: async def init_integration( hass: HomeAssistant, mock_config_entry: MockConfigEntry, - mock_diffusers: list[MagicMock] = [mock_diffuser(hublot="lot123")], + mock_diffusers: list[MagicMock], ) -> None: """Initialize the Rituals Perfume Genie integration with the given Config Entry and Diffuser list.""" mock_config_entry.add_to_hass(hass) diff --git a/tests/components/rituals_perfume_genie/test_init.py b/tests/components/rituals_perfume_genie/test_init.py index d1001d1ad93..435e762a646 100644 --- a/tests/components/rituals_perfume_genie/test_init.py +++ b/tests/components/rituals_perfume_genie/test_init.py @@ -12,6 +12,7 @@ from homeassistant.helpers import entity_registry as er from .common import ( init_integration, mock_config_entry, + mock_diffuser, mock_diffuser_v1_battery_cartridge, ) @@ -31,7 +32,7 @@ async def test_config_entry_not_ready(hass: HomeAssistant) -> None: async def test_config_entry_unload(hass: HomeAssistant) -> None: """Test the Rituals Perfume Genie configuration entry setup and unloading.""" config_entry = mock_config_entry(unique_id="id_123_unload") - await init_integration(hass, config_entry) + await init_integration(hass, config_entry, [mock_diffuser(hublot="lot123")]) await hass.config_entries.async_unload(config_entry.entry_id) await hass.async_block_till_done()