1
mirror of https://github.com/home-assistant/core synced 2024-08-28 03:36:46 +02:00
ha-core/homeassistant/components/minecraft_server/diagnostics.py
elmurato 370e3166ee
Add diagnostics support in Minecraft Server (#101787)
* Add diagnostics support

* Return diagnostics dict directly

* Use syrupy snapshots for assertions in diagnostics test

* Use parametrize for testing diagnostics

* Remove unnecessary side_effect in patch
2023-10-13 15:40:50 +02:00

32 lines
1.1 KiB
Python

"""Diagnostics for the Minecraft Server integration."""
from collections.abc import Iterable
from dataclasses import asdict
from typing import Any
from homeassistant.components.diagnostics.util import async_redact_data
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_ADDRESS, CONF_NAME
from homeassistant.core import HomeAssistant
from .const import DOMAIN
TO_REDACT: Iterable[Any] = {CONF_ADDRESS, CONF_NAME, "players_list"}
async def async_get_config_entry_diagnostics(
hass: HomeAssistant, config_entry: ConfigEntry
) -> dict[str, Any]:
"""Return diagnostics for a config entry."""
coordinator = hass.data[DOMAIN][config_entry.entry_id]
return {
"config_entry": {
"version": config_entry.version,
"unique_id": config_entry.unique_id,
"entry_id": config_entry.entry_id,
},
"config_entry_data": async_redact_data(config_entry.data, TO_REDACT),
"config_entry_options": async_redact_data(config_entry.options, TO_REDACT),
"server_data": async_redact_data(asdict(coordinator.data), TO_REDACT),
}