Use TestCase.addCleanup (#36560)
This commit is contained in:
parent
85ba29012f
commit
c33edbe5bb
103 changed files with 179 additions and 254 deletions
|
@ -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()."""
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue