update test

moralmunky · Jul 12, 2024 · 6913201 · 6913201
1 parent 8bdc803
commit 6913201
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/test_diagnostics.py b/tests/test_diagnostics.py
@@ -6,7 +6,7 @@
 from homeassistant.const import CONF_HOST, CONF_PASSWORD, CONF_PORT, CONF_USERNAME
 from pytest_homeassistant_custom_component.common import MockConfigEntry
 
-from custom_components.mail_and_packages.const import DOMAIN
+from custom_components.mail_and_packages.const import CONF_AMAZON_FWDS, DOMAIN
 from custom_components.mail_and_packages.diagnostics import (
     async_get_config_entry_diagnostics,
     async_get_device_diagnostics,
@@ -31,6 +31,7 @@ async def test_config_entry_diagnostics(hass):
     assert result["config"]["data"][CONF_PORT] == 993
     assert result["config"]["data"][CONF_PASSWORD] == "**REDACTED**"
     assert result["config"]["data"][CONF_USERNAME] == "**REDACTED**"
+    assert result["config"]["data"][CONF_AMAZON_FWDS] == "**REDACTED**"
 
 
 @pytest.mark.asyncio