Home-Assistant
/
core
mirror of https://github.com/home-assistant/core.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Use freezegun in qnap_qsw tests (#99041)

pull/99058/head
Erik Montnemery 2023-08-25 15:59:08 +02:00 committed by GitHub
parent 64306ec053
commit 65d555b138
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
tests/components/qnap_qsw/test_coordinator.py
View File

@ -3,6 +3,7 @@
from unittest.mock import patch from unittest.mock import patch
from aioqsw.exceptions import APIError, QswError from aioqsw.exceptions import APIError, QswError
from freezegun.api import FrozenDateTimeFactory
from homeassistant.components.qnap_qsw.const import DOMAIN from homeassistant.components.qnap_qsw.const import DOMAIN
from homeassistant.components.qnap_qsw.coordinator import ( from homeassistant.components.qnap_qsw.coordinator import (
@ -11,7 +12,6 @@ from homeassistant.components.qnap_qsw.coordinator import (
) )
from homeassistant.const import STATE_UNAVAILABLE from homeassistant.const import STATE_UNAVAILABLE
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.util.dt import utcnow
from .util import ( from .util import (
CONFIG, CONFIG,
@ -31,7 +31,9 @@ from .util import (
from tests.common import MockConfigEntry, async_fire_time_changed from tests.common import MockConfigEntry, async_fire_time_changed
async def test_coordinator_client_connector_error(hass: HomeAssistant) -> None: async def test_coordinator_client_connector_error(
hass: HomeAssistant, freezer: FrozenDateTimeFactory
) -> None:
"""Test ClientConnectorError on coordinator update.""" """Test ClientConnectorError on coordinator update."""
entry = MockConfigEntry(domain=DOMAIN, data=CONFIG) entry = MockConfigEntry(domain=DOMAIN, data=CONFIG)
@ -99,7 +101,8 @@ async def test_coordinator_client_connector_error(hass: HomeAssistant) -> None:
mock_users_login.reset_mock() mock_users_login.reset_mock()
mock_system_sensor.side_effect = QswError mock_system_sensor.side_effect = QswError
async_fire_time_changed(hass, utcnow() + DATA_SCAN_INTERVAL) freezer.tick(DATA_SCAN_INTERVAL)
async_fire_time_changed(hass)
await hass.async_block_till_done() await hass.async_block_till_done()
mock_system_sensor.assert_called_once() mock_system_sensor.assert_called_once()
@ -110,14 +113,16 @@ async def test_coordinator_client_connector_error(hass: HomeAssistant) -> None:
assert state.state == STATE_UNAVAILABLE assert state.state == STATE_UNAVAILABLE
mock_firmware_update_check.side_effect = APIError mock_firmware_update_check.side_effect = APIError
async_fire_time_changed(hass, utcnow() + FW_SCAN_INTERVAL) freezer.tick(FW_SCAN_INTERVAL)
async_fire_time_changed(hass)
await hass.async_block_till_done() await hass.async_block_till_done()
mock_firmware_update_check.assert_called_once() mock_firmware_update_check.assert_called_once()
mock_firmware_update_check.reset_mock() mock_firmware_update_check.reset_mock()
mock_firmware_update_check.side_effect = QswError mock_firmware_update_check.side_effect = QswError
async_fire_time_changed(hass, utcnow() + FW_SCAN_INTERVAL) freezer.tick(FW_SCAN_INTERVAL)
async_fire_time_changed(hass)
await hass.async_block_till_done() await hass.async_block_till_done()
mock_firmware_update_check.assert_called_once() mock_firmware_update_check.assert_called_once()