Add diagnostics integration (#64330)

tests/components/diagnostics/test_init.py

@@ -0,0 +1,56 @@
+"""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"}