evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

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

Browse source
This commit is contained in:
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
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.core import callback
from homeassistant.helpers import aiohttp_client from homeassistant.helpers import aiohttp_client
from homeassistant.helpers.storage import Store from homeassistant.helpers.storage import Store
from homeassistant.util import dt from homeassistant.util import dt as dt_util
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)
@ -95,12 +95,12 @@ class Auth:
if not self._prefs[STORAGE_ACCESS_TOKEN]: if not self._prefs[STORAGE_ACCESS_TOKEN]:
return False 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( preemptive_expire_time = expire_time - timedelta(
seconds=PREEMPTIVE_REFRESH_TTL_IN_SECONDS 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): async def _async_request_new_token(self, lwa_params):
try: try:
@ -130,7 +130,7 @@ class Auth:
access_token = response_json["access_token"] access_token = response_json["access_token"]
refresh_token = response_json["refresh_token"] refresh_token = response_json["refresh_token"]
expires_in = response_json["expires_in"] 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( await self._async_update_preferences(
access_token, refresh_token, expire_time.isoformat() 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.exceptions import HomeAssistantError
from homeassistant.helpers import integration_platform from homeassistant.helpers import integration_platform
from homeassistant.helpers.json import save_json 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 homeassistant.util.json import json_loads_object
from .const import DOMAIN, EXCLUDE_FROM_BACKUP, LOGGER from .const import DOMAIN, EXCLUDE_FROM_BACKUP, LOGGER
@ -176,7 +176,7 @@ class BackupManager:
raise result raise result
backup_name = f"Core {HAVERSION}" backup_name = f"Core {HAVERSION}"
date_str = dt.now().isoformat() date_str = dt_util.now().isoformat()
slug = _generate_slug(date_str, backup_name) slug = _generate_slug(date_str, backup_name)
backup_data = { 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.entity_platform import AddEntitiesCallback
from homeassistant.helpers.restore_state import RestoreEntity from homeassistant.helpers.restore_state import RestoreEntity
from homeassistant.helpers.storage import Store from homeassistant.helpers.storage import Store
from homeassistant.util import dt from homeassistant.util import dt as dt_util
from .const import DOMAIN from .const import DOMAIN
from .entity import BroadlinkEntity from .entity import BroadlinkEntity
@ -330,8 +330,8 @@ class BroadlinkRemote(BroadlinkEntity, RemoteEntity, RestoreEntity):
) )
try: try:
start_time = dt.utcnow() start_time = dt_util.utcnow()
while (dt.utcnow() - start_time) < LEARNING_TIMEOUT: while (dt_util.utcnow() - start_time) < LEARNING_TIMEOUT:
await asyncio.sleep(1) await asyncio.sleep(1)
try: try:
code = await device.async_request(device.api.check_data) code = await device.async_request(device.api.check_data)
@ -368,8 +368,8 @@ class BroadlinkRemote(BroadlinkEntity, RemoteEntity, RestoreEntity):
) )
try: try:
start_time = dt.utcnow() start_time = dt_util.utcnow()
while (dt.utcnow() - start_time) < LEARNING_TIMEOUT: while (dt_util.utcnow() - start_time) < LEARNING_TIMEOUT:
await asyncio.sleep(1) await asyncio.sleep(1)
found = await device.async_request(device.api.check_frequency) found = await device.async_request(device.api.check_frequency)
if found: if found:
@ -403,8 +403,8 @@ class BroadlinkRemote(BroadlinkEntity, RemoteEntity, RestoreEntity):
) )
try: try:
start_time = dt.utcnow() start_time = dt_util.utcnow()
while (dt.utcnow() - start_time) < LEARNING_TIMEOUT: while (dt_util.utcnow() - start_time) < LEARNING_TIMEOUT:
await asyncio.sleep(1) await asyncio.sleep(1)
try: try:
code = await device.async_request(device.api.check_data) 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 broadlink.exceptions import AuthorizationError, BroadlinkException
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed 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__) _LOGGER = logging.getLogger(__name__)
@ -64,7 +64,7 @@ class BroadlinkUpdateManager(ABC):
except (BroadlinkException, OSError) as err: except (BroadlinkException, OSError) as err:
if self.available and ( 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)) or isinstance(err, (AuthorizationError, OSError))
): ):
self.available = False self.available = False
@ -84,7 +84,7 @@ class BroadlinkUpdateManager(ABC):
self.device.api.host[0], self.device.api.host[0],
) )
self.available = True self.available = True
self.last_update = dt.utcnow() self.last_update = dt_util.utcnow()
return data return data
@abstractmethod @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 import generate_entity_id
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType 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__) _LOGGER = logging.getLogger(__name__)
@ -204,8 +204,8 @@ class WebDavCalendarData:
@Throttle(MIN_TIME_BETWEEN_UPDATES) @Throttle(MIN_TIME_BETWEEN_UPDATES)
def update(self): def update(self):
"""Get the latest data.""" """Get the latest data."""
start_of_today = dt.start_of_local_day() start_of_today = dt_util.start_of_local_day()
start_of_tomorrow = dt.start_of_local_day() + timedelta(days=self.days) 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 # We have to retrieve the results for the whole day as the server
# won't return events that have already started # won't return events that have already started
@ -312,7 +312,7 @@ class WebDavCalendarData:
@staticmethod @staticmethod
def is_over(vevent): def is_over(vevent):
"""Return if the event is over.""" """Return if the event is over."""
return dt.now() >= WebDavCalendarData.to_datetime( return dt_util.now() >= WebDavCalendarData.to_datetime(
WebDavCalendarData.get_end_date(vevent) WebDavCalendarData.get_end_date(vevent)
) )
@ -321,7 +321,7 @@ class WebDavCalendarData:
"""Return a datetime.""" """Return a datetime."""
if isinstance(obj, datetime): if isinstance(obj, datetime):
return WebDavCalendarData.to_local(obj) 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 @staticmethod
def to_local(obj: datetime | date) -> datetime | date: 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. used by the caldav client and dateutil so the datetime can be copied.
""" """
if isinstance(obj, datetime): if isinstance(obj, datetime):
return dt.as_local(obj) return dt_util.as_local(obj)
return obj return obj
@staticmethod @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.entity_component import EntityComponent
from homeassistant.helpers.template import DATE_STR_FORMAT from homeassistant.helpers.template import DATE_STR_FORMAT
from homeassistant.helpers.typing import ConfigType from homeassistant.helpers.typing import ConfigType
from homeassistant.util import dt from homeassistant.util import dt as dt_util
from .const import ( from .const import (
CONF_EVENT, 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.""" """Convert all keys that are datetime values to local timezone."""
for k in keys: for k in keys:
if (value := obj.get(k)) and isinstance(value, datetime.datetime): 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 obj
return validate 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: def get_date(date: dict[str, Any]) -> datetime.datetime:
"""Get the dateTime from date or dateTime as a local.""" """Get the dateTime from date or dateTime as a local."""
if "date" in date: if "date" in date:
parsed_date = dt.parse_date(date["date"]) parsed_date = dt_util.parse_date(date["date"])
assert parsed_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) 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 assert parsed_datetime
return dt.as_local(parsed_datetime) return dt_util.as_local(parsed_datetime)
@dataclasses.dataclass @dataclasses.dataclass
@ -380,7 +380,7 @@ def _api_event_dict_factory(obj: Iterable[tuple[str, Any]]) -> dict[str, Any]:
result: dict[str, Any] = {} result: dict[str, Any] = {}
for name, value in obj: for name, value in obj:
if isinstance(value, datetime.datetime): 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): elif isinstance(value, datetime.date):
result[name] = {"date": value.isoformat()} result[name] = {"date": value.isoformat()}
else: else:
@ -393,14 +393,14 @@ def _get_datetime_local(
) -> datetime.datetime: ) -> datetime.datetime:
"""Convert a calendar event date/datetime to a datetime if needed.""" """Convert a calendar event date/datetime to a datetime if needed."""
if isinstance(dt_or_d, datetime.datetime): if isinstance(dt_or_d, datetime.datetime):
return dt.as_local(dt_or_d) return dt_util.as_local(dt_or_d)
return dt.start_of_local_day(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]: 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.""" """Convert a calendar event date/datetime to a datetime if needed."""
if isinstance(dt_or_d, datetime.datetime): 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()} 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.""" """Have we reached the offset time specified in the event title."""
if offset_time == datetime.timedelta(): if offset_time == datetime.timedelta():
return False return False
return start + offset_time <= dt.now(start.tzinfo) return start + offset_time <= dt_util.now(start.tzinfo)
class CalendarEntity(Entity): class CalendarEntity(Entity):
@ -467,7 +467,7 @@ class CalendarEntity(Entity):
if (event := self.event) is None: if (event := self.event) is None:
return STATE_OFF return STATE_OFF
now = dt.now() now = dt_util.now()
if event.start_datetime_local <= now < event.end_datetime_local: if event.start_datetime_local <= now < event.end_datetime_local:
return STATE_ON return STATE_ON
@ -529,8 +529,8 @@ class CalendarEventView(http.HomeAssistantView):
if start is None or end is None: if start is None or end is None:
return web.Response(status=HTTPStatus.BAD_REQUEST) return web.Response(status=HTTPStatus.BAD_REQUEST)
try: try:
start_date = dt.parse_datetime(start) start_date = dt_util.parse_datetime(start)
end_date = dt.parse_datetime(end) end_date = dt_util.parse_datetime(end)
except (ValueError, AttributeError): except (ValueError, AttributeError):
return web.Response(status=HTTPStatus.BAD_REQUEST) return web.Response(status=HTTPStatus.BAD_REQUEST)
if start_date is None or end_date is None: if start_date is None or end_date is None:
@ -540,7 +540,9 @@ class CalendarEventView(http.HomeAssistantView):
try: try:
calendar_event_list = await entity.async_get_events( 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: except HomeAssistantError as err:
_LOGGER.debug("Error reading events: %s", 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 import ssl
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.util import dt from homeassistant.util import dt as dt_util
from .const import TIMEOUT from .const import TIMEOUT
from .errors import ( from .errors import (
@ -52,4 +52,4 @@ async def get_cert_expiry_timestamp(
raise ValidationFailure(err.args[0]) from err raise ValidationFailure(err.args[0]) from err
ts_seconds = ssl.cert_time_to_seconds(cert["notAfter"]) 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.components.mailbox import CONTENT_TYPE_MPEG, Mailbox, StreamError
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from homeassistant.util import dt from homeassistant.util import dt as dt_util
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)
@ -34,7 +34,9 @@ class DemoMailbox(Mailbox):
self._messages: dict[str, dict[str, Any]] = {} self._messages: dict[str, dict[str, Any]] = {}
txt = "Lorem ipsum dolor sit amet, consectetur adipiscing elit. " txt = "Lorem ipsum dolor sit amet, consectetur adipiscing elit. "
for idx in range(0, 10): 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))}" msgtxt = f"Message {idx + 1}. {txt * (1 + idx * (idx % 2))}"
msgsha = sha1(msgtxt.encode("utf-8")).hexdigest() msgsha = sha1(msgtxt.encode("utf-8")).hexdigest()
msg = { 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.const import STATE_OFF, STATE_ON
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_registry as er from homeassistant.helpers import entity_registry as er
from homeassistant.util import dt from homeassistant.util import dt as dt_util
from . import ( from . import (
TEST_SET_RESPONSE, TEST_SET_RESPONSE,
@ -88,7 +88,7 @@ async def test_binary_sensor_async_setup_entry(
async_fire_time_changed( async_fire_time_changed(
hass, 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() await hass.async_block_till_done()
@ -110,7 +110,7 @@ async def test_binary_sensor_async_setup_entry(
async_fire_time_changed( async_fire_time_changed(
hass, 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() 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.const import ATTR_ENTITY_ID
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_registry as er from homeassistant.helpers import entity_registry as er
from homeassistant.util import dt from homeassistant.util import dt as dt_util
from . import ( from . import (
TEST_SET_RESPONSE, TEST_SET_RESPONSE,
@ -144,7 +144,7 @@ async def test_sensor_platform(
async_fire_time_changed( async_fire_time_changed(
hass, 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() 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.const import STATE_UNAVAILABLE
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.util import dt from homeassistant.util import dt as dt_util
from . import async_init_integration 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", "homeassistant.components.anova.AnovaPrecisionCooker.update",
side_effect=AnovaOffline(), 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() await hass.async_block_till_done()
state = hass.states.get("sensor.anova_precision_cooker_water_temperature") 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.components.broadlink.heartbeat import BroadlinkHeartbeat
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.util import dt from homeassistant.util import dt as dt_util
from . import get_device 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: with patch(DEVICE_PING) as mock_ping:
async_fire_time_changed( async_fire_time_changed(
hass, dt.utcnow() + BroadlinkHeartbeat.HEARTBEAT_INTERVAL hass, dt_util.utcnow() + BroadlinkHeartbeat.HEARTBEAT_INTERVAL
) )
await hass.async_block_till_done() 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: with patch(DEVICE_PING) as mock_ping:
async_fire_time_changed( async_fire_time_changed(
hass, hass,
dt.utcnow() + BroadlinkHeartbeat.HEARTBEAT_INTERVAL // 2, dt_util.utcnow() + BroadlinkHeartbeat.HEARTBEAT_INTERVAL // 2,
) )
await hass.async_block_till_done() 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: with patch(DEVICE_PING) as mock_ping:
async_fire_time_changed( async_fire_time_changed(
hass, dt.utcnow() + BroadlinkHeartbeat.HEARTBEAT_INTERVAL hass, dt_util.utcnow() + BroadlinkHeartbeat.HEARTBEAT_INTERVAL
) )
assert mock_ping.call_count == 0 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: with patch(DEVICE_PING) as mock_ping:
async_fire_time_changed( async_fire_time_changed(
hass, dt.utcnow() + BroadlinkHeartbeat.HEARTBEAT_INTERVAL hass, dt_util.utcnow() + BroadlinkHeartbeat.HEARTBEAT_INTERVAL
) )
await hass.async_block_till_done() 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.core import HomeAssistant
from homeassistant.helpers import device_registry as dr, entity_registry as er from homeassistant.helpers import device_registry as dr, entity_registry as er
from homeassistant.helpers.entity_component import async_update_entity 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 from . import get_device
@ -384,7 +384,9 @@ async def test_scb1e_sensor_update(
} }
target_time = ( 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) 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.const import STATE_OFF, STATE_ON
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.setup import async_setup_component 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"} DEVICE_DATA = {"name": "Private Calendar", "device_id": "Private Calendar"}
@ -360,7 +360,7 @@ def get_api_events(hass_client):
def _local_datetime(hours, minutes): def _local_datetime(hours, minutes):
"""Build a datetime object for testing in the correct timezone.""" """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): def _mocked_dav_client(*names, calendars=None):
@ -693,7 +693,9 @@ async def test_all_day_event_returned_early(
hass, hass,
calendar, calendar,
config, 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, hass,
calendar, calendar,
config, 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, hass,
calendar, calendar,
config, 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, hass,
calendar, calendar,
config, 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, hass,
calendar, calendar,
config, 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, hass,
calendar, calendar,
config, 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) @pytest.mark.parametrize("set_tz", ["utc"], indirect=True)
@patch( @patch(
"homeassistant.util.dt.now", "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( async def test_event_rrule_hourly_on_first(
mock_now, hass: HomeAssistant, calendar, set_tz 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) @pytest.mark.parametrize("set_tz", ["utc"], indirect=True)
@patch( @patch(
"homeassistant.util.dt.now", "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( async def test_event_rrule_hourly_on_last(
mock_now, hass: HomeAssistant, calendar, set_tz mock_now, hass: HomeAssistant, calendar, set_tz
@ -975,7 +985,7 @@ async def test_event_rrule_hourly_on_last(
@patch( @patch(
"homeassistant.util.dt.now", "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( async def test_event_rrule_hourly_off_first(
mock_now, hass: HomeAssistant, calendar mock_now, hass: HomeAssistant, calendar
@ -991,7 +1001,7 @@ async def test_event_rrule_hourly_off_first(
@patch( @patch(
"homeassistant.util.dt.now", "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( async def test_event_rrule_hourly_off_last(
mock_now, hass: HomeAssistant, calendar mock_now, hass: HomeAssistant, calendar
@ -1007,7 +1017,7 @@ async def test_event_rrule_hourly_off_last(
@patch( @patch(
"homeassistant.util.dt.now", "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( async def test_event_rrule_hourly_ended(
mock_now, hass: HomeAssistant, calendar mock_now, hass: HomeAssistant, calendar

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

@ -1,12 +1,12 @@
"""Helpers for Cert Expiry tests.""" """Helpers for Cert Expiry tests."""
from datetime import datetime, timedelta from datetime import datetime, timedelta
from homeassistant.util import dt from homeassistant.util import dt as dt_util
def static_datetime(): def static_datetime():
"""Build a datetime object for testing in the correct timezone.""" """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): 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 from homeassistant.core import HomeAssistant
import homeassistant.helpers.issue_registry as ir import homeassistant.helpers.issue_registry as ir
from homeassistant.setup import async_setup_component from homeassistant.setup import async_setup_component
from homeassistant.util import dt from homeassistant.util import dt as dt_util
from . import mock_cloud 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): with patch("homeassistant.components.cloud.Cloud.is_logged_in", False):
await mock_cloud(hass) 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() await hass.async_block_till_done()
assert not issue_registry.async_get_issue( 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): with patch("homeassistant.components.cloud.Cloud.is_logged_in", True):
await mock_cloud(hass) 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() await hass.async_block_till_done()
assert issue_registry.async_get_issue( 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.core import HomeAssistant
from homeassistant.helpers import entity_registry as er from homeassistant.helpers import entity_registry as er
import homeassistant.helpers.issue_registry as ir 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 from tests.common import async_fire_time_changed
@ -123,7 +123,7 @@ async def test_template_render(
# Give time for template to load # Give time for template to load
async_fire_time_changed( async_fire_time_changed(
hass, hass,
dt.utcnow() + timedelta(minutes=1), dt_util.utcnow() + timedelta(minutes=1),
) )
await hass.async_block_till_done() 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 # Give time for template to load
async_fire_time_changed( async_fire_time_changed(
hass, hass,
dt.utcnow() + timedelta(minutes=1), dt_util.utcnow() + timedelta(minutes=1),
) )
await hass.async_block_till_done() 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.core import HomeAssistant
from homeassistant.helpers import device_registry as dr, entity_registry as er 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 from .test_gateway import DECONZ_WEB_REQUEST, setup_deconz_integration
@ -785,7 +785,7 @@ async def test_sensors(
async_fire_time_changed( async_fire_time_changed(
hass, 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() 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.core import HomeAssistant
from homeassistant.setup import async_setup_component 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.common import async_fire_time_changed, async_mock_service
from tests.components.vacuum import common from tests.components.vacuum import common
@ -179,7 +179,7 @@ async def test_methods(hass: HomeAssistant) -> None:
state = hass.states.get(ENTITY_VACUUM_STATE) state = hass.states.get(ENTITY_VACUUM_STATE)
assert state.state == STATE_RETURNING 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() await hass.async_block_till_done()
state = hass.states.get(ENTITY_VACUUM_STATE) state = hass.states.get(ENTITY_VACUUM_STATE)
assert state.state == STATE_DOCKED 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.core import HomeAssistant
from homeassistant.helpers import entity_registry as er 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 . import configure_integration
from .const import PLCNET_ATTACHED from .const import PLCNET_ATTACHED
@ -66,7 +66,7 @@ async def test_update_attached_to_router(
mock_device.plcnet.async_get_network_overview = AsyncMock( mock_device.plcnet.async_get_network_overview = AsyncMock(
side_effect=DeviceUnavailable 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() await hass.async_block_till_done()
state = hass.states.get(state_key) 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( mock_device.plcnet.async_get_network_overview = AsyncMock(
return_value=PLCNET_ATTACHED 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() await hass.async_block_till_done()
state = hass.states.get(state_key) 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.core import HomeAssistant
from homeassistant.helpers import entity_registry as er 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 . import configure_integration
from .const import CONNECTED_STATIONS, DISCOVERY_INFO, NO_CONNECTED_STATIONS from .const import CONNECTED_STATIONS, DISCOVERY_INFO, NO_CONNECTED_STATIONS
@ -40,13 +40,13 @@ async def test_device_tracker(
entry = configure_integration(hass) entry = configure_integration(hass)
await hass.config_entries.async_setup(entry.entry_id) await hass.config_entries.async_setup(entry.entry_id)
await hass.async_block_till_done() 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() await hass.async_block_till_done()
# Enable entity # Enable entity
entity_registry.async_update_entity(state_key, disabled_by=None) entity_registry.async_update_entity(state_key, disabled_by=None)
await hass.async_block_till_done() 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() await hass.async_block_till_done()
state = hass.states.get(state_key) state = hass.states.get(state_key)
@ -62,7 +62,7 @@ async def test_device_tracker(
mock_device.device.async_get_wifi_connected_station = AsyncMock( mock_device.device.async_get_wifi_connected_station = AsyncMock(
return_value=NO_CONNECTED_STATIONS 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() await hass.async_block_till_done()
state = hass.states.get(state_key) state = hass.states.get(state_key)
@ -73,7 +73,7 @@ async def test_device_tracker(
mock_device.device.async_get_wifi_connected_station = AsyncMock( mock_device.device.async_get_wifi_connected_station = AsyncMock(
side_effect=DeviceUnavailable 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() await hass.async_block_till_done()
state = hass.states.get(state_key) 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.const import ATTR_FRIENDLY_NAME, STATE_UNAVAILABLE, EntityCategory
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_registry as er 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 . import configure_integration
from .mock import MockDevice from .mock import MockDevice
@ -57,7 +57,7 @@ async def test_update_connected_wifi_clients(
mock_device.device.async_get_wifi_connected_station = AsyncMock( mock_device.device.async_get_wifi_connected_station = AsyncMock(
side_effect=DeviceUnavailable 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() await hass.async_block_till_done()
state = hass.states.get(state_key) state = hass.states.get(state_key)
@ -66,7 +66,7 @@ async def test_update_connected_wifi_clients(
# Emulate state change # Emulate state change
mock_device.reset() 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() await hass.async_block_till_done()
state = hass.states.get(state_key) 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( mock_device.device.async_get_wifi_neighbor_access_points = AsyncMock(
side_effect=DeviceUnavailable 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() await hass.async_block_till_done()
state = hass.states.get(state_key) state = hass.states.get(state_key)
@ -112,7 +112,7 @@ async def test_update_neighboring_wifi_networks(
# Emulate state change # Emulate state change
mock_device.reset() 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() await hass.async_block_till_done()
state = hass.states.get(state_key) 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( mock_device.plcnet.async_get_network_overview = AsyncMock(
side_effect=DeviceUnavailable 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() await hass.async_block_till_done()
state = hass.states.get(state_key) state = hass.states.get(state_key)
@ -157,7 +157,7 @@ async def test_update_connected_plc_devices(
# Emulate state change # Emulate state change
mock_device.reset() 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() await hass.async_block_till_done()
state = hass.states.get(state_key) 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.exceptions import HomeAssistantError
from homeassistant.helpers import entity_registry as er from homeassistant.helpers import entity_registry as er
from homeassistant.helpers.update_coordinator import REQUEST_REFRESH_DEFAULT_COOLDOWN 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 . import configure_integration
from .mock import MockDevice 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( mock_device.device.async_get_wifi_guest_access.return_value = WifiGuestAccessGet(
enabled=True 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() await hass.async_block_till_done()
state = hass.states.get(state_key) state = hass.states.get(state_key)
@ -115,7 +115,7 @@ async def test_update_enable_guest_wifi(
turn_off.assert_called_once_with(False) turn_off.assert_called_once_with(False)
async_fire_time_changed( 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() await hass.async_block_till_done()
@ -137,7 +137,7 @@ async def test_update_enable_guest_wifi(
turn_on.assert_called_once_with(True) turn_on.assert_called_once_with(True)
async_fire_time_changed( 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() await hass.async_block_till_done()
@ -177,7 +177,7 @@ async def test_update_enable_leds(
# Emulate state change # Emulate state change
mock_device.device.async_get_led_setting.return_value = True 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() await hass.async_block_till_done()
state = hass.states.get(state_key) state = hass.states.get(state_key)
@ -200,7 +200,7 @@ async def test_update_enable_leds(
turn_off.assert_called_once_with(False) turn_off.assert_called_once_with(False)
async_fire_time_changed( 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() await hass.async_block_till_done()
@ -220,7 +220,7 @@ async def test_update_enable_leds(
turn_on.assert_called_once_with(True) turn_on.assert_called_once_with(True)
async_fire_time_changed( 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() await hass.async_block_till_done()
@ -268,7 +268,7 @@ async def test_device_failure(
api = getattr(mock_device.device, get_method) api = getattr(mock_device.device, get_method)
api.side_effect = DeviceUnavailable 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() await hass.async_block_till_done()
state = hass.states.get(state_key) 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.config_entries import SOURCE_USER
from homeassistant.const import CONF_NAME, STATE_UNAVAILABLE from homeassistant.const import CONF_NAME, STATE_UNAVAILABLE
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.util import dt from homeassistant.util import dt as dt_util
from . import RetrieveDNS from . import RetrieveDNS
@ -97,7 +97,7 @@ async def test_sensor_no_response(hass: HomeAssistant) -> None:
): ):
async_fire_time_changed( async_fire_time_changed(
hass, hass,
dt.utcnow() + timedelta(minutes=10), dt_util.utcnow() + timedelta(minutes=10),
) )
await hass.async_block_till_done() await hass.async_block_till_done()