Use TestCase.addCleanup (#36560)

This commit is contained in:
Paulus Schoutsen 2020-06-08 12:26:40 -07:00 committed by GitHub
parent 85ba29012f
commit c33edbe5bb
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
103 changed files with 179 additions and 254 deletions

View file

@ -92,8 +92,9 @@ class TestMQTTComponent(unittest.TestCase):
self.hass = get_test_home_assistant()
mock_mqtt_component(self.hass)
self.calls = []
self.addCleanup(self.tear_down_cleanup)
def tearDown(self): # pylint: disable=invalid-name
def tear_down_cleanup(self):
"""Stop everything that was started."""
self.hass.stop()
@ -307,8 +308,9 @@ class TestMQTTCallbacks(unittest.TestCase):
self.hass = get_test_home_assistant()
mock_mqtt_client(self.hass)
self.calls = []
self.addCleanup(self.tear_down_cleanup)
def tearDown(self): # pylint: disable=invalid-name
def tear_down_cleanup(self):
"""Stop everything that was started."""
self.hass.stop()