From 13a9efb6a6045bbe9afc3f01be12cbd66ba3a332 Mon Sep 17 00:00:00 2001 From: "Mr. Bubbles" Date: Wed, 26 Jun 2024 13:36:01 +0200 Subject: [PATCH] Convert dataclass to dict in pyLoad diagnostics (#120552) --- homeassistant/components/pyload/diagnostics.py | 3 ++- .../components/pyload/snapshots/test_diagnostics.ambr | 11 +++++++++-- 2 files changed, 11 insertions(+), 3 deletions(-) diff --git a/homeassistant/components/pyload/diagnostics.py b/homeassistant/components/pyload/diagnostics.py index d18e5a5fe0d..95ff37bf9f8 100644 --- a/homeassistant/components/pyload/diagnostics.py +++ b/homeassistant/components/pyload/diagnostics.py @@ -2,6 +2,7 @@ from __future__ import annotations +from dataclasses import asdict from typing import Any from homeassistant.components.diagnostics import async_redact_data @@ -22,5 +23,5 @@ async def async_get_config_entry_diagnostics( return { "config_entry_data": async_redact_data(dict(config_entry.data), TO_REDACT), - "pyload_data": pyload_data, + "pyload_data": asdict(pyload_data), } diff --git a/tests/components/pyload/snapshots/test_diagnostics.ambr b/tests/components/pyload/snapshots/test_diagnostics.ambr index 8c3e110f2ec..0e078e000c9 100644 --- a/tests/components/pyload/snapshots/test_diagnostics.ambr +++ b/tests/components/pyload/snapshots/test_diagnostics.ambr @@ -10,8 +10,15 @@ 'verify_ssl': False, }), 'pyload_data': dict({ - '__type': "", - 'repr': 'pyLoadData(pause=False, active=1, queue=6, total=37, speed=5405963.0, download=True, reconnect=False, captcha=False, free_space=99999999999)', + 'active': 1, + 'captcha': False, + 'download': True, + 'free_space': 99999999999, + 'pause': False, + 'queue': 6, + 'reconnect': False, + 'speed': 5405963.0, + 'total': 37, }), }) # ---