57 lines
1.6 KiB
Python
57 lines
1.6 KiB
Python
|
"""Test the Diagnostics integration."""
|
||
|
|
from http import HTTPStatus
|
||
|
|
from unittest.mock import AsyncMock, Mock
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from homeassistant.components.websocket_api.const import TYPE_RESULT
|
||
|
|
from homeassistant.setup import async_setup_component
|
||
|
|
|
||
|
|
from tests.common import MockConfigEntry, mock_platform
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture(autouse=True)
|
||
|
|
async def mock_diagnostics_integration(hass):
|
||
|
|
"""Mock a diagnostics integration."""
|
||
|
|
hass.config.components.add("fake_integration")
|
||
|
|
mock_platform(
|
||
|
|
hass,
|
||
|
|
"fake_integration.diagnostics",
|
||
|
|
Mock(
|
||
|
|
async_get_config_entry_diagnostics=AsyncMock(
|
||
|
|
return_value={
|
||
|
|
"hello": "info",
|
||
|
|
}
|
||
|
|
),
|
||
|
|
),
|
||
|
|
)
|
||
|
|
assert await async_setup_component(hass, "diagnostics", {})
|
||
|
|
|
||
|
|
|
||
|
|
async def test_websocket_info(hass, hass_ws_client):
|
||
|
|
"""Test camera_thumbnail websocket command."""
|
||
|
|
client = await hass_ws_client(hass)
|
||
|
|
await client.send_json({"id": 5, "type": "diagnostics/list"})
|
||
|
|
|
||
|
|
msg = await client.receive_json()
|
||
|
|
|
||
|
|
assert msg["id"] == 5
|
||
|
|
assert msg["type"] == TYPE_RESULT
|
||
|
|
assert msg["success"]
|
||
|
|
assert msg["result"] == [
|
||
|
|
{"domain": "fake_integration", "handlers": {"config_entry": True}}
|
||
|
|
]
|
||
|
|
|
||
|
|
|
||
|
|
async def test_download_diagnostics(hass, hass_client):
|
||
|
|
"""Test record service."""
|
||
|
|
config_entry = MockConfigEntry(domain="fake_integration")
|
||
|
|
config_entry.add_to_hass(hass)
|
||
|
|
|
||
|
|
client = await hass_client()
|
||
|
|
response = await client.get(
|
||
|
|
f"/api/diagnostics/config_entry/{config_entry.entry_id}"
|
||
|
|
)
|
||
|
|
assert response.status == HTTPStatus.OK
|
||
|
|
assert await response.json() == {"hello": "info"}
|