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

Use setup_test_component_platform helper for time entity component tests instead of hass.components (#114411)

Browse source
This commit is contained in:
Jan-Philipp Benecke 2024-03-29 07:34:00 +01:00 • committed by GitHub
parent ae635a5586
commit a102230498
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 30 additions and 66 deletions
Download patch file Download diff file Expand all files Collapse all files

25
tests/components/time/test_init.py
View file

@ -2,7 +2,7 @@
from datetime import time
from homeassistant.components.time import DOMAIN, SERVICE_SET_VALUE, TimeEntity
from homeassistant.components.time import DOMAIN, SERVICE_SET_VALUE
from homeassistant.const import (
ATTR_ENTITY_ID,
ATTR_FRIENDLY_NAME,
@ -12,23 +12,18 @@ from homeassistant.const import (
from homeassistant.core import HomeAssistant
from homeassistant.setup import async_setup_component
class MockTimeEntity(TimeEntity):
"""Mock time device to use in tests."""
def __init__(self, native_value=time(12, 0, 0)) -> None:
"""Initialize mock time entity."""
self._attr_native_value = native_value
async def async_set_value(self, value: time) -> None:
"""Set the value of the time."""
self._attr_native_value = value
from tests.common import setup_test_component_platform
from tests.components.time.common import MockTimeEntity
async def test_date(hass: HomeAssistant, enable_custom_integrations: None) -> None:
async def test_date(hass: HomeAssistant) -> None:
"""Test time entity."""
platform = getattr(hass.components, f"test.{DOMAIN}")
platform.init()
entity = MockTimeEntity(
name="test",
unique_id="unique_time",
native_value=time(1, 2, 3),
)
setup_test_component_platform(hass, DOMAIN, [entity])
assert await async_setup_component(hass, DOMAIN, {DOMAIN: {CONF_PLATFORM: "test"}})
await hass.async_block_till_done()