From 95d228cacec4c90120af071ef27d07a9644f15ae Mon Sep 17 00:00:00 2001
From: Eliot Wong <eliotw@google.com>
Date: Thu, 1 Oct 2020 03:42:23 -0400
Subject: [PATCH] Rewrite worldclock unittest tests to pytest style test
 functions (#40922)

---
 tests/components/worldclock/test_sensor.py | 73 +++++++++++-----------
 1 file changed, 38 insertions(+), 35 deletions(-)

diff --git a/tests/components/worldclock/test_sensor.py b/tests/components/worldclock/test_sensor.py
index e029ce783d3..8ed808b9d04 100644
--- a/tests/components/worldclock/test_sensor.py
+++ b/tests/components/worldclock/test_sensor.py
@@ -1,49 +1,52 @@
 """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
 
-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):
-    """Test the World clock sensor."""
+async def test_time(hass, time_zone):
+    """Test the time at a different location."""
+    config = {"sensor": {"platform": "worldclock", "time_zone": "America/New_York"}}
 
-    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")
+    assert await async_setup_component(
+        hass,
+        "sensor",
+        config,
+    )
+    await hass.async_block_till_done()
 
-    def test_time(self):
-        """Test the time at a different location."""
-        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 = hass.states.get("sensor.worldclock_sensor")
+    assert state is not None
 
-        state = self.hass.states.get("sensor.worldclock_sensor")
-        assert state is not None
+    assert state.state == dt_util.now(time_zone=time_zone).strftime("%H:%M")
 
-        assert state.state == dt_util.now(time_zone=self.time_zone).strftime("%H:%M")
 
-    def test_time_format(self):
-        """Test time_format setting."""
-        time_format = "%a, %b %d, %Y %I:%M %p"
-        config = {
-            "sensor": {
-                "platform": "worldclock",
-                "time_zone": "America/New_York",
-                "time_format": time_format,
-            }
+async def test_time_format(hass, time_zone):
+    """Test time_format setting."""
+    time_format = "%a, %b %d, %Y %I:%M %p"
+    config = {
+        "sensor": {
+            "platform": "worldclock",
+            "time_zone": "America/New_York",
+            "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 state is not None
+    assert await async_setup_component(
+        hass,
+        "sensor",
+        config,
+    )
+    await hass.async_block_till_done()
 
-        assert state.state == dt_util.now(time_zone=self.time_zone).strftime(
-            time_format
-        )
+    state = hass.states.get("sensor.worldclock_sensor")
+    assert state is not None
+
+    assert state.state == dt_util.now(time_zone=time_zone).strftime(time_format)