diff --git a/tests/components/wilight/test_init.py b/tests/components/wilight/test_init.py index 24efdaaa8e1..5aadce3caea 100644 --- a/tests/components/wilight/test_init.py +++ b/tests/components/wilight/test_init.py @@ -4,6 +4,7 @@ from unittest.mock import patch import pytest import pywilight from pywilight.const import DOMAIN +import requests from homeassistant.config_entries import ConfigEntryState from homeassistant.core import HomeAssistant @@ -41,7 +42,11 @@ def mock_dummy_device_from_host(): async def test_config_entry_not_ready(hass: HomeAssistant) -> None: """Test the WiLight configuration entry not ready.""" - entry = await setup_integration(hass) + with patch( + "pywilight.device_from_host", + side_effect=requests.exceptions.Timeout, + ): + entry = await setup_integration(hass) assert entry.state is ConfigEntryState.SETUP_RETRY