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

Add missing hass type hint in flux tests (#124078)

Browse source
This commit is contained in:
epenet 2024-08-18 13:53:59 +02:00 • committed by GitHub
parent ec0012209d
commit c4fee124b3
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 49 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

65
tests/components/flux/test_switch.py
View file

@ -1,5 +1,6 @@
"""The tests for the Flux switch platform."""
from datetime import date, datetime
from unittest.mock import patch
from freezegun import freeze_time
@ -187,7 +188,9 @@ async def test_flux_when_switch_is_off(
sunset_time = test_time.replace(hour=17, minute=0, second=0)
sunrise_time = test_time.replace(hour=5, minute=0, second=0)
def event_date(hass, event, now=None):
def event_date(
hass: HomeAssistant, event: str, now: date | datetime | None = None
) -> datetime | None:
if event == SUN_EVENT_SUNRISE:
return sunrise_time
return sunset_time
@ -242,7 +245,9 @@ async def test_flux_before_sunrise(
sunset_time = test_time.replace(hour=17, minute=0, second=0)
sunrise_time = test_time.replace(hour=5, minute=0, second=5)
def event_date(hass, event, now=None):
def event_date(
hass: HomeAssistant, event: str, now: date | datetime | None = None
) -> datetime | None:
if event == SUN_EVENT_SUNRISE:
return sunrise_time
return sunset_time
@ -364,7 +369,9 @@ async def test_flux_after_sunrise_before_sunset(
sunset_time = test_time.replace(hour=17, minute=0, second=0)
sunrise_time = test_time.replace(hour=5, minute=0, second=0)
def event_date(hass, event, now=None):
def event_date(
hass: HomeAssistant, event: str, now: date | datetime | None = None
) -> datetime | None:
if event == SUN_EVENT_SUNRISE:
return sunrise_time
return sunset_time
@ -426,7 +433,9 @@ async def test_flux_after_sunset_before_stop(
sunset_time = test_time.replace(hour=17, minute=0, second=0)
sunrise_time = test_time.replace(hour=5, minute=0, second=0)
def event_date(hass, event, now=None):
def event_date(
hass: HomeAssistant, event: str, now: date | datetime | None = None
) -> datetime | None:
if event == SUN_EVENT_SUNRISE:
return sunrise_time
return sunset_time
@ -489,7 +498,9 @@ async def test_flux_after_stop_before_sunrise(
sunset_time = test_time.replace(hour=17, minute=0, second=0)
sunrise_time = test_time.replace(hour=5, minute=0, second=0)
def event_date(hass, event, now=None):
def event_date(
hass: HomeAssistant, event: str, now: date | datetime | None = None
) -> datetime | None:
if event == SUN_EVENT_SUNRISE:
return sunrise_time
return sunset_time
@ -551,7 +562,9 @@ async def test_flux_with_custom_start_stop_times(
sunset_time = test_time.replace(hour=17, minute=0, second=0)
sunrise_time = test_time.replace(hour=5, minute=0, second=0)
def event_date(hass, event, now=None):
def event_date(
hass: HomeAssistant, event: str, now: date | datetime | None = None
) -> datetime | None:
if event == SUN_EVENT_SUNRISE:
return sunrise_time
return sunset_time
@ -618,7 +631,9 @@ async def test_flux_before_sunrise_stop_next_day(
sunset_time = test_time.replace(hour=17, minute=0, second=0)
sunrise_time = test_time.replace(hour=5, minute=0, second=0)
def event_date(hass, event, now=None):
def event_date(
hass: HomeAssistant, event: str, now: date | datetime | None = None
) -> datetime | None:
if event == SUN_EVENT_SUNRISE:
return sunrise_time
return sunset_time
@ -684,7 +699,9 @@ async def test_flux_after_sunrise_before_sunset_stop_next_day(
sunset_time = test_time.replace(hour=17, minute=0, second=0)
sunrise_time = test_time.replace(hour=5, minute=0, second=0)
def event_date(hass, event, now=None):
def event_date(
hass: HomeAssistant, event: str, now: date | datetime | None = None
) -> datetime | None:
if event == SUN_EVENT_SUNRISE:
return sunrise_time
return sunset_time
@ -750,7 +767,9 @@ async def test_flux_after_sunset_before_midnight_stop_next_day(
sunset_time = test_time.replace(hour=17, minute=0, second=0)
sunrise_time = test_time.replace(hour=5, minute=0, second=0)
def event_date(hass, event, now=None):
def event_date(
hass: HomeAssistant, event: str, now: date | datetime | None = None
) -> datetime | None:
if event == SUN_EVENT_SUNRISE:
return sunrise_time
return sunset_time
@ -816,7 +835,9 @@ async def test_flux_after_sunset_after_midnight_stop_next_day(
sunset_time = test_time.replace(hour=17, minute=0, second=0)
sunrise_time = test_time.replace(hour=5, minute=0, second=0)
def event_date(hass, event, now=None):
def event_date(
hass: HomeAssistant, event: str, now: date | datetime | None = None
) -> datetime | None:
if event == SUN_EVENT_SUNRISE:
return sunrise_time
return sunset_time
@ -882,7 +903,9 @@ async def test_flux_after_stop_before_sunrise_stop_next_day(
sunset_time = test_time.replace(hour=17, minute=0, second=0)
sunrise_time = test_time.replace(hour=5, minute=0, second=0)
def event_date(hass, event, now=None):
def event_date(
hass: HomeAssistant, event: str, now: date | datetime | None = None
) -> datetime | None:
if event == SUN_EVENT_SUNRISE:
return sunrise_time
return sunset_time
@ -945,7 +968,9 @@ async def test_flux_with_custom_colortemps(
sunset_time = test_time.replace(hour=17, minute=0, second=0)
sunrise_time = test_time.replace(hour=5, minute=0, second=0)
def event_date(hass, event, now=None):
def event_date(
hass: HomeAssistant, event: str, now: date | datetime | None = None
) -> datetime | None:
if event == SUN_EVENT_SUNRISE:
return sunrise_time
return sunset_time
@ -1010,7 +1035,9 @@ async def test_flux_with_custom_brightness(
sunset_time = test_time.replace(hour=17, minute=0, second=0)
sunrise_time = test_time.replace(hour=5, minute=0, second=0)
def event_date(hass, event, now=None):
def event_date(
hass: HomeAssistant, event: str, now: date | datetime | None = None
) -> datetime | None:
if event == SUN_EVENT_SUNRISE:
return sunrise_time
return sunset_time
@ -1091,7 +1118,9 @@ async def test_flux_with_multiple_lights(
sunset_time = test_time.replace(hour=17, minute=0, second=0)
sunrise_time = test_time.replace(hour=5, minute=0, second=0)
def event_date(hass, event, now=None):
def event_date(
hass: HomeAssistant, event: str, now: date | datetime | None = None
) -> datetime | None:
if event == SUN_EVENT_SUNRISE:
return sunrise_time
return sunset_time
@ -1158,7 +1187,9 @@ async def test_flux_with_mired(
sunset_time = test_time.replace(hour=17, minute=0, second=0)
sunrise_time = test_time.replace(hour=5, minute=0, second=0)
def event_date(hass, event, now=None):
def event_date(
hass: HomeAssistant, event: str, now: date | datetime | None = None
) -> datetime | None:
if event == SUN_EVENT_SUNRISE:
return sunrise_time
return sunset_time
@ -1219,7 +1250,9 @@ async def test_flux_with_rgb(
sunset_time = test_time.replace(hour=17, minute=0, second=0)
sunrise_time = test_time.replace(hour=5, minute=0, second=0)
def event_date(hass, event, now=None):
def event_date(
hass: HomeAssistant, event: str, date: date | datetime | None = None
) -> datetime | None:
if event == SUN_EVENT_SUNRISE:
return sunrise_time
return sunset_time