fix: some minor fixes

Signed-off-by: Rafal Zielinski <sq4ind@gmail.com>

tests/conftest.py

@@ -3,7 +3,7 @@ import pytest
 from unittest.mock import AsyncMock, MagicMock, patch
 
 from homeassistant.core import HomeAssistant
-from homeassistant.helpers.aiohttp_client import async_get_clientsession
+from homeassistant.config_entries import ConfigEntry
 
 from custom_components.adguard_control_hub.const import (
     CONF_HOST,
@@ -50,8 +50,8 @@ def mock_adguard_api():
 
 
 @pytest.fixture
-def mock_config_entry():
-    """Mock config entry."""
+def mock_config_entry_data():
+    """Mock config entry data."""
     return {
         CONF_HOST: "192.168.1.100",
         CONF_PORT: 3000,
@@ -63,7 +63,11 @@ def mock_config_entry():
 
 
 @pytest.fixture
-def mock_setup_entry():
-    """Mock setup entry."""
-    with patch("custom_components.adguard_control_hub.AdGuardHomeAPI") as mock_api_class:
-        yield mock_api_class
+def mock_config_entry(mock_config_entry_data):
+    """Mock config entry."""
+    entry = MagicMock(spec=ConfigEntry)
+    entry.data = mock_config_entry_data
+    entry.entry_id = "test_entry_id"
+    entry.title = "AdGuard Home (192.168.1.100)"
+    entry.domain = DOMAIN
+    return entry