mirror of
https://github.com/elisspace/core.git
synced 2026-09-30 06:19:55 +00:00
Import util.dt as dt_util in components/[e-f]* (#93758)
This commit is contained in:
@@ -11,7 +11,7 @@ from homeassistant.const import CONF_PASSWORD, CONF_USERNAME
|
||||
from homeassistant.core import HomeAssistant
|
||||
from homeassistant.helpers.update_coordinator import UpdateFailed
|
||||
from homeassistant.setup import async_setup_component
|
||||
from homeassistant.util import dt
|
||||
from homeassistant.util import dt as dt_util
|
||||
|
||||
from .common import TEST_PASSWORD, TEST_USER_ID
|
||||
|
||||
@@ -83,7 +83,7 @@ async def test_device(
|
||||
|
||||
call_count = aioclient_mock.call_count
|
||||
|
||||
async_fire_time_changed(hass, dt.utcnow() + timedelta(seconds=90))
|
||||
async_fire_time_changed(hass, dt_util.utcnow() + timedelta(seconds=90))
|
||||
await hass.async_block_till_done()
|
||||
|
||||
assert aioclient_mock.call_count == call_count + 6
|
||||
|
||||
@@ -6,7 +6,7 @@ from homeassistant.config_entries import ConfigEntry
|
||||
from homeassistant.const import CONF_HOST
|
||||
from homeassistant.core import HomeAssistant
|
||||
from homeassistant.helpers import entity_registry as er
|
||||
from homeassistant.util import dt
|
||||
from homeassistant.util import dt as dt_util
|
||||
|
||||
from tests.common import MockConfigEntry, async_fire_time_changed, load_fixture
|
||||
from tests.test_util.aiohttp import AiohttpClientMocker
|
||||
@@ -96,5 +96,5 @@ async def enable_all_entities(hass, config_entry_id, time_till_next_update):
|
||||
]:
|
||||
registry.async_update_entity(entry.entity_id, **{"disabled_by": None})
|
||||
await hass.async_block_till_done()
|
||||
async_fire_time_changed(hass, dt.utcnow() + time_till_next_update)
|
||||
async_fire_time_changed(hass, dt_util.utcnow() + time_till_next_update)
|
||||
await hass.async_block_till_done()
|
||||
|
||||
@@ -7,7 +7,7 @@ from homeassistant.components.fronius.coordinator import (
|
||||
FroniusInverterUpdateCoordinator,
|
||||
)
|
||||
from homeassistant.core import HomeAssistant
|
||||
from homeassistant.util import dt
|
||||
from homeassistant.util import dt as dt_util
|
||||
|
||||
from . import mock_responses, setup_fronius_integration
|
||||
|
||||
@@ -26,7 +26,7 @@ async def test_adaptive_update_interval(
|
||||
mock_inverter_data.reset_mock()
|
||||
|
||||
async_fire_time_changed(
|
||||
hass, dt.utcnow() + FroniusInverterUpdateCoordinator.default_interval
|
||||
hass, dt_util.utcnow() + FroniusInverterUpdateCoordinator.default_interval
|
||||
)
|
||||
await hass.async_block_till_done()
|
||||
mock_inverter_data.assert_called_once()
|
||||
@@ -36,14 +36,15 @@ async def test_adaptive_update_interval(
|
||||
# first 3 bad requests at default interval - 4th has different interval
|
||||
for _ in range(3):
|
||||
async_fire_time_changed(
|
||||
hass, dt.utcnow() + FroniusInverterUpdateCoordinator.default_interval
|
||||
hass,
|
||||
dt_util.utcnow() + FroniusInverterUpdateCoordinator.default_interval,
|
||||
)
|
||||
await hass.async_block_till_done()
|
||||
assert mock_inverter_data.call_count == 3
|
||||
mock_inverter_data.reset_mock()
|
||||
|
||||
async_fire_time_changed(
|
||||
hass, dt.utcnow() + FroniusInverterUpdateCoordinator.error_interval
|
||||
hass, dt_util.utcnow() + FroniusInverterUpdateCoordinator.error_interval
|
||||
)
|
||||
await hass.async_block_till_done()
|
||||
assert mock_inverter_data.call_count == 1
|
||||
@@ -52,14 +53,14 @@ async def test_adaptive_update_interval(
|
||||
mock_inverter_data.side_effect = None
|
||||
# next successful request resets to default interval
|
||||
async_fire_time_changed(
|
||||
hass, dt.utcnow() + FroniusInverterUpdateCoordinator.error_interval
|
||||
hass, dt_util.utcnow() + FroniusInverterUpdateCoordinator.error_interval
|
||||
)
|
||||
await hass.async_block_till_done()
|
||||
mock_inverter_data.assert_called_once()
|
||||
mock_inverter_data.reset_mock()
|
||||
|
||||
async_fire_time_changed(
|
||||
hass, dt.utcnow() + FroniusInverterUpdateCoordinator.default_interval
|
||||
hass, dt_util.utcnow() + FroniusInverterUpdateCoordinator.default_interval
|
||||
)
|
||||
await hass.async_block_till_done()
|
||||
mock_inverter_data.assert_called_once()
|
||||
@@ -70,7 +71,8 @@ async def test_adaptive_update_interval(
|
||||
# first 3 requests at default interval - 4th has different interval
|
||||
for _ in range(3):
|
||||
async_fire_time_changed(
|
||||
hass, dt.utcnow() + FroniusInverterUpdateCoordinator.default_interval
|
||||
hass,
|
||||
dt_util.utcnow() + FroniusInverterUpdateCoordinator.default_interval,
|
||||
)
|
||||
await hass.async_block_till_done()
|
||||
# BadStatusError does 3 silent retries for inverter endpoint * 3 request intervals = 9
|
||||
|
||||
@@ -8,7 +8,7 @@ from homeassistant.components.fronius.coordinator import (
|
||||
from homeassistant.components.sensor import DOMAIN as SENSOR_DOMAIN
|
||||
from homeassistant.core import HomeAssistant
|
||||
from homeassistant.helpers import device_registry as dr
|
||||
from homeassistant.util import dt
|
||||
from homeassistant.util import dt as dt_util
|
||||
|
||||
from . import enable_all_entities, mock_responses, setup_fronius_integration
|
||||
|
||||
@@ -43,7 +43,7 @@ async def test_symo_inverter(
|
||||
# Second test at daytime when inverter is producing
|
||||
mock_responses(aioclient_mock, night=False)
|
||||
async_fire_time_changed(
|
||||
hass, dt.utcnow() + FroniusInverterUpdateCoordinator.default_interval
|
||||
hass, dt_util.utcnow() + FroniusInverterUpdateCoordinator.default_interval
|
||||
)
|
||||
await hass.async_block_till_done()
|
||||
assert len(hass.states.async_all(domain_filter=SENSOR_DOMAIN)) == 56
|
||||
@@ -65,7 +65,7 @@ async def test_symo_inverter(
|
||||
# Third test at nighttime - additional AC entities default to 0
|
||||
mock_responses(aioclient_mock, night=True)
|
||||
async_fire_time_changed(
|
||||
hass, dt.utcnow() + FroniusInverterUpdateCoordinator.default_interval
|
||||
hass, dt_util.utcnow() + FroniusInverterUpdateCoordinator.default_interval
|
||||
)
|
||||
await hass.async_block_till_done()
|
||||
assert_state("sensor.symo_20_ac_current", 0)
|
||||
@@ -150,7 +150,7 @@ async def test_symo_power_flow(
|
||||
hass: HomeAssistant, aioclient_mock: AiohttpClientMocker
|
||||
) -> None:
|
||||
"""Test Fronius Symo power flow entities."""
|
||||
async_fire_time_changed(hass, dt.utcnow())
|
||||
async_fire_time_changed(hass, dt_util.utcnow())
|
||||
|
||||
def assert_state(entity_id, expected_state):
|
||||
state = hass.states.get(entity_id)
|
||||
@@ -176,7 +176,7 @@ async def test_symo_power_flow(
|
||||
# Second test at daytime when inverter is producing
|
||||
mock_responses(aioclient_mock, night=False)
|
||||
async_fire_time_changed(
|
||||
hass, dt.utcnow() + FroniusPowerFlowUpdateCoordinator.default_interval
|
||||
hass, dt_util.utcnow() + FroniusPowerFlowUpdateCoordinator.default_interval
|
||||
)
|
||||
await hass.async_block_till_done()
|
||||
# 54 because power_flow `rel_SelfConsumption` and `P_PV` is not `null` anymore
|
||||
@@ -193,7 +193,7 @@ async def test_symo_power_flow(
|
||||
# Third test at nighttime - default values are used
|
||||
mock_responses(aioclient_mock, night=True)
|
||||
async_fire_time_changed(
|
||||
hass, dt.utcnow() + FroniusPowerFlowUpdateCoordinator.default_interval
|
||||
hass, dt_util.utcnow() + FroniusPowerFlowUpdateCoordinator.default_interval
|
||||
)
|
||||
await hass.async_block_till_done()
|
||||
assert len(hass.states.async_all(domain_filter=SENSOR_DOMAIN)) == 54
|
||||
|
||||
@@ -21,7 +21,7 @@ from homeassistant.components.websocket_api.const import TYPE_RESULT
|
||||
from homeassistant.core import HomeAssistant
|
||||
from homeassistant.loader import async_get_integration
|
||||
from homeassistant.setup import async_setup_component
|
||||
from homeassistant.util import dt
|
||||
from homeassistant.util import dt as dt_util
|
||||
|
||||
from tests.common import MockUser, async_capture_events, async_fire_time_changed
|
||||
from tests.typing import WebSocketGenerator
|
||||
@@ -228,7 +228,7 @@ async def test_themes_save_storage(
|
||||
)
|
||||
|
||||
# To trigger the call_later
|
||||
async_fire_time_changed(hass, dt.utcnow() + timedelta(seconds=60))
|
||||
async_fire_time_changed(hass, dt_util.utcnow() + timedelta(seconds=60))
|
||||
# To execute the save
|
||||
await hass.async_block_till_done()
|
||||
|
||||
|
||||
@@ -15,7 +15,7 @@ from homeassistant.const import (
|
||||
)
|
||||
from homeassistant.core import HomeAssistant
|
||||
from homeassistant.helpers import device_registry as dr, entity_registry as er
|
||||
from homeassistant.util import dt
|
||||
from homeassistant.util import dt as dt_util
|
||||
|
||||
from tests.common import MockConfigEntry, async_fire_time_changed
|
||||
|
||||
@@ -76,7 +76,7 @@ async def test_binary_sensors(
|
||||
|
||||
# Test unknown/missing data
|
||||
mock_fully_kiosk.getDeviceInfo.return_value = {}
|
||||
async_fire_time_changed(hass, dt.utcnow() + UPDATE_INTERVAL)
|
||||
async_fire_time_changed(hass, dt_util.utcnow() + UPDATE_INTERVAL)
|
||||
await hass.async_block_till_done()
|
||||
|
||||
state = hass.states.get("binary_sensor.amazon_fire_plugged_in")
|
||||
@@ -85,7 +85,7 @@ async def test_binary_sensors(
|
||||
|
||||
# Test failed update
|
||||
mock_fully_kiosk.getDeviceInfo.side_effect = FullyKioskError("error", "status")
|
||||
async_fire_time_changed(hass, dt.utcnow() + UPDATE_INTERVAL)
|
||||
async_fire_time_changed(hass, dt_util.utcnow() + UPDATE_INTERVAL)
|
||||
await hass.async_block_till_done()
|
||||
|
||||
state = hass.states.get("binary_sensor.amazon_fire_plugged_in")
|
||||
|
||||
@@ -6,7 +6,7 @@ import homeassistant.components.number as number
|
||||
from homeassistant.const import ATTR_ENTITY_ID, STATE_UNKNOWN
|
||||
from homeassistant.core import HomeAssistant
|
||||
from homeassistant.helpers import device_registry as dr, entity_registry as er
|
||||
from homeassistant.util import dt
|
||||
from homeassistant.util import dt as dt_util
|
||||
|
||||
from tests.common import MockConfigEntry, async_fire_time_changed
|
||||
|
||||
@@ -52,7 +52,7 @@ async def test_numbers(
|
||||
|
||||
# Test invalid numeric data
|
||||
mock_fully_kiosk.getSettings.return_value = {"screenBrightness": "invalid"}
|
||||
async_fire_time_changed(hass, dt.utcnow() + UPDATE_INTERVAL)
|
||||
async_fire_time_changed(hass, dt_util.utcnow() + UPDATE_INTERVAL)
|
||||
await hass.async_block_till_done()
|
||||
|
||||
state = hass.states.get("number.amazon_fire_screen_brightness")
|
||||
@@ -61,7 +61,7 @@ async def test_numbers(
|
||||
|
||||
# Test unknown/missing data
|
||||
mock_fully_kiosk.getSettings.return_value = {}
|
||||
async_fire_time_changed(hass, dt.utcnow() + UPDATE_INTERVAL)
|
||||
async_fire_time_changed(hass, dt_util.utcnow() + UPDATE_INTERVAL)
|
||||
await hass.async_block_till_done()
|
||||
|
||||
state = hass.states.get("number.amazon_fire_screensaver_timer")
|
||||
|
||||
@@ -18,7 +18,7 @@ from homeassistant.const import (
|
||||
)
|
||||
from homeassistant.core import HomeAssistant
|
||||
from homeassistant.helpers import device_registry as dr, entity_registry as er
|
||||
from homeassistant.util import dt
|
||||
from homeassistant.util import dt as dt_util
|
||||
|
||||
from tests.common import MockConfigEntry, async_fire_time_changed
|
||||
|
||||
@@ -141,7 +141,7 @@ async def test_sensors_sensors(
|
||||
|
||||
# Test unknown/missing data
|
||||
mock_fully_kiosk.getDeviceInfo.return_value = {}
|
||||
async_fire_time_changed(hass, dt.utcnow() + UPDATE_INTERVAL)
|
||||
async_fire_time_changed(hass, dt_util.utcnow() + UPDATE_INTERVAL)
|
||||
await hass.async_block_till_done()
|
||||
|
||||
state = hass.states.get("sensor.amazon_fire_internal_storage_free_space")
|
||||
@@ -150,7 +150,7 @@ async def test_sensors_sensors(
|
||||
|
||||
# Test failed update
|
||||
mock_fully_kiosk.getDeviceInfo.side_effect = FullyKioskError("error", "status")
|
||||
async_fire_time_changed(hass, dt.utcnow() + UPDATE_INTERVAL)
|
||||
async_fire_time_changed(hass, dt_util.utcnow() + UPDATE_INTERVAL)
|
||||
await hass.async_block_till_done()
|
||||
|
||||
state = hass.states.get("sensor.amazon_fire_internal_storage_free_space")
|
||||
@@ -176,7 +176,7 @@ async def test_url_sensor_truncating(
|
||||
mock_fully_kiosk.getDeviceInfo.return_value = {
|
||||
"currentPage": long_url,
|
||||
}
|
||||
async_fire_time_changed(hass, dt.utcnow() + UPDATE_INTERVAL)
|
||||
async_fire_time_changed(hass, dt_util.utcnow() + UPDATE_INTERVAL)
|
||||
await hass.async_block_till_done()
|
||||
|
||||
state = hass.states.get("sensor.amazon_fire_current_page")
|
||||
|
||||
Reference in New Issue
Block a user