Use TestCase.addCleanup (#36560)

2020-06-08 12:26:40 -07:00 · 2020-06-08 12:26:40 -07:00 · c33edbe5bb
commit c33edbe5bb
parent 85ba29012f
103 changed files with 179 additions and 254 deletions
--- a/tests/components/influxdb/test_init.py
+++ b/tests/components/influxdb/test_init.py
@ -29,8 +29,9 @@ class TestInfluxDB(unittest.TestCase):
        self.hass = get_test_home_assistant()
        self.handler_method = None
        self.hass.bus.listen = mock.Mock()
+        self.addCleanup(self.tear_down_cleanup)

-    def tearDown(self):
+    def tear_down_cleanup(self):
        """Clear data."""
        self.hass.stop()