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

Import `util.dt` as `dt_util` in `components/[a-d]*` (#93756)

pull/93668/head
Ville Skyttä 2023-05-30 00:02:06 +03:00 committed by GitHub
parent 5aadd7ffc9
commit ced67e5800
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
24 changed files with 121 additions and 105 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
homeassistant/components/alexa/auth.py
View File

@ -12,7 +12,7 @@ from homeassistant.const import CONF_CLIENT_ID, CONF_CLIENT_SECRET
from homeassistant.core import callback
from homeassistant.helpers import aiohttp_client
from homeassistant.helpers.storage import Store
from homeassistant.util import dt
from homeassistant.util import dt as dt_util
_LOGGER = logging.getLogger(__name__)
@ -95,12 +95,12 @@ class Auth:
if not self._prefs[STORAGE_ACCESS_TOKEN]:
return False
expire_time = dt.parse_datetime(self._prefs[STORAGE_EXPIRE_TIME])
expire_time = dt_util.parse_datetime(self._prefs[STORAGE_EXPIRE_TIME])
preemptive_expire_time = expire_time - timedelta(
seconds=PREEMPTIVE_REFRESH_TTL_IN_SECONDS
)
return dt.utcnow() < preemptive_expire_time
return dt_util.utcnow() < preemptive_expire_time
async def _async_request_new_token(self, lwa_params):
try:
@ -130,7 +130,7 @@ class Auth:
access_token = response_json["access_token"]
refresh_token = response_json["refresh_token"]
expires_in = response_json["expires_in"]
expire_time = dt.utcnow() + timedelta(seconds=expires_in)
expire_time = dt_util.utcnow() + timedelta(seconds=expires_in)
await self._async_update_preferences(
access_token, refresh_token, expire_time.isoformat()

4
homeassistant/components/backup/manager.py
View File

@ -18,7 +18,7 @@ from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers import integration_platform
from homeassistant.helpers.json import save_json
from homeassistant.util import dt
from homeassistant.util import dt as dt_util
from homeassistant.util.json import json_loads_object
from .const import DOMAIN, EXCLUDE_FROM_BACKUP, LOGGER
@ -176,7 +176,7 @@ class BackupManager:
raise result
backup_name = f"Core {HAVERSION}"
date_str = dt.now().isoformat()
date_str = dt_util.now().isoformat()
slug = _generate_slug(date_str, backup_name)
backup_data = {

14
homeassistant/components/broadlink/remote.py
View File

@ -39,7 +39,7 @@ from homeassistant.helpers import config_validation as cv
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.restore_state import RestoreEntity
from homeassistant.helpers.storage import Store
from homeassistant.util import dt
from homeassistant.util import dt as dt_util
from .const import DOMAIN
from .entity import BroadlinkEntity
@ -330,8 +330,8 @@ class BroadlinkRemote(BroadlinkEntity, RemoteEntity, RestoreEntity):
)
try:
start_time = dt.utcnow()
while (dt.utcnow() - start_time) < LEARNING_TIMEOUT:
start_time = dt_util.utcnow()
while (dt_util.utcnow() - start_time) < LEARNING_TIMEOUT:
await asyncio.sleep(1)
try:
code = await device.async_request(device.api.check_data)
@ -368,8 +368,8 @@ class BroadlinkRemote(BroadlinkEntity, RemoteEntity, RestoreEntity):
)
try:
start_time = dt.utcnow()
while (dt.utcnow() - start_time) < LEARNING_TIMEOUT:
start_time = dt_util.utcnow()
while (dt_util.utcnow() - start_time) < LEARNING_TIMEOUT:
await asyncio.sleep(1)
found = await device.async_request(device.api.check_frequency)
if found:
@ -403,8 +403,8 @@ class BroadlinkRemote(BroadlinkEntity, RemoteEntity, RestoreEntity):
)
try:
start_time = dt.utcnow()
while (dt.utcnow() - start_time) < LEARNING_TIMEOUT:
start_time = dt_util.utcnow()
while (dt_util.utcnow() - start_time) < LEARNING_TIMEOUT:
await asyncio.sleep(1)
try:
code = await device.async_request(device.api.check_data)

6
homeassistant/components/broadlink/updater.py
View File

@ -6,7 +6,7 @@ import logging
from broadlink.exceptions import AuthorizationError, BroadlinkException
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
from homeassistant.util import dt
from homeassistant.util import dt as dt_util
_LOGGER = logging.getLogger(__name__)
@ -64,7 +64,7 @@ class BroadlinkUpdateManager(ABC):
except (BroadlinkException, OSError) as err:
if self.available and (
dt.utcnow() - self.last_update > self.SCAN_INTERVAL * 3
dt_util.utcnow() - self.last_update > self.SCAN_INTERVAL * 3
or isinstance(err, (AuthorizationError, OSError))
):
self.available = False
@ -84,7 +84,7 @@ class BroadlinkUpdateManager(ABC):
self.device.api.host[0],
)
self.available = True
self.last_update = dt.utcnow()
self.last_update = dt_util.utcnow()
return data
@abstractmethod

12
homeassistant/components/caldav/calendar.py
View File

@ -29,7 +29,7 @@ import homeassistant.helpers.config_validation as cv
from homeassistant.helpers.entity import generate_entity_id
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from homeassistant.util import Throttle, dt
from homeassistant.util import Throttle, dt as dt_util
_LOGGER = logging.getLogger(__name__)
@ -204,8 +204,8 @@ class WebDavCalendarData:
@Throttle(MIN_TIME_BETWEEN_UPDATES)
def update(self):
"""Get the latest data."""
start_of_today = dt.start_of_local_day()
start_of_tomorrow = dt.start_of_local_day() + timedelta(days=self.days)
start_of_today = dt_util.start_of_local_day()
start_of_tomorrow = dt_util.start_of_local_day() + timedelta(days=self.days)
# We have to retrieve the results for the whole day as the server
# won't return events that have already started
@ -312,7 +312,7 @@ class WebDavCalendarData:
@staticmethod
def is_over(vevent):
"""Return if the event is over."""
return dt.now() >= WebDavCalendarData.to_datetime(
return dt_util.now() >= WebDavCalendarData.to_datetime(
WebDavCalendarData.get_end_date(vevent)
)
@ -321,7 +321,7 @@ class WebDavCalendarData:
"""Return a datetime."""
if isinstance(obj, datetime):
return WebDavCalendarData.to_local(obj)
return datetime.combine(obj, time.min).replace(tzinfo=dt.DEFAULT_TIME_ZONE)
return datetime.combine(obj, time.min).replace(tzinfo=dt_util.DEFAULT_TIME_ZONE)
@staticmethod
def to_local(obj: datetime | date) -> datetime | date:
@ -332,7 +332,7 @@ class WebDavCalendarData:
used by the caldav client and dateutil so the datetime can be copied.
"""
if isinstance(obj, datetime):
return dt.as_local(obj)
return dt_util.as_local(obj)
return obj
@staticmethod

32
homeassistant/components/calendar/__init__.py
View File

@ -31,7 +31,7 @@ from homeassistant.helpers.entity import Entity
from homeassistant.helpers.entity_component import EntityComponent
from homeassistant.helpers.template import DATE_STR_FORMAT
from homeassistant.helpers.typing import ConfigType
from homeassistant.util import dt
from homeassistant.util import dt as dt_util
from .const import (
CONF_EVENT,
@ -117,7 +117,7 @@ def _as_local_timezone(*keys: Any) -> Callable[[dict[str, Any]], dict[str, Any]]
"""Convert all keys that are datetime values to local timezone."""
for k in keys:
if (value := obj.get(k)) and isinstance(value, datetime.datetime):
obj[k] = dt.as_local(value)
obj[k] = dt_util.as_local(value)
return obj
return validate
@ -294,14 +294,14 @@ async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
def get_date(date: dict[str, Any]) -> datetime.datetime:
"""Get the dateTime from date or dateTime as a local."""
if "date" in date:
parsed_date = dt.parse_date(date["date"])
parsed_date = dt_util.parse_date(date["date"])
assert parsed_date
return dt.start_of_local_day(
return dt_util.start_of_local_day(
datetime.datetime.combine(parsed_date, datetime.time.min)
)
parsed_datetime = dt.parse_datetime(date["dateTime"])
parsed_datetime = dt_util.parse_datetime(date["dateTime"])
assert parsed_datetime
return dt.as_local(parsed_datetime)
return dt_util.as_local(parsed_datetime)
@dataclasses.dataclass
@ -380,7 +380,7 @@ def _api_event_dict_factory(obj: Iterable[tuple[str, Any]]) -> dict[str, Any]:
result: dict[str, Any] = {}
for name, value in obj:
if isinstance(value, datetime.datetime):
result[name] = {"dateTime": dt.as_local(value).isoformat()}
result[name] = {"dateTime": dt_util.as_local(value).isoformat()}
elif isinstance(value, datetime.date):
result[name] = {"date": value.isoformat()}
else:
@ -393,14 +393,14 @@ def _get_datetime_local(
) -> datetime.datetime:
"""Convert a calendar event date/datetime to a datetime if needed."""
if isinstance(dt_or_d, datetime.datetime):
return dt.as_local(dt_or_d)
return dt.start_of_local_day(dt_or_d)
return dt_util.as_local(dt_or_d)
return dt_util.start_of_local_day(dt_or_d)
def _get_api_date(dt_or_d: datetime.datetime | datetime.date) -> dict[str, str]:
"""Convert a calendar event date/datetime to a datetime if needed."""
if isinstance(dt_or_d, datetime.datetime):
return {"dateTime": dt.as_local(dt_or_d).isoformat()}
return {"dateTime": dt_util.as_local(dt_or_d).isoformat()}
return {"date": dt_or_d.isoformat()}
@ -433,7 +433,7 @@ def is_offset_reached(
"""Have we reached the offset time specified in the event title."""
if offset_time == datetime.timedelta():
return False
return start + offset_time <= dt.now(start.tzinfo)
return start + offset_time <= dt_util.now(start.tzinfo)
class CalendarEntity(Entity):
@ -467,7 +467,7 @@ class CalendarEntity(Entity):
if (event := self.event) is None:
return STATE_OFF
now = dt.now()
now = dt_util.now()
if event.start_datetime_local <= now < event.end_datetime_local:
return STATE_ON
@ -529,8 +529,8 @@ class CalendarEventView(http.HomeAssistantView):
if start is None or end is None:
return web.Response(status=HTTPStatus.BAD_REQUEST)
try:
start_date = dt.parse_datetime(start)
end_date = dt.parse_datetime(end)
start_date = dt_util.parse_datetime(start)
end_date = dt_util.parse_datetime(end)
except (ValueError, AttributeError):
return web.Response(status=HTTPStatus.BAD_REQUEST)
if start_date is None or end_date is None:
@ -540,7 +540,9 @@ class CalendarEventView(http.HomeAssistantView):
try:
calendar_event_list = await entity.async_get_events(
request.app["hass"], dt.as_local(start_date), dt.as_local(end_date)
request.app["hass"],
dt_util.as_local(start_date),
dt_util.as_local(end_date),
)
except HomeAssistantError as err:
_LOGGER.debug("Error reading events: %s", err)

4
homeassistant/components/cert_expiry/helper.py
View File

@ -3,7 +3,7 @@ import socket
import ssl
from homeassistant.core import HomeAssistant
from homeassistant.util import dt
from homeassistant.util import dt as dt_util
from .const import TIMEOUT
from .errors import (
@ -52,4 +52,4 @@ async def get_cert_expiry_timestamp(
raise ValidationFailure(err.args[0]) from err
ts_seconds = ssl.cert_time_to_seconds(cert["notAfter"])
return dt.utc_from_timestamp(ts_seconds)
return dt_util.utc_from_timestamp(ts_seconds)

6
homeassistant/components/demo/mailbox.py
View File

@ -9,7 +9,7 @@ from typing import Any
from homeassistant.components.mailbox import CONTENT_TYPE_MPEG, Mailbox, StreamError
from homeassistant.core import HomeAssistant
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from homeassistant.util import dt
from homeassistant.util import dt as dt_util
_LOGGER = logging.getLogger(__name__)
@ -34,7 +34,9 @@ class DemoMailbox(Mailbox):
self._messages: dict[str, dict[str, Any]] = {}
txt = "Lorem ipsum dolor sit amet, consectetur adipiscing elit. "
for idx in range(0, 10):
msgtime = int(dt.as_timestamp(dt.utcnow()) - 3600 * 24 * (10 - idx))
msgtime = int(
dt_util.as_timestamp(dt_util.utcnow()) - 3600 * 24 * (10 - idx)
)
msgtxt = f"Message {idx + 1}. {txt * (1 + idx * (idx % 2))}"
msgsha = sha1(msgtxt.encode("utf-8")).hexdigest()
msg = {

6
tests/components/advantage_air/test_binary_sensor.py
View File

@ -5,7 +5,7 @@ from homeassistant.config_entries import RELOAD_AFTER_UPDATE_DELAY
from homeassistant.const import STATE_OFF, STATE_ON
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 . import (
TEST_SET_RESPONSE,
@ -88,7 +88,7 @@ async def test_binary_sensor_async_setup_entry(
async_fire_time_changed(
hass,
dt.utcnow() + timedelta(seconds=RELOAD_AFTER_UPDATE_DELAY + 1),
dt_util.utcnow() + timedelta(seconds=RELOAD_AFTER_UPDATE_DELAY + 1),
)
await hass.async_block_till_done()
@ -110,7 +110,7 @@ async def test_binary_sensor_async_setup_entry(
async_fire_time_changed(
hass,
dt.utcnow() + timedelta(seconds=RELOAD_AFTER_UPDATE_DELAY + 1),
dt_util.utcnow() + timedelta(seconds=RELOAD_AFTER_UPDATE_DELAY + 1),
)
await hass.async_block_till_done()

4
tests/components/advantage_air/test_sensor.py
View File

@ -11,7 +11,7 @@ from homeassistant.config_entries import RELOAD_AFTER_UPDATE_DELAY
from homeassistant.const import ATTR_ENTITY_ID
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 . import (
TEST_SET_RESPONSE,
@ -144,7 +144,7 @@ async def test_sensor_platform(
async_fire_time_changed(
hass,
dt.utcnow() + timedelta(seconds=RELOAD_AFTER_UPDATE_DELAY + 1),
dt_util.utcnow() + timedelta(seconds=RELOAD_AFTER_UPDATE_DELAY + 1),
)
await hass.async_block_till_done()

4
tests/components/anova/test_sensor.py
View File

@ -8,7 +8,7 @@ from anova_wifi import AnovaApi, AnovaOffline
from homeassistant.const import STATE_UNAVAILABLE
from homeassistant.core import HomeAssistant
from homeassistant.util import dt
from homeassistant.util import dt as dt_util
from . import async_init_integration
@ -54,7 +54,7 @@ async def test_update_failed(hass: HomeAssistant, anova_api: AnovaApi) -> None:
"homeassistant.components.anova.AnovaPrecisionCooker.update",
side_effect=AnovaOffline(),
):
async_fire_time_changed(hass, dt.utcnow() + timedelta(seconds=61))
async_fire_time_changed(hass, dt_util.utcnow() + timedelta(seconds=61))
await hass.async_block_till_done()
state = hass.states.get("sensor.anova_precision_cooker_water_temperature")

10
tests/components/broadlink/test_heartbeat.py
View File

@ -5,7 +5,7 @@ import pytest
from homeassistant.components.broadlink.heartbeat import BroadlinkHeartbeat
from homeassistant.core import HomeAssistant
from homeassistant.util import dt
from homeassistant.util import dt as dt_util
from . import get_device
@ -48,7 +48,7 @@ async def test_heartbeat_trigger_right_time(hass: HomeAssistant) -> None:
with patch(DEVICE_PING) as mock_ping:
async_fire_time_changed(
hass, dt.utcnow() + BroadlinkHeartbeat.HEARTBEAT_INTERVAL
hass, dt_util.utcnow() + BroadlinkHeartbeat.HEARTBEAT_INTERVAL
)
await hass.async_block_till_done()
@ -66,7 +66,7 @@ async def test_heartbeat_do_not_trigger_before_time(hass: HomeAssistant) -> None
with patch(DEVICE_PING) as mock_ping:
async_fire_time_changed(
hass,
dt.utcnow() + BroadlinkHeartbeat.HEARTBEAT_INTERVAL // 2,
dt_util.utcnow() + BroadlinkHeartbeat.HEARTBEAT_INTERVAL // 2,
)
await hass.async_block_till_done()
@ -85,7 +85,7 @@ async def test_heartbeat_unload(hass: HomeAssistant) -> None:
with patch(DEVICE_PING) as mock_ping:
async_fire_time_changed(
hass, dt.utcnow() + BroadlinkHeartbeat.HEARTBEAT_INTERVAL
hass, dt_util.utcnow() + BroadlinkHeartbeat.HEARTBEAT_INTERVAL
)
assert mock_ping.call_count == 0
@ -105,7 +105,7 @@ async def test_heartbeat_do_not_unload(hass: HomeAssistant) -> None:
with patch(DEVICE_PING) as mock_ping:
async_fire_time_changed(
hass, dt.utcnow() + BroadlinkHeartbeat.HEARTBEAT_INTERVAL
hass, dt_util.utcnow() + BroadlinkHeartbeat.HEARTBEAT_INTERVAL
)
await hass.async_block_till_done()

6
tests/components/broadlink/test_sensors.py
View File

@ -7,7 +7,7 @@ from homeassistant.const import ATTR_FRIENDLY_NAME, Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers import device_registry as dr, entity_registry as er
from homeassistant.helpers.entity_component import async_update_entity
from homeassistant.util import dt
from homeassistant.util import dt as dt_util
from . import get_device
@ -384,7 +384,9 @@ async def test_scb1e_sensor_update(
}
target_time = (
dt.utcnow() + BroadlinkSP4UpdateManager.SCAN_INTERVAL * 3 + timedelta(seconds=1)
dt_util.utcnow()
+ BroadlinkSP4UpdateManager.SCAN_INTERVAL * 3
+ timedelta(seconds=1)
)
mock_setup = await device.setup_entry(hass, mock_api=mock_api)

36
tests/components/caldav/test_calendar.py
View File

@ -9,7 +9,7 @@ import pytest
from homeassistant.const import STATE_OFF, STATE_ON
from homeassistant.core import HomeAssistant
from homeassistant.setup import async_setup_component
from homeassistant.util import dt
from homeassistant.util import dt as dt_util
DEVICE_DATA = {"name": "Private Calendar", "device_id": "Private Calendar"}
@ -360,7 +360,7 @@ def get_api_events(hass_client):
def _local_datetime(hours, minutes):
"""Build a datetime object for testing in the correct timezone."""
return dt.as_local(datetime.datetime(2017, 11, 27, hours, minutes, 0))
return dt_util.as_local(datetime.datetime(2017, 11, 27, hours, minutes, 0))
def _mocked_dav_client(*names, calendars=None):
@ -693,7 +693,9 @@ async def test_all_day_event_returned_early(
hass,
calendar,
config,
datetime.datetime(2017, 11, 27, 0, 30).replace(tzinfo=dt.DEFAULT_TIME_ZONE),
datetime.datetime(2017, 11, 27, 0, 30).replace(
tzinfo=dt_util.DEFAULT_TIME_ZONE
),
)
@ -711,7 +713,9 @@ async def test_all_day_event_returned_mid(
hass,
calendar,
config,
datetime.datetime(2017, 11, 27, 12, 30).replace(tzinfo=dt.DEFAULT_TIME_ZONE),
datetime.datetime(2017, 11, 27, 12, 30).replace(
tzinfo=dt_util.DEFAULT_TIME_ZONE
),
)
@ -729,7 +733,9 @@ async def test_all_day_event_returned_late(
hass,
calendar,
config,
datetime.datetime(2017, 11, 27, 23, 30).replace(tzinfo=dt.DEFAULT_TIME_ZONE),
datetime.datetime(2017, 11, 27, 23, 30).replace(
tzinfo=dt_util.DEFAULT_TIME_ZONE
),
)
@ -883,7 +889,7 @@ async def test_event_rrule_all_day_early(hass: HomeAssistant, calendar, set_tz)
hass,
calendar,
config,
datetime.datetime(2016, 12, 1, 0, 30).replace(tzinfo=dt.DEFAULT_TIME_ZONE),
datetime.datetime(2016, 12, 1, 0, 30).replace(tzinfo=dt_util.DEFAULT_TIME_ZONE),
)
@ -899,7 +905,9 @@ async def test_event_rrule_all_day_mid(hass: HomeAssistant, calendar, set_tz) ->
hass,
calendar,
config,
datetime.datetime(2016, 12, 1, 17, 30).replace(tzinfo=dt.DEFAULT_TIME_ZONE),
datetime.datetime(2016, 12, 1, 17, 30).replace(
tzinfo=dt_util.DEFAULT_TIME_ZONE
),
)
@ -915,14 +923,16 @@ async def test_event_rrule_all_day_late(hass: HomeAssistant, calendar, set_tz) -
hass,
calendar,
config,
datetime.datetime(2016, 12, 1, 23, 30).replace(tzinfo=dt.DEFAULT_TIME_ZONE),
datetime.datetime(2016, 12, 1, 23, 30).replace(
tzinfo=dt_util.DEFAULT_TIME_ZONE
),
)
@pytest.mark.parametrize("set_tz", ["utc"], indirect=True)
@patch(
"homeassistant.util.dt.now",
return_value=dt.as_local(datetime.datetime(2015, 11, 27, 0, 15)),
return_value=dt_util.as_local(datetime.datetime(2015, 11, 27, 0, 15)),
)
async def test_event_rrule_hourly_on_first(
mock_now, hass: HomeAssistant, calendar, set_tz
@ -949,7 +959,7 @@ async def test_event_rrule_hourly_on_first(
@pytest.mark.parametrize("set_tz", ["utc"], indirect=True)
@patch(
"homeassistant.util.dt.now",
return_value=dt.as_local(datetime.datetime(2015, 11, 27, 11, 15)),
return_value=dt_util.as_local(datetime.datetime(2015, 11, 27, 11, 15)),
)
async def test_event_rrule_hourly_on_last(
mock_now, hass: HomeAssistant, calendar, set_tz
@ -975,7 +985,7 @@ async def test_event_rrule_hourly_on_last(
@patch(
"homeassistant.util.dt.now",
return_value=dt.as_local(datetime.datetime(2015, 11, 27, 0, 45)),
return_value=dt_util.as_local(datetime.datetime(2015, 11, 27, 0, 45)),
)
async def test_event_rrule_hourly_off_first(
mock_now, hass: HomeAssistant, calendar
@ -991,7 +1001,7 @@ async def test_event_rrule_hourly_off_first(
@patch(
"homeassistant.util.dt.now",
return_value=dt.as_local(datetime.datetime(2015, 11, 27, 11, 45)),
return_value=dt_util.as_local(datetime.datetime(2015, 11, 27, 11, 45)),
)
async def test_event_rrule_hourly_off_last(
mock_now, hass: HomeAssistant, calendar
@ -1007,7 +1017,7 @@ async def test_event_rrule_hourly_off_last(
@patch(
"homeassistant.util.dt.now",
return_value=dt.as_local(datetime.datetime(2015, 11, 27, 12, 15)),
return_value=dt_util.as_local(datetime.datetime(2015, 11, 27, 12, 15)),
)
async def test_event_rrule_hourly_ended(
mock_now, hass: HomeAssistant, calendar

4
tests/components/cert_expiry/helpers.py
View File

@ -1,12 +1,12 @@
"""Helpers for Cert Expiry tests."""
from datetime import datetime, timedelta
from homeassistant.util import dt
from homeassistant.util import dt as dt_util
def static_datetime():
"""Build a datetime object for testing in the correct timezone."""
return dt.as_utc(datetime(2020, 6, 12, 8, 0, 0))
return dt_util.as_utc(datetime(2020, 6, 12, 8, 0, 0))
def future_timestamp(days):

6
tests/components/cloud/test_repairs.py
View File

@ -10,7 +10,7 @@ from homeassistant.components.repairs import DOMAIN as REPAIRS_DOMAIN
from homeassistant.core import HomeAssistant
import homeassistant.helpers.issue_registry as ir
from homeassistant.setup import async_setup_component
from homeassistant.util import dt
from homeassistant.util import dt as dt_util
from . import mock_cloud
@ -28,7 +28,7 @@ async def test_do_not_create_repair_issues_at_startup_if_not_logged_in(
with patch("homeassistant.components.cloud.Cloud.is_logged_in", False):
await mock_cloud(hass)
async_fire_time_changed(hass, dt.utcnow() + timedelta(hours=1))
async_fire_time_changed(hass, dt_util.utcnow() + timedelta(hours=1))
await hass.async_block_till_done()
assert not issue_registry.async_get_issue(
@ -51,7 +51,7 @@ async def test_create_repair_issues_at_startup_if_logged_in(
with patch("homeassistant.components.cloud.Cloud.is_logged_in", True):
await mock_cloud(hass)
async_fire_time_changed(hass, dt.utcnow() + timedelta(hours=1))
async_fire_time_changed(hass, dt_util.utcnow() + timedelta(hours=1))
await hass.async_block_till_done()
assert issue_registry.async_get_issue(

6
tests/components/command_line/test_sensor.py
View File

@ -13,7 +13,7 @@ from homeassistant.components.sensor import DOMAIN as SENSOR_DOMAIN
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_registry as er
import homeassistant.helpers.issue_registry as ir
from homeassistant.util import dt
from homeassistant.util import dt as dt_util
from tests.common import async_fire_time_changed
@ -123,7 +123,7 @@ async def test_template_render(
# Give time for template to load
async_fire_time_changed(
hass,
dt.utcnow() + timedelta(minutes=1),
dt_util.utcnow() + timedelta(minutes=1),
)
await hass.async_block_till_done()
@ -158,7 +158,7 @@ async def test_template_render_with_quote(hass: HomeAssistant) -> None:
# Give time for template to load
async_fire_time_changed(
hass,
dt.utcnow() + timedelta(minutes=1),
dt_util.utcnow() + timedelta(minutes=1),
)
await hass.async_block_till_done()

4
tests/components/deconz/test_sensor.py
View File

@ -24,7 +24,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 .test_gateway import DECONZ_WEB_REQUEST, setup_deconz_integration
@ -785,7 +785,7 @@ async def test_sensors(
async_fire_time_changed(
hass,
dt.utcnow() + timedelta(seconds=RELOAD_AFTER_UPDATE_DELAY + 1),
dt_util.utcnow() + timedelta(seconds=RELOAD_AFTER_UPDATE_DELAY + 1),
)
await hass.async_block_till_done()

4
tests/components/demo/test_vacuum.py
View File

@ -38,7 +38,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant
from homeassistant.setup import async_setup_component
from homeassistant.util import dt
from homeassistant.util import dt as dt_util
from tests.common import async_fire_time_changed, async_mock_service
from tests.components.vacuum import common
@ -179,7 +179,7 @@ async def test_methods(hass: HomeAssistant) -> None:
state = hass.states.get(ENTITY_VACUUM_STATE)
assert state.state == STATE_RETURNING
async_fire_time_changed(hass, dt.utcnow() + timedelta(seconds=31))
async_fire_time_changed(hass, dt_util.utcnow() + timedelta(seconds=31))
await hass.async_block_till_done()
state = hass.states.get(ENTITY_VACUUM_STATE)
assert state.state == STATE_DOCKED

6
tests/components/devolo_home_network/test_binary_sensor.py
View File

@ -18,7 +18,7 @@ from homeassistant.const import (
)
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 . import configure_integration
from .const import PLCNET_ATTACHED
@ -66,7 +66,7 @@ async def test_update_attached_to_router(
mock_device.plcnet.async_get_network_overview = AsyncMock(
side_effect=DeviceUnavailable
)
async_fire_time_changed(hass, dt.utcnow() + LONG_UPDATE_INTERVAL)
async_fire_time_changed(hass, dt_util.utcnow() + LONG_UPDATE_INTERVAL)
await hass.async_block_till_done()
state = hass.states.get(state_key)
@ -77,7 +77,7 @@ async def test_update_attached_to_router(
mock_device.plcnet.async_get_network_overview = AsyncMock(
return_value=PLCNET_ATTACHED
)
async_fire_time_changed(hass, dt.utcnow() + LONG_UPDATE_INTERVAL)
async_fire_time_changed(hass, dt_util.utcnow() + LONG_UPDATE_INTERVAL)
await hass.async_block_till_done()
state = hass.states.get(state_key)

10
tests/components/devolo_home_network/test_device_tracker.py
View File

@ -18,7 +18,7 @@ from homeassistant.const import (
)
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 . import configure_integration
from .const import CONNECTED_STATIONS, DISCOVERY_INFO, NO_CONNECTED_STATIONS
@ -40,13 +40,13 @@ async def test_device_tracker(
entry = configure_integration(hass)
await hass.config_entries.async_setup(entry.entry_id)
await hass.async_block_till_done()
async_fire_time_changed(hass, dt.utcnow() + LONG_UPDATE_INTERVAL)
async_fire_time_changed(hass, dt_util.utcnow() + LONG_UPDATE_INTERVAL)
await hass.async_block_till_done()
# Enable entity
entity_registry.async_update_entity(state_key, disabled_by=None)
await hass.async_block_till_done()
async_fire_time_changed(hass, dt.utcnow() + LONG_UPDATE_INTERVAL)
async_fire_time_changed(hass, dt_util.utcnow() + LONG_UPDATE_INTERVAL)
await hass.async_block_till_done()
state = hass.states.get(state_key)
@ -62,7 +62,7 @@ async def test_device_tracker(
mock_device.device.async_get_wifi_connected_station = AsyncMock(
return_value=NO_CONNECTED_STATIONS
)
async_fire_time_changed(hass, dt.utcnow() + LONG_UPDATE_INTERVAL)
async_fire_time_changed(hass, dt_util.utcnow() + LONG_UPDATE_INTERVAL)
await hass.async_block_till_done()
state = hass.states.get(state_key)
@ -73,7 +73,7 @@ async def test_device_tracker(
mock_device.device.async_get_wifi_connected_station = AsyncMock(
side_effect=DeviceUnavailable
)
async_fire_time_changed(hass, dt.utcnow() + LONG_UPDATE_INTERVAL)
async_fire_time_changed(hass, dt_util.utcnow() + LONG_UPDATE_INTERVAL)
await hass.async_block_till_done()
state = hass.states.get(state_key)

14
tests/components/devolo_home_network/test_sensor.py
View File

@ -12,7 +12,7 @@ from homeassistant.components.sensor import DOMAIN, SensorStateClass
from homeassistant.const import ATTR_FRIENDLY_NAME, STATE_UNAVAILABLE, EntityCategory
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 . import configure_integration
from .mock import MockDevice
@ -57,7 +57,7 @@ async def test_update_connected_wifi_clients(
mock_device.device.async_get_wifi_connected_station = AsyncMock(
side_effect=DeviceUnavailable
)
async_fire_time_changed(hass, dt.utcnow() + SHORT_UPDATE_INTERVAL)
async_fire_time_changed(hass, dt_util.utcnow() + SHORT_UPDATE_INTERVAL)
await hass.async_block_till_done()
state = hass.states.get(state_key)
@ -66,7 +66,7 @@ async def test_update_connected_wifi_clients(
# Emulate state change
mock_device.reset()
async_fire_time_changed(hass, dt.utcnow() + SHORT_UPDATE_INTERVAL)
async_fire_time_changed(hass, dt_util.utcnow() + SHORT_UPDATE_INTERVAL)
await hass.async_block_till_done()
state = hass.states.get(state_key)
@ -103,7 +103,7 @@ async def test_update_neighboring_wifi_networks(
mock_device.device.async_get_wifi_neighbor_access_points = AsyncMock(
side_effect=DeviceUnavailable
)
async_fire_time_changed(hass, dt.utcnow() + LONG_UPDATE_INTERVAL)
async_fire_time_changed(hass, dt_util.utcnow() + LONG_UPDATE_INTERVAL)
await hass.async_block_till_done()
state = hass.states.get(state_key)
@ -112,7 +112,7 @@ async def test_update_neighboring_wifi_networks(
# Emulate state change
mock_device.reset()
async_fire_time_changed(hass, dt.utcnow() + LONG_UPDATE_INTERVAL)
async_fire_time_changed(hass, dt_util.utcnow() + LONG_UPDATE_INTERVAL)
await hass.async_block_till_done()
state = hass.states.get(state_key)
@ -148,7 +148,7 @@ async def test_update_connected_plc_devices(
mock_device.plcnet.async_get_network_overview = AsyncMock(
side_effect=DeviceUnavailable
)
async_fire_time_changed(hass, dt.utcnow() + LONG_UPDATE_INTERVAL)
async_fire_time_changed(hass, dt_util.utcnow() + LONG_UPDATE_INTERVAL)
await hass.async_block_till_done()
state = hass.states.get(state_key)
@ -157,7 +157,7 @@ async def test_update_connected_plc_devices(
# Emulate state change
mock_device.reset()
async_fire_time_changed(hass, dt.utcnow() + LONG_UPDATE_INTERVAL)
async_fire_time_changed(hass, dt_util.utcnow() + LONG_UPDATE_INTERVAL)
await hass.async_block_till_done()
state = hass.states.get(state_key)

16
tests/components/devolo_home_network/test_switch.py
View File

@ -24,7 +24,7 @@ from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers import entity_registry as er
from homeassistant.helpers.update_coordinator import REQUEST_REFRESH_DEFAULT_COOLDOWN
from homeassistant.util import dt
from homeassistant.util import dt as dt_util
from . import configure_integration
from .mock import MockDevice
@ -90,7 +90,7 @@ async def test_update_enable_guest_wifi(
mock_device.device.async_get_wifi_guest_access.return_value = WifiGuestAccessGet(
enabled=True
)
async_fire_time_changed(hass, dt.utcnow() + SHORT_UPDATE_INTERVAL)
async_fire_time_changed(hass, dt_util.utcnow() + SHORT_UPDATE_INTERVAL)
await hass.async_block_till_done()
state = hass.states.get(state_key)
@ -115,7 +115,7 @@ async def test_update_enable_guest_wifi(
turn_off.assert_called_once_with(False)
async_fire_time_changed(
hass, dt.utcnow() + timedelta(seconds=REQUEST_REFRESH_DEFAULT_COOLDOWN)
hass, dt_util.utcnow() + timedelta(seconds=REQUEST_REFRESH_DEFAULT_COOLDOWN)
)
await hass.async_block_till_done()
@ -137,7 +137,7 @@ async def test_update_enable_guest_wifi(
turn_on.assert_called_once_with(True)
async_fire_time_changed(
hass, dt.utcnow() + timedelta(seconds=REQUEST_REFRESH_DEFAULT_COOLDOWN)
hass, dt_util.utcnow() + timedelta(seconds=REQUEST_REFRESH_DEFAULT_COOLDOWN)
)
await hass.async_block_till_done()
@ -177,7 +177,7 @@ async def test_update_enable_leds(
# Emulate state change
mock_device.device.async_get_led_setting.return_value = True
async_fire_time_changed(hass, dt.utcnow() + SHORT_UPDATE_INTERVAL)
async_fire_time_changed(hass, dt_util.utcnow() + SHORT_UPDATE_INTERVAL)
await hass.async_block_till_done()
state = hass.states.get(state_key)
@ -200,7 +200,7 @@ async def test_update_enable_leds(
turn_off.assert_called_once_with(False)
async_fire_time_changed(
hass, dt.utcnow() + timedelta(seconds=REQUEST_REFRESH_DEFAULT_COOLDOWN)
hass, dt_util.utcnow() + timedelta(seconds=REQUEST_REFRESH_DEFAULT_COOLDOWN)
)
await hass.async_block_till_done()
@ -220,7 +220,7 @@ async def test_update_enable_leds(
turn_on.assert_called_once_with(True)
async_fire_time_changed(
hass, dt.utcnow() + timedelta(seconds=REQUEST_REFRESH_DEFAULT_COOLDOWN)
hass, dt_util.utcnow() + timedelta(seconds=REQUEST_REFRESH_DEFAULT_COOLDOWN)
)
await hass.async_block_till_done()
@ -268,7 +268,7 @@ async def test_device_failure(
api = getattr(mock_device.device, get_method)
api.side_effect = DeviceUnavailable
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(state_key)

4
tests/components/dnsip/test_sensor.py
View File

@ -17,7 +17,7 @@ from homeassistant.components.dnsip.const import (
from homeassistant.config_entries import SOURCE_USER
from homeassistant.const import CONF_NAME, STATE_UNAVAILABLE
from homeassistant.core import HomeAssistant
from homeassistant.util import dt
from homeassistant.util import dt as dt_util
from . import RetrieveDNS
@ -97,7 +97,7 @@ async def test_sensor_no_response(hass: HomeAssistant) -> None:
):
async_fire_time_changed(
hass,
dt.utcnow() + timedelta(minutes=10),
dt_util.utcnow() + timedelta(minutes=10),
)
await hass.async_block_till_done()