1
0
mirror of https://github.com/home-assistant/core.git synced 2026-05-08 17:49:37 +01:00

Small cleanup in sensors test for system monitor (#161616)

This commit is contained in:
Simone Chemelli
2026-01-26 12:30:17 +01:00
committed by GitHub
parent f8797ef06a
commit bb45b23ba9
@@ -2,7 +2,7 @@
from datetime import timedelta
import socket
from unittest.mock import Mock, PropertyMock, patch
from unittest.mock import Mock, patch
from freezegun.api import FrozenDateTimeFactory
from psutil._common import sdiskpart, sdiskusage, shwtemp, snetio, snicaddr
@@ -664,7 +664,7 @@ async def test_sensor_with_param_exception(
entity_id: str,
) -> None:
"""Test the sensor."""
setattr(mock_psutil, psutil_attr, PropertyMock(side_effect=exception))
setattr(mock_psutil, psutil_attr, Mock(side_effect=exception))
mock_config_entry.add_to_hass(hass)
await hass.config_entries.async_setup(mock_config_entry.entry_id)
await hass.async_block_till_done()
@@ -697,7 +697,7 @@ async def test_sensor_without_param_exception(
entity_id: str,
) -> None:
"""Test the sensor."""
setattr(mock_psutil, psutil_attr, PropertyMock(side_effect=exception))
setattr(mock_psutil, psutil_attr, Mock(side_effect=exception))
mock_config_entry.add_to_hass(hass)
await hass.config_entries.async_setup(mock_config_entry.entry_id)
await hass.async_block_till_done()