From 6f3b7d009d4577a480d33d52c89f2f5dd52f92a9 Mon Sep 17 00:00:00 2001
From: Shay Levy <levyshay1@gmail.com>
Date: Fri, 28 Oct 2022 15:48:16 +0300
Subject: [PATCH] Add diagnostics to webostv (#81133)

---
 .../components/webostv/diagnostics.py         | 52 ++++++++++++++++
 tests/components/webostv/conftest.py          |  2 +
 tests/components/webostv/test_diagnostics.py  | 61 +++++++++++++++++++
 3 files changed, 115 insertions(+)
 create mode 100644 homeassistant/components/webostv/diagnostics.py
 create mode 100644 tests/components/webostv/test_diagnostics.py

diff --git a/homeassistant/components/webostv/diagnostics.py b/homeassistant/components/webostv/diagnostics.py
new file mode 100644
index 00000000000..ce62f51b540
--- /dev/null
+++ b/homeassistant/components/webostv/diagnostics.py
@@ -0,0 +1,52 @@
+"""Diagnostics support for LG webOS Smart TV."""
+from __future__ import annotations
+
+from typing import Any
+
+from aiowebostv import WebOsClient
+
+from homeassistant.components.diagnostics import async_redact_data
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.const import CONF_CLIENT_SECRET, CONF_HOST, CONF_UNIQUE_ID
+from homeassistant.core import HomeAssistant
+
+from .const import DATA_CONFIG_ENTRY, DOMAIN
+
+TO_REDACT = {
+    CONF_CLIENT_SECRET,
+    CONF_UNIQUE_ID,
+    CONF_HOST,
+    "device_id",
+    "deviceUUID",
+    "icon",
+    "largeIcon",
+}
+
+
+async def async_get_config_entry_diagnostics(
+    hass: HomeAssistant, entry: ConfigEntry
+) -> dict[str, Any]:
+    """Return diagnostics for a config entry."""
+    client: WebOsClient = hass.data[DOMAIN][DATA_CONFIG_ENTRY][entry.entry_id].client
+
+    client_data = {
+        "is_registered": client.is_registered(),
+        "is_connected": client.is_connected(),
+        "current_app_id": client.current_app_id,
+        "current_channel": client.current_channel,
+        "apps": client.apps,
+        "inputs": client.inputs,
+        "system_info": client.system_info,
+        "software_info": client.software_info,
+        "hello_info": client.hello_info,
+        "sound_output": client.sound_output,
+        "is_on": client.is_on,
+    }
+
+    return async_redact_data(
+        {
+            "entry": entry.as_dict(),
+            "client": client_data,
+        },
+        TO_REDACT,
+    )
diff --git a/tests/components/webostv/conftest.py b/tests/components/webostv/conftest.py
index 05f1be66d00..c8333c84447 100644
--- a/tests/components/webostv/conftest.py
+++ b/tests/components/webostv/conftest.py
@@ -39,6 +39,8 @@ def client_fixture():
         client.sound_output = "speaker"
         client.muted = False
         client.is_on = True
+        client.is_registered = Mock(return_value=True)
+        client.is_connected = Mock(return_value=True)
 
         async def mock_state_update_callback():
             await client.register_state_update_callback.call_args[0][0](client)
diff --git a/tests/components/webostv/test_diagnostics.py b/tests/components/webostv/test_diagnostics.py
new file mode 100644
index 00000000000..707f83b2fcf
--- /dev/null
+++ b/tests/components/webostv/test_diagnostics.py
@@ -0,0 +1,61 @@
+"""Tests for the diagnostics data provided by LG webOS Smart TV."""
+from aiohttp import ClientSession
+
+from homeassistant.components.diagnostics import REDACTED
+from homeassistant.core import HomeAssistant
+
+from . import setup_webostv
+
+from tests.components.diagnostics import get_diagnostics_for_config_entry
+
+
+async def test_diagnostics(
+    hass: HomeAssistant, hass_client: ClientSession, client
+) -> None:
+    """Test diagnostics."""
+    entry = await setup_webostv(hass)
+    assert await get_diagnostics_for_config_entry(hass, hass_client, entry) == {
+        "client": {
+            "is_registered": True,
+            "is_connected": True,
+            "current_app_id": "com.webos.app.livetv",
+            "current_channel": {
+                "channelId": "ch1id",
+                "channelName": "Channel 1",
+                "channelNumber": "1",
+            },
+            "apps": {
+                "com.webos.app.livetv": {
+                    "icon": REDACTED,
+                    "id": "com.webos.app.livetv",
+                    "largeIcon": REDACTED,
+                    "title": "Live TV",
+                }
+            },
+            "inputs": {
+                "in1": {"appId": "app0", "id": "in1", "label": "Input01"},
+                "in2": {"appId": "app1", "id": "in2", "label": "Input02"},
+            },
+            "system_info": {"modelName": "TVFAKE"},
+            "software_info": {"major_ver": "major", "minor_ver": "minor"},
+            "hello_info": {"deviceUUID": "**REDACTED**"},
+            "sound_output": "speaker",
+            "is_on": True,
+        },
+        "entry": {
+            "entry_id": entry.entry_id,
+            "version": 1,
+            "domain": "webostv",
+            "title": "fake_webos",
+            "data": {
+                "client_secret": "**REDACTED**",
+                "host": "**REDACTED**",
+            },
+            "options": {},
+            "pref_disable_new_entities": False,
+            "pref_disable_polling": False,
+            "source": "user",
+            "unique_id": REDACTED,
+            "disabled_by": None,
+        },
+    }