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
103 changed files with 179 additions and 254 deletions

View File

@@ -18,8 +18,9 @@ class TestLightRfxtrx(unittest.TestCase):
"""Set up things to be run when tests are started."""
self.hass = get_test_home_assistant()
mock_component(self.hass, "rfxtrx")
self.addCleanup(self.tear_down_cleanup)
def tearDown(self):
def tear_down_cleanup(self):
"""Stop everything that was started."""
rfxtrx_core.RECEIVED_EVT_SUBSCRIBERS = []
rfxtrx_core.RFX_DEVICES = {}