fix: some minor fixes

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

tests/test_init.py

@@ -1,5 +1,5 @@
 """Test AdGuard Control Hub initialization."""
-from unittest.mock import patch
+from unittest.mock import patch, AsyncMock
 
 import pytest
 from homeassistant.config_entries import ConfigEntryState
@@ -9,39 +9,71 @@ from homeassistant.core import HomeAssistant
 from custom_components.adguard_control_hub.const import DOMAIN
 
 
-async def test_setup_entry(hass: HomeAssistant, mock_config_entry, mock_adguard_api) -> None:
+@pytest.mark.asyncio
+async def test_setup_entry(hass: HomeAssistant, mock_config_entry_data, mock_adguard_api) -> None:
     """Test successful setup of entry."""
     with patch(
         "custom_components.adguard_control_hub.AdGuardHomeAPI"
     ) as mock_api_class:
         mock_api_class.return_value = mock_adguard_api
 
-        entry = hass.config_entries.async_add(mock_config_entry)
+        # Create a mock config entry
+        from homeassistant.config_entries import ConfigEntry
+        entry = ConfigEntry(
+            version=1,
+            minor_version=1,
+            domain=DOMAIN,
+            title="Test AdGuard",
+            data=mock_config_entry_data,
+            options={},
+            source="test",
+            unique_id="test_unique_id",
+        )
+
+        # Add entry to hass
+        hass.config_entries._entries[entry.entry_id] = entry
 
         with patch("custom_components.adguard_control_hub.PLATFORMS", []):
-            await hass.config_entries.async_setup(entry.entry_id)
+            result = await hass.config_entries.async_setup(entry.entry_id)
             await hass.async_block_till_done()
 
-        assert entry.state == ConfigEntryState.LOADED
+        assert result is True
         assert DOMAIN in hass.data
 
 
-async def test_unload_entry(hass: HomeAssistant, mock_config_entry, mock_adguard_api) -> None:
+@pytest.mark.asyncio
+async def test_unload_entry(hass: HomeAssistant, mock_config_entry_data, mock_adguard_api) -> None:
     """Test successful unload of entry."""
     with patch(
         "custom_components.adguard_control_hub.AdGuardHomeAPI"
     ) as mock_api_class:
         mock_api_class.return_value = mock_adguard_api
 
-        entry = hass.config_entries.async_add(mock_config_entry)
+        # Create a mock config entry
+        from homeassistant.config_entries import ConfigEntry
+        entry = ConfigEntry(
+            version=1,
+            minor_version=1,
+            domain=DOMAIN,
+            title="Test AdGuard",
+            data=mock_config_entry_data,
+            options={},
+            source="test",
+            unique_id="test_unique_id",
+        )
+
+        # Add entry to hass
+        hass.config_entries._entries[entry.entry_id] = entry
 
         with patch("custom_components.adguard_control_hub.PLATFORMS", []):
+            # Setup first
             await hass.config_entries.async_setup(entry.entry_id)
             await hass.async_block_till_done()
 
-            assert entry.state == ConfigEntryState.LOADED
+            assert DOMAIN in hass.data
 
-            await hass.config_entries.async_unload(entry.entry_id)
+            # Then unload
+            result = await hass.config_entries.async_unload(entry.entry_id)
             await hass.async_block_till_done()
 
-            assert entry.state == ConfigEntryState.NOT_LOADED
+            assert result is True