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

@ -25,10 +25,7 @@ class TestRestSensorSetup(unittest.TestCase):
def setUp(self):
"""Set up things to be run when tests are started."""
self.hass = get_test_home_assistant()
def tearDown(self):
"""Stop everything that was started."""
self.hass.stop()
self.addCleanup(self.hass.stop)
def test_setup_missing_config(self):
"""Test setup with configuration missing required entries."""
@ -237,10 +234,7 @@ class TestRestSensor(unittest.TestCase):
self.resource_template,
self.json_attrs_path,
)
def tearDown(self):
"""Stop everything that was started."""
self.hass.stop()
self.addCleanup(self.hass.stop)
def update_side_effect(self, data, headers):
"""Side effect function for mocking RestData.update()."""