Implement option to download diagnostics (#1218)

custom_components/powercalc/diagnostics.py

@@ -0,0 +1,11 @@
+from homeassistant.core import HomeAssistant
+from homeassistant.config_entries import ConfigEntry
+
+
+async def async_get_config_entry_diagnostics(
+    hass: HomeAssistant, entry: ConfigEntry
+) -> dict:
+    """Return diagnostics for a config entry."""
+    return {
+        "entry": entry.as_dict(),
+    }

tests/conftest.py

@@ -4,8 +4,16 @@
     mock_area_registry,
     mock_device_registry,
     mock_registry,
+    MockConfigEntry
+)
+from homeassistant.const import CONF_ENTITY_ID
+from custom_components.powercalc.const import (
+    DOMAIN,
+    CONF_FIXED,
+    CONF_POWER,
+    CONF_SENSOR_TYPE,
+    SensorType
 )
-
 
 @pytest.fixture(autouse=True)
 def auto_enable_custom_integrations(enable_custom_integrations):
@@ -34,3 +42,20 @@ def device_reg(hass):
 def entity_reg(hass):
     """Return an empty, loaded, registry."""
     return mock_registry(hass)
+
+
+@pytest.fixture
+def mock_config_entry() -> MockConfigEntry:
+    """Return the default mocked config entry."""
+    return MockConfigEntry(
+        domain=DOMAIN,
+        data={
+            CONF_SENSOR_TYPE: SensorType.VIRTUAL_POWER,
+            CONF_ENTITY_ID: "input_boolean.test",
+            CONF_FIXED: {
+                CONF_POWER: 50,
+            }
+        },
+        unique_id="aabbccddeeff",
+        title="test"
+    )

tests/test_diagnostics.py

@@ -0,0 +1,10 @@
+from homeassistant.core import HomeAssistant
+from pytest_homeassistant_custom_component.common import MockConfigEntry
+from custom_components.powercalc.diagnostics import async_get_config_entry_diagnostics
+
+
+async def test_diagnostics(hass: HomeAssistant, mock_config_entry: MockConfigEntry) -> None:
+    diagnostics_data = await async_get_config_entry_diagnostics(hass, mock_config_entry)
+    assert diagnostics_data == {
+        "entry": mock_config_entry.as_dict()
+    }