jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-23 21:27:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

Rewrite worldclock unittest tests to pytest style test functions (#40922)

Browse Source
This commit is contained in:
Eliot Wong 2020-10-01 03:42:23 -04:00 committed by GitHub
parent d95f83b5fb
commit 95d228cace
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 38 additions and 35 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

53
tests/components/worldclock/test_sensor.py
View File

@ -1,33 +1,34 @@
"""The test for the World clock sensor platform.""" """The test for the World clock sensor platform."""
import unittest import pytest
from homeassistant.setup import setup_component from homeassistant.setup import async_setup_component
import homeassistant.util.dt as dt_util import homeassistant.util.dt as dt_util
from tests.common import get_test_home_assistant
@pytest.fixture
def time_zone():
"""Fixture for time zone."""
return dt_util.get_time_zone("America/New_York")
class TestWorldClockSensor(unittest.TestCase): async def test_time(hass, time_zone):
"""Test the World clock sensor."""
def setUp(self):
"""Set up things to be run when tests are started."""
self.hass = get_test_home_assistant()
self.time_zone = dt_util.get_time_zone("America/New_York")
def test_time(self):
"""Test the time at a different location.""" """Test the time at a different location."""
config = {"sensor": {"platform": "worldclock", "time_zone": "America/New_York"}} config = {"sensor": {"platform": "worldclock", "time_zone": "America/New_York"}}
assert setup_component(self.hass, "sensor", config)
self.hass.block_till_done()
self.addCleanup(self.hass.stop)
state = self.hass.states.get("sensor.worldclock_sensor") assert await async_setup_component(
hass,
"sensor",
config,
)
await hass.async_block_till_done()
state = hass.states.get("sensor.worldclock_sensor")
assert state is not None assert state is not None
assert state.state == dt_util.now(time_zone=self.time_zone).strftime("%H:%M") assert state.state == dt_util.now(time_zone=time_zone).strftime("%H:%M")
def test_time_format(self):
async def test_time_format(hass, time_zone):
"""Test time_format setting.""" """Test time_format setting."""
time_format = "%a, %b %d, %Y %I:%M %p" time_format = "%a, %b %d, %Y %I:%M %p"
config = { config = {
@ -37,13 +38,15 @@ class TestWorldClockSensor(unittest.TestCase):
"time_format": time_format, "time_format": time_format,
} }
} }
assert setup_component(self.hass, "sensor", config)
self.hass.block_till_done()
self.addCleanup(self.hass.stop)
state = self.hass.states.get("sensor.worldclock_sensor") assert await async_setup_component(
hass,
"sensor",
config,
)
await hass.async_block_till_done()
state = hass.states.get("sensor.worldclock_sensor")
assert state is not None assert state is not None
assert state.state == dt_util.now(time_zone=self.time_zone).strftime( assert state.state == dt_util.now(time_zone=time_zone).strftime(time_format)
time_format
)