From 4cd222e70aa241c4d9e644581f3e2e49942081de Mon Sep 17 00:00:00 2001
From: Paulus Schoutsen <balloob@gmail.com>
Date: Tue, 18 Jan 2022 12:02:37 -0800
Subject: [PATCH] Add evil genius labs diagnostics (#64339)

---
 .../evil_genius_labs/diagnostics.py           | 24 +++++++++++++++++++
 tests/components/diagnostics/__init__.py      | 23 ++++++++++++++++++
 tests/components/diagnostics/test_init.py     | 17 +++++--------
 .../evil_genius_labs/test_diagnostics.py      | 19 +++++++++++++++
 4 files changed, 72 insertions(+), 11 deletions(-)
 create mode 100644 homeassistant/components/evil_genius_labs/diagnostics.py
 create mode 100644 tests/components/evil_genius_labs/test_diagnostics.py

diff --git a/homeassistant/components/evil_genius_labs/diagnostics.py b/homeassistant/components/evil_genius_labs/diagnostics.py
new file mode 100644
index 00000000000..d7439c29d9b
--- /dev/null
+++ b/homeassistant/components/evil_genius_labs/diagnostics.py
@@ -0,0 +1,24 @@
+"""Diagnostics support for Evil Genius Labs."""
+from __future__ import annotations
+
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.core import HomeAssistant
+
+from . import EvilGeniusUpdateCoordinator
+from .const import DOMAIN
+
+
+async def async_get_config_entry_diagnostics(
+    hass: HomeAssistant, config_entry: ConfigEntry
+) -> dict:
+    """Return diagnostics for a config entry."""
+    coordinator: EvilGeniusUpdateCoordinator = hass.data[DOMAIN][config_entry.entry_id]
+
+    return {
+        "info": {
+            **coordinator.info,
+            "wiFiSsidDefault": "REDACTED",
+            "wiFiSSID": "REDACTED",
+        },
+        "data": coordinator.data,
+    }
diff --git a/tests/components/diagnostics/__init__.py b/tests/components/diagnostics/__init__.py
index 620640cd217..5cf56913f60 100644
--- a/tests/components/diagnostics/__init__.py
+++ b/tests/components/diagnostics/__init__.py
@@ -1 +1,24 @@
 """Tests for the Diagnostics integration."""
+from http import HTTPStatus
+
+from homeassistant.setup import async_setup_component
+
+from tests.common import MockConfigEntry
+
+
+async def get_diagnostics_for_config_entry(hass, hass_client, domain_or_config_entry):
+    """Return the diagnostics config entry for the specified domain."""
+    if isinstance(domain_or_config_entry, str):
+        config_entry = MockConfigEntry(domain=domain_or_config_entry)
+        config_entry.add_to_hass(hass)
+    else:
+        config_entry = domain_or_config_entry
+
+    assert await async_setup_component(hass, "diagnostics", {})
+
+    client = await hass_client()
+    response = await client.get(
+        f"/api/diagnostics/config_entry/{config_entry.entry_id}"
+    )
+    assert response.status == HTTPStatus.OK
+    return await response.json()
diff --git a/tests/components/diagnostics/test_init.py b/tests/components/diagnostics/test_init.py
index 5c329b00d8e..60bd28a862a 100644
--- a/tests/components/diagnostics/test_init.py
+++ b/tests/components/diagnostics/test_init.py
@@ -1,5 +1,4 @@
 """Test the Diagnostics integration."""
-from http import HTTPStatus
 from unittest.mock import AsyncMock, Mock
 
 import pytest
@@ -7,7 +6,9 @@ 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
+from . import get_diagnostics_for_config_entry
+
+from tests.common import mock_platform
 
 
 @pytest.fixture(autouse=True)
@@ -45,12 +46,6 @@ async def test_websocket_info(hass, hass_ws_client):
 
 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"}
+    assert await get_diagnostics_for_config_entry(
+        hass, hass_client, "fake_integration"
+    ) == {"hello": "info"}
diff --git a/tests/components/evil_genius_labs/test_diagnostics.py b/tests/components/evil_genius_labs/test_diagnostics.py
new file mode 100644
index 00000000000..980cfa4eab8
--- /dev/null
+++ b/tests/components/evil_genius_labs/test_diagnostics.py
@@ -0,0 +1,19 @@
+"""Test evil genius labs diagnostics."""
+import pytest
+
+from tests.components.diagnostics import get_diagnostics_for_config_entry
+
+
+@pytest.mark.parametrize("platforms", [[]])
+async def test_entry_diagnostics(
+    hass, hass_client, setup_evil_genius_labs, config_entry, data_fixture, info_fixture
+):
+    """Test config entry diagnostics."""
+    assert await get_diagnostics_for_config_entry(hass, hass_client, config_entry) == {
+        "info": {
+            **info_fixture,
+            "wiFiSsidDefault": "REDACTED",
+            "wiFiSSID": "REDACTED",
+        },
+        "data": data_fixture,
+    }