@@ -90,10 +90,10 @@ async def test_setup_entry_connection_failure(hass: HomeAssistant, mock_config_e
|
||||
mock_api.test_connection = AsyncMock(return_value=False)
|
||||
|
||||
with patch("custom_components.adguard_hub.AdGuardHomeAPI", return_value=mock_api), \
|
||||
patch("custom_components.adguard_hub.async_get_clientsession"), \
|
||||
pytest.raises(Exception): # Should raise ConfigEntryNotReady
|
||||
patch("custom_components.adguard_hub.async_get_clientsession"):
|
||||
|
||||
await async_setup_entry(hass, mock_config_entry)
|
||||
with pytest.raises(Exception): # Should raise ConfigEntryNotReady
|
||||
await async_setup_entry(hass, mock_config_entry)
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
@@ -154,7 +154,7 @@ def test_services_registration(hass: HomeAssistant):
|
||||
"""Test that services are properly registered."""
|
||||
from custom_components.adguard_hub.services import AdGuardControlHubServices
|
||||
|
||||
# Create services without running inside an existing event loop
|
||||
# Create services without async context
|
||||
services = AdGuardControlHubServices(hass)
|
||||
services.register_services()
|
||||
|
||||
@@ -162,9 +162,9 @@ def test_services_registration(hass: HomeAssistant):
|
||||
assert hass.services.has_service(DOMAIN, "block_services")
|
||||
assert hass.services.has_service(DOMAIN, "unblock_services")
|
||||
assert hass.services.has_service(DOMAIN, "emergency_unblock")
|
||||
assert hass.services.has_service(DOMAIN, "bulk_update_clients")
|
||||
assert hass.services.has_service(DOMAIN, "add_client")
|
||||
assert hass.services.has_service(DOMAIN, "remove_client")
|
||||
assert hass.services.has_service(DOMAIN, "bulk_update_clients")
|
||||
|
||||
# Clean up
|
||||
services.unregister_services()
|
||||
|
Reference in New Issue
Block a user