jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-15 17:27:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use entry.as_dict() in WattTime diagnostics (#80122)

Browse Source
This commit is contained in:
Aaron Bach 2022-10-11 12:15:21 -06:00 committed by GitHub
parent 687987f05b
commit d4465e4e69
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 21 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
homeassistant/components/watttime/diagnostics.py
View File

@ -9,17 +9,25 @@ from homeassistant.const import (
CONF_LATITUDE, CONF_LATITUDE,
CONF_LONGITUDE, CONF_LONGITUDE,
CONF_PASSWORD, CONF_PASSWORD,
CONF_UNIQUE_ID,
CONF_USERNAME, CONF_USERNAME,
) )
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
from .const import DOMAIN from .const import CONF_BALANCING_AUTHORITY, CONF_BALANCING_AUTHORITY_ABBREV, DOMAIN
CONF_TITLE = "title"
TO_REDACT = { TO_REDACT = {
CONF_BALANCING_AUTHORITY,
CONF_BALANCING_AUTHORITY_ABBREV,
CONF_LATITUDE, CONF_LATITUDE,
CONF_LONGITUDE, CONF_LONGITUDE,
CONF_PASSWORD, CONF_PASSWORD,
# Config entry title and unique ID may contain sensitive data:
CONF_TITLE,
CONF_UNIQUE_ID,
CONF_USERNAME, CONF_USERNAME,
} }
@ -32,10 +40,7 @@ async def async_get_config_entry_diagnostics(
return async_redact_data( return async_redact_data(
{ {
"entry": { "entry": entry.as_dict(),
"data": dict(entry.data),
"options": dict(entry.options),
},
"data": coordinator.data, "data": coordinator.data,
}, },
TO_REDACT, TO_REDACT,

13
tests/components/watttime/test_diagnostics.py
View File

@ -8,15 +8,24 @@ async def test_entry_diagnostics(hass, config_entry, hass_client, setup_watttime
"""Test config entry diagnostics.""" """Test config entry diagnostics."""
assert await get_diagnostics_for_config_entry(hass, hass_client, config_entry) == { assert await get_diagnostics_for_config_entry(hass, hass_client, config_entry) == {
"entry": { "entry": {
"entry_id": config_entry.entry_id,
"version": 1,
"domain": "watttime",
"title": REDACTED,
"data": { "data": {
"username": REDACTED, "username": REDACTED,
"password": REDACTED, "password": REDACTED,
"latitude": REDACTED, "latitude": REDACTED,
"longitude": REDACTED, "longitude": REDACTED,
"balancing_authority": "PJM New Jersey", "balancing_authority": REDACTED,
"balancing_authority_abbreviation": "PJM_NJ", "balancing_authority_abbreviation": REDACTED,
}, },
"options": {}, "options": {},
"pref_disable_new_entities": False,
"pref_disable_polling": False,
"source": "user",
"unique_id": REDACTED,
"disabled_by": None,
}, },
"data": { "data": {
"freq": "300", "freq": "300",