core/tests/components/prosegur/test_diagnostics.py

22 lines
634 B
Python

"""Test Prosegur diagnostics."""
from unittest.mock import patch
from tests.components.diagnostics import get_diagnostics_for_config_entry
async def test_diagnostics(hass, hass_client, init_integration, mock_install):
"""Test generating diagnostics for a config entry."""
with patch(
"pyprosegur.installation.Installation.retrieve", return_value=mock_install
):
diag = await get_diagnostics_for_config_entry(
hass, hass_client, init_integration
)
assert diag == {
"installation": {"contract": "1234abcd"},
"activity": {"event": "armed"},
}