44 lines
1.1 KiB
Python
44 lines
1.1 KiB
Python
"""Test API functionality."""
|
|
import pytest
|
|
from unittest.mock import AsyncMock, MagicMock
|
|
from custom_components.adguard_hub.api import AdGuardHomeAPI
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_session():
|
|
"""Mock aiohttp session."""
|
|
session = MagicMock()
|
|
response = MagicMock()
|
|
response.raise_for_status = MagicMock()
|
|
response.json = AsyncMock(return_value={"status": "ok"})
|
|
response.status = 200
|
|
response.content_length = 100
|
|
session.request = AsyncMock(return_value=response)
|
|
return session
|
|
|
|
|
|
async def test_api_connection(mock_session):
|
|
"""Test API connection."""
|
|
api = AdGuardHomeAPI(
|
|
host="test-host",
|
|
port=3000,
|
|
username="admin",
|
|
password="password",
|
|
session=mock_session
|
|
)
|
|
|
|
result = await api.test_connection()
|
|
assert result is True
|
|
|
|
|
|
async def test_api_get_status(mock_session):
|
|
"""Test getting status."""
|
|
api = AdGuardHomeAPI(
|
|
host="test-host",
|
|
port=3000,
|
|
session=mock_session
|
|
)
|
|
|
|
status = await api.get_status()
|
|
assert status == {"status": "ok"}
|