diff --git a/tests/components/co2signal/snapshots/test_diagnostics.ambr b/tests/components/co2signal/snapshots/test_diagnostics.ambr new file mode 100644 index 00000000000..ffb35edfbbb --- /dev/null +++ b/tests/components/co2signal/snapshots/test_diagnostics.ambr @@ -0,0 +1,33 @@ +# serializer version: 1 +# name: test_entry_diagnostics + dict({ + 'config_entry': dict({ + 'data': dict({ + 'api_key': '**REDACTED**', + 'location': '', + }), + 'disabled_by': None, + 'domain': 'co2signal', + 'entry_id': '904a74160aa6f335526706bee85dfb83', + 'options': dict({ + }), + 'pref_disable_new_entities': False, + 'pref_disable_polling': False, + 'source': 'user', + 'title': 'Mock Title', + 'unique_id': None, + 'version': 1, + }), + 'data': dict({ + 'countryCode': 'FR', + 'data': dict({ + 'carbonIntensity': 45.98623190095805, + 'fossilFuelPercentage': 5.461182741937103, + }), + 'status': 'ok', + 'units': dict({ + 'carbonIntensity': 'gCO2eq/kWh', + }), + }), + }) +# --- diff --git a/tests/components/co2signal/test_diagnostics.py b/tests/components/co2signal/test_diagnostics.py index c73409fa59b..ed73cb960b5 100644 --- a/tests/components/co2signal/test_diagnostics.py +++ b/tests/components/co2signal/test_diagnostics.py @@ -1,8 +1,9 @@ """Test the CO2Signal diagnostics.""" from unittest.mock import patch +from syrupy import SnapshotAssertion + from homeassistant.components.co2signal import DOMAIN -from homeassistant.components.diagnostics import REDACTED from homeassistant.const import CONF_API_KEY from homeassistant.core import HomeAssistant from homeassistant.setup import async_setup_component @@ -15,11 +16,15 @@ from tests.typing import ClientSessionGenerator async def test_entry_diagnostics( - hass: HomeAssistant, hass_client: ClientSessionGenerator + hass: HomeAssistant, + hass_client: ClientSessionGenerator, + snapshot: SnapshotAssertion, ) -> None: """Test config entry diagnostics.""" config_entry = MockConfigEntry( - domain=DOMAIN, data={CONF_API_KEY: "api_key", "location": ""} + domain=DOMAIN, + data={CONF_API_KEY: "api_key", "location": ""}, + entry_id="904a74160aa6f335526706bee85dfb83", ) config_entry.add_to_hass(hass) with patch("CO2Signal.get_latest", return_value=VALID_PAYLOAD): @@ -27,10 +32,4 @@ async def test_entry_diagnostics( result = await get_diagnostics_for_config_entry(hass, hass_client, config_entry) - config_entry_dict = config_entry.as_dict() - config_entry_dict["data"][CONF_API_KEY] = REDACTED - - assert result == { - "config_entry": config_entry_dict, - "data": VALID_PAYLOAD, - } + assert result == snapshot