starred/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2026-09-26 01:11:51 -04:00
Code Issues Packages Projects Releases Wiki Activity

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
1 changed files with 49 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/components/flux/test_switch.py
+49 -16
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