Use TestCase.addCleanup (#36560)

tests/components/alert/test_init.py

@@ -125,10 +125,7 @@ class TestAlert(unittest.TestCase):
         """Set up things to be run when tests are started."""
         self.hass = get_test_home_assistant()
         self._setup_notify()
-
-    def tearDown(self):
-        """Stop everything that was started."""
-        self.hass.stop()
+        self.addCleanup(self.hass.stop)
 
     def _setup_notify(self):
         events = []

tests/components/apns/test_notify.py

@@ -30,8 +30,9 @@ class TestApns(unittest.TestCase):
     def setUp(self):  # pylint: disable=invalid-name
         """Set up things to be run when tests are started."""
         self.hass = get_test_home_assistant()
+        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()
 

tests/components/aurora/test_binary_sensor.py

@@ -20,8 +20,9 @@ class TestAuroraSensorSetUp(unittest.TestCase):
         self.hass.config.latitude = self.lat
         self.hass.config.longitude = self.lon
         self.entities = []
+        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()
 

tests/components/blackbird/test_media_player.py

@@ -203,8 +203,9 @@ class TestBlackbirdMediaPlayer(unittest.TestCase):
         self.media_player = self.hass.data[DATA_BLACKBIRD]["/dev/ttyUSB0-3"]
         self.media_player.hass = self.hass
         self.media_player.entity_id = "media_player.zone_3"
+        self.addCleanup(self.tear_down_cleanup)
 
-    def tearDown(self):
+    def tear_down_cleanup(self):
         """Tear down the test case."""
         self.hass.stop()
 

tests/components/bom/test_sensor.py

@@ -60,10 +60,7 @@ class TestBOMWeatherSensor(unittest.TestCase):
         """Set up things to be run when tests are started."""
         self.hass = get_test_home_assistant()
         self.config = VALID_CONFIG
-
-    def tearDown(self):
-        """Stop everything that was started."""
-        self.hass.stop()
+        self.addCleanup(self.hass.stop)
 
     @patch("requests.get", side_effect=mocked_requests)
     def test_setup(self, mock_get):

tests/components/canary/test_init.py

@@ -43,8 +43,9 @@ class TestCanary(unittest.TestCase):
     def setUp(self):
         """Initialize values for this test case class."""
         self.hass = get_test_home_assistant()
+        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()
 

tests/components/canary/test_sensor.py

@@ -34,10 +34,7 @@ class TestCanarySensorSetup(unittest.TestCase):
         """Initialize values for this testcase class."""
         self.hass = get_test_home_assistant()
         self.config = copy.deepcopy(VALID_CONFIG)
-
-    def tearDown(self):
-        """Stop everything that was started."""
-        self.hass.stop()
+        self.addCleanup(self.hass.stop)
 
     def test_setup_sensors(self):
         """Test the sensor setup."""

tests/components/coinmarketcap/test_sensor.py

@@ -23,10 +23,7 @@ class TestCoinMarketCapSensor(unittest.TestCase):
         """Set up things to be run when tests are started."""
         self.hass = get_test_home_assistant()
         self.config = VALID_CONFIG
-
-    def tearDown(self):
-        """Stop everything that was started."""
-        self.hass.stop()
+        self.addCleanup(self.hass.stop)
 
     @patch(
         "coinmarketcap.Market.ticker",

tests/components/command_line/test_binary_sensor.py

@@ -14,10 +14,7 @@ class TestCommandSensorBinarySensor(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(self):
         """Test sensor setup."""

tests/components/command_line/test_notify.py

@@ -16,8 +16,9 @@ class TestCommandLine(unittest.TestCase):
     def setUp(self):  # pylint: disable=invalid-name
         """Set up things to be run when tests are started."""
         self.hass = get_test_home_assistant()
+        self.addCleanup(self.tear_down_cleanup)
 
-    def tearDown(self):  # pylint: disable=invalid-name
+    def tear_down_cleanup(self):
         """Stop down everything that was started."""
         self.hass.stop()
 

tests/components/command_line/test_sensor.py

@@ -14,10 +14,7 @@ class TestCommandSensorSensor(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 update_side_effect(self, data):
         """Side effect function for mocking CommandSensorData.update()."""

tests/components/command_line/test_switch.py

@@ -20,10 +20,7 @@ class TestCommandSwitch(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_state_none(self):
         """Test with none state."""

tests/components/configurator/test_init.py

@@ -15,11 +15,7 @@ class TestConfigurator(unittest.TestCase):
     def setUp(self):
         """Set up things to be run when tests are started."""
         self.hass = get_test_home_assistant()
-
-    # pylint: disable=invalid-name
-    def tearDown(self):
-        """Stop everything that was started."""
-        self.hass.stop()
+        self.addCleanup(self.hass.stop)
 
     def test_request_least_info(self):
         """Test request config with least amount of data."""

tests/components/darksky/test_sensor.py

@@ -105,8 +105,9 @@ class TestDarkSkySetup(unittest.TestCase):
         self.lat = self.hass.config.latitude = 37.8267
         self.lon = self.hass.config.longitude = -122.423
         self.entities = []
+        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()
 

tests/components/darksky/test_weather.py

@@ -23,8 +23,9 @@ class TestDarkSky(unittest.TestCase):
         self.hass.config.units = METRIC_SYSTEM
         self.lat = self.hass.config.latitude = 37.8267
         self.lon = self.hass.config.longitude = -122.423
+        self.addCleanup(self.tear_down_cleanup)
 
-    def tearDown(self):
+    def tear_down_cleanup(self):
         """Stop down everything that was started."""
         self.hass.stop()
 

tests/components/datadog/test_init.py

@@ -21,8 +21,9 @@ class TestDatadog(unittest.TestCase):
     def setUp(self):  # pylint: disable=invalid-name
         """Set up things to be run when tests are started."""
         self.hass = get_test_home_assistant()
+        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()
 

tests/components/demo/test_geo_location.py

@@ -26,10 +26,7 @@ class TestDemoPlatform(unittest.TestCase):
     def setUp(self):
         """Initialize values for this testcase class."""
         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_platform(self):
         """Test setup of demo platform via configuration."""

tests/components/demo/test_notify.py

@@ -32,8 +32,9 @@ class TestNotifyDemo(unittest.TestCase):
             self.events.append(event)
 
         self.hass.bus.listen(demo.EVENT_NOTIFY, record_event)
+        self.addCleanup(self.tear_down_cleanup)
 
-    def tearDown(self):  # pylint: disable=invalid-name
+    def tear_down_cleanup(self):
         """Stop down everything that was started."""
         self.hass.stop()
 

tests/components/demo/test_remote.py

@@ -24,8 +24,9 @@ class TestDemoRemote(unittest.TestCase):
         )
         self.hass.block_till_done()
 
-    # pylint: disable=invalid-name
-    def tearDown(self):
+        self.addCleanup(self.tear_down_cleanup)
+
+    def tear_down_cleanup(self):
         """Stop down everything that was started."""
         self.hass.stop()
 

tests/components/dte_energy_bridge/test_sensor.py

@@ -17,10 +17,7 @@ class TestDteEnergyBridgeSetup(unittest.TestCase):
     def setUp(self):
         """Initialize values for this testcase class."""
         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_with_config(self):
         """Test the platform setup with configuration."""

tests/components/dyson/test_climate.py

@@ -111,8 +111,9 @@ class DysonTest(unittest.TestCase):
     def setUp(self):  # pylint: disable=invalid-name
         """Set up things to be run when tests are started."""
         self.hass = get_test_home_assistant()
+        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()
 

tests/components/dyson/test_fan.py

@@ -136,8 +136,9 @@ class DysonSetupTest(unittest.TestCase):
     def setUp(self):  # pylint: disable=invalid-name
         """Set up things to be run when tests are started."""
         self.hass = get_test_home_assistant()
+        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()
 
@@ -173,8 +174,9 @@ class DysonTest(unittest.TestCase):
     def setUp(self):  # pylint: disable=invalid-name
         """Set up things to be run when tests are started."""
         self.hass = get_test_home_assistant()
+        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()
 

tests/components/dyson/test_init.py

@@ -41,8 +41,9 @@ class DysonTest(unittest.TestCase):
     def setUp(self):  # pylint: disable=invalid-name
         """Set up things to be run when tests are started."""
         self.hass = get_test_home_assistant()
+        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()
 

tests/components/dyson/test_sensor.py

@@ -84,8 +84,9 @@ class DysonTest(unittest.TestCase):
     def setUp(self):  # pylint: disable=invalid-name
         """Set up things to be run when tests are started."""
         self.hass = get_test_home_assistant()
+        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()
 

tests/components/dyson/test_vacuum.py

@@ -70,8 +70,9 @@ class DysonTest(unittest.TestCase):
     def setUp(self):  # pylint: disable=invalid-name
         """Set up things to be run when tests are started."""
         self.hass = get_test_home_assistant()
+        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()
 

tests/components/efergy/test_sensor.py

@@ -75,8 +75,9 @@ class TestEfergySensor(unittest.TestCase):
         """Initialize values for this test case class."""
         self.hass = get_test_home_assistant()
         self.config = ONE_SENSOR_CONFIG
+        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()
 

tests/components/fail2ban/test_sensor.py

@@ -66,10 +66,7 @@ class TestBanSensor(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)
 
     @patch("os.path.isfile", Mock(return_value=True))
     def test_setup(self):

tests/components/fan/test_init.py

@@ -21,8 +21,9 @@ class TestFanEntity(unittest.TestCase):
     def setUp(self):
         """Set up test data."""
         self.fan = BaseFan()
+        self.addCleanup(self.tear_down_cleanup)
 
-    def tearDown(self):
+    def tear_down_cleanup(self):
         """Tear down unit test data."""
         self.fan = None
 

tests/components/feedreader/test_init.py

@@ -42,10 +42,7 @@ class TestFeedreaderComponent(unittest.TestCase):
         data_file = self.hass.config.path(f"{feedreader.DOMAIN}.pickle")
         if exists(data_file):
             remove(data_file)
-
-    def tearDown(self):
-        """Stop everything that was started."""
-        self.hass.stop()
+        self.addCleanup(self.hass.stop)
 
     def test_setup_one_feed(self):
         """Test the general setup of this component."""

tests/components/file/test_notify.py

@@ -17,8 +17,9 @@ class TestNotifyFile(unittest.TestCase):
     def setUp(self):  # pylint: disable=invalid-name
         """Set up things to be run when tests are started."""
         self.hass = get_test_home_assistant()
+        self.addCleanup(self.tear_down_cleanup)
 
-    def tearDown(self):  # pylint: disable=invalid-name
+    def tear_down_cleanup(self):
         """Stop down everything that was started."""
         self.hass.stop()
 

tests/components/geo_rss_events/test_sensor.py

@@ -42,10 +42,7 @@ class TestGeoRssServiceUpdater(unittest.TestCase):
         """Initialize values for this testcase class."""
         self.hass = get_test_home_assistant()
         # self.config = VALID_CONFIG_WITHOUT_CATEGORIES
-
-    def tearDown(self):
-        """Stop everything that was started."""
-        self.hass.stop()
+        self.addCleanup(self.hass.stop)
 
     @staticmethod
     def _generate_mock_feed_entry(

tests/components/google_wifi/test_sensor.py

@@ -40,10 +40,7 @@ class TestGoogleWifiSetup(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)
 
     @requests_mock.Mocker()
     def test_setup_minimum(self, mock_req):
@@ -92,10 +89,7 @@ class TestGoogleWifiSensor(unittest.TestCase):
         self.hass = get_test_home_assistant()
         with requests_mock.Mocker() as mock_req:
             self.setup_api(MOCK_DATA, mock_req)
-
-    def tearDown(self):
-        """Stop everything that was started."""
-        self.hass.stop()
+        self.addCleanup(self.hass.stop)
 
     def setup_api(self, data, mock_req):
         """Set up API with fake data."""

tests/components/group/test_init.py

@@ -28,11 +28,7 @@ class TestComponentsGroup(unittest.TestCase):
     def setUp(self):
         """Set up things to be run when tests are started."""
         self.hass = get_test_home_assistant()
-
-    # pylint: disable=invalid-name
-    def tearDown(self):
-        """Stop everything that was started."""
-        self.hass.stop()
+        self.addCleanup(self.hass.stop)
 
     def test_setup_group_with_mixed_groupable_states(self):
         """Try to set up a group with mixed groupable states."""

tests/components/group/test_notify.py

@@ -63,8 +63,9 @@ class TestNotifyGroup(unittest.TestCase):
         ).result()
 
         assert self.service is not None
+        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()
 

tests/components/hddtemp/test_sensor.py

@@ -87,10 +87,7 @@ class TestHDDTempSensor(unittest.TestCase):
                 "model": "WDC WD15EARS-00Z5B1",
             },
         }
-
-    def tearDown(self):
-        """Stop everything that was started."""
-        self.hass.stop()
+        self.addCleanup(self.hass.stop)
 
     @patch("telnetlib.Telnet", new=TelnetMock)
     def test_hddtemp_min_config(self):

tests/components/history/test_init.py

@@ -26,8 +26,9 @@ class TestComponentHistory(unittest.TestCase):
     def setUp(self):  # pylint: disable=invalid-name
         """Set up things to be run when tests are started."""
         self.hass = get_test_home_assistant()
+        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()
 

tests/components/history_stats/test_sensor.py

@@ -23,10 +23,7 @@ class TestHistoryStatsSensor(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(self):
         """Test the history statistics sensor setup."""

tests/components/homeassistant/test_init.py

@@ -120,11 +120,7 @@ class TestComponentsCore(unittest.TestCase):
 
         self.hass.states.set("light.Bowl", STATE_ON)
         self.hass.states.set("light.Ceiling", STATE_OFF)
-
-    # pylint: disable=invalid-name
-    def tearDown(self):
-        """Stop everything that was started."""
-        self.hass.stop()
+        self.addCleanup(self.hass.stop)
 
     def test_is_on(self):
         """Test is_on method."""

tests/components/homematic/test_notify.py

@@ -14,8 +14,9 @@ class TestHomematicNotify(unittest.TestCase):
     def setUp(self):  # pylint: disable=invalid-name
         """Set up things to be run when tests are started."""
         self.hass = get_test_home_assistant()
+        self.addCleanup(self.tear_down_cleanup)
 
-    def tearDown(self):  # pylint: disable=invalid-name
+    def tear_down_cleanup(self):
         """Stop down everything that was started."""
         self.hass.stop()
 

tests/components/imap_email_content/test_sensor.py

@@ -37,10 +37,7 @@ class EmailContentSensor(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_allowed_sender(self):
         """Test emails from allowed sender."""

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()
 

tests/components/kira/test_init.py

@@ -45,8 +45,9 @@ class TestKiraSetup(unittest.TestCase):
         self._module_patcher.start()
 
         self.work_dir = tempfile.mkdtemp()
+        self.addCleanup(self.tear_down_cleanup)
 
-    def tearDown(self):
+    def tear_down_cleanup(self):
         """Stop everything that was started."""
         self.hass.stop()
         self._module_patcher.stop()

tests/components/kira/test_remote.py

@@ -30,10 +30,7 @@ class TestKiraSensor(unittest.TestCase):
         self.mock_kira = MagicMock()
         self.hass.data[kira.DOMAIN] = {kira.CONF_REMOTE: {}}
         self.hass.data[kira.DOMAIN][kira.CONF_REMOTE]["kira"] = self.mock_kira
-
-    def tearDown(self):
-        """Stop everything that was started."""
-        self.hass.stop()
+        self.addCleanup(self.hass.stop)
 
     def test_service_call(self):
         """Test Kira's ability to send commands."""

tests/components/kira/test_sensor.py

@@ -28,10 +28,7 @@ class TestKiraSensor(unittest.TestCase):
         mock_kira = MagicMock()
         self.hass.data[kira.DOMAIN] = {kira.CONF_SENSOR: {}}
         self.hass.data[kira.DOMAIN][kira.CONF_SENSOR]["kira"] = mock_kira
-
-    def tearDown(self):
-        """Stop everything that was started."""
-        self.hass.stop()
+        self.addCleanup(self.hass.stop)
 
     # pylint: disable=protected-access
     def test_kira_sensor_callback(self):

tests/components/light/test_init.py

@@ -32,9 +32,9 @@ class TestLight(unittest.TestCase):
     def setUp(self):
         """Set up things to be run when tests are started."""
         self.hass = get_test_home_assistant()
+        self.addCleanup(self.tear_down_cleanup)
 
-    # pylint: disable=invalid-name
-    def tearDown(self):
+    def tear_down_cleanup(self):
         """Stop everything that was started."""
         self.hass.stop()
 

tests/components/logbook/test_init.py

@@ -44,10 +44,7 @@ class TestComponentLogbook(unittest.TestCase):
         init_recorder_component(self.hass)  # Force an in memory DB
         with patch("homeassistant.components.http.start_http_server_and_save_config"):
             assert setup_component(self.hass, logbook.DOMAIN, self.EMPTY_CONFIG)
-
-    def tearDown(self):
-        """Stop everything that was started."""
-        self.hass.stop()
+        self.addCleanup(self.hass.stop)
 
     def test_service_call_create_logbook_entry(self):
         """Test if service call create log book entry."""

tests/components/logentries/test_init.py

@@ -16,8 +16,9 @@ class TestLogentries(unittest.TestCase):
     def setUp(self):  # pylint: disable=invalid-name
         """Set up things to be run when tests are started."""
         self.hass = get_test_home_assistant()
+        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()
 

tests/components/london_air/test_sensor.py

@@ -18,10 +18,7 @@ class TestLondonAirSensor(unittest.TestCase):
         """Initialize values for this testcase class."""
         self.hass = get_test_home_assistant()
         self.config = VALID_CONFIG
-
-    def tearDown(self):
-        """Stop everything that was started."""
-        self.hass.stop()
+        self.addCleanup(self.hass.stop)
 
     @requests_mock.Mocker()
     def test_setup(self, mock_req):

tests/components/manual_mqtt/test_alarm_control_panel.py

@@ -35,8 +35,9 @@ class TestAlarmControlPanelManualMqtt(unittest.TestCase):
         self.hass = get_test_home_assistant()
         self.hass.config_entries._async_schedule_save = Mock()
         self.mock_publish = mock_mqtt_component(self.hass)
+        self.addCleanup(self.tear_down_cleanup)
 
-    def tearDown(self):  # pylint: disable=invalid-name
+    def tear_down_cleanup(self):
         """Stop down everything that was started."""
         self.hass.stop()
 

tests/components/media_player/test_async_helpers.py

@@ -146,8 +146,9 @@ class TestAsyncMediaPlayer(unittest.TestCase):
         """Set up things to be run when tests are started."""
         self.hass = get_test_home_assistant()
         self.player = AsyncMediaPlayer(self.hass)
+        self.addCleanup(self.tear_down_cleanup)
 
-    def tearDown(self):
+    def tear_down_cleanup(self):
         """Shut down test instance."""
         self.hass.stop()
 
@@ -207,8 +208,9 @@ class TestSyncMediaPlayer(unittest.TestCase):
         """Set up things to be run when tests are started."""
         self.hass = get_test_home_assistant()
         self.player = SyncMediaPlayer(self.hass)
+        self.addCleanup(self.tear_down_cleanup)
 
-    def tearDown(self):
+    def tear_down_cleanup(self):
         """Shut down test instance."""
         self.hass.stop()
 

tests/components/mochad/test_light.py

@@ -28,10 +28,7 @@ class TestMochadSwitchSetup(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)
 
     @mock.patch("homeassistant.components.mochad.light.MochadLight")
     def test_setup_adds_proper_devices(self, mock_light):

tests/components/mochad/test_switch.py

@@ -30,10 +30,7 @@ class TestMochadSwitchSetup(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)
 
     @mock.patch("homeassistant.components.mochad.switch.MochadSwitch")
     def test_setup_adds_proper_devices(self, mock_switch):

tests/components/mold_indicator/test_sensor.py

@@ -32,8 +32,9 @@ class TestSensorMoldIndicator(unittest.TestCase):
         self.hass.states.set(
             "test.indoorhumidity", "50", {ATTR_UNIT_OF_MEASUREMENT: UNIT_PERCENTAGE}
         )
+        self.addCleanup(self.tear_down_cleanup)
 
-    def tearDown(self):
+    def tear_down_cleanup(self):
         """Stop down everything that was started."""
         self.hass.stop()
 

tests/components/mqtt/test_init.py

@@ -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()
 

tests/components/nsw_fuel_station/test_sensor.py

@@ -79,10 +79,7 @@ class TestNSWFuelStation(unittest.TestCase):
         """Set up things to be run when tests are started."""
         self.hass = get_test_home_assistant()
         self.config = VALID_CONFIG
-
-    def tearDown(self):
-        """Stop everything that was started."""
-        self.hass.stop()
+        self.addCleanup(self.hass.stop)
 
     @patch(
         "homeassistant.components.nsw_fuel_station.sensor.FuelCheckClient",

tests/components/nx584/test_binary_sensor.py

@@ -36,8 +36,9 @@ class TestNX584SensorSetup(unittest.TestCase):
         client = nx584_client.Client.return_value
         client.list_zones.return_value = self.fake_zones
         client.get_version.return_value = "1.1"
+        self.addCleanup(self.tear_down_cleanup)
 
-    def tearDown(self):
+    def tear_down_cleanup(self):
         """Stop everything that was started."""
         self.hass.stop()
         self._mock_client.stop()

tests/components/openhardwaremonitor/test_sensor.py

@@ -21,8 +21,9 @@ class TestOpenHardwareMonitorSetup(unittest.TestCase):
                 "port": 8085,
             }
         }
+        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()
 

tests/components/panel_iframe/test_init.py

@@ -14,10 +14,7 @@ class TestPanelIframe(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_wrong_config(self):
         """Test setup with wrong configuration."""

tests/components/pilight/test_init.py

@@ -71,8 +71,9 @@ class TestPilight(unittest.TestCase):
         """Set up things to be run when tests are started."""
         self.hass = get_test_home_assistant()
         self.skip_teardown_stop = False
+        self.addCleanup(self.tear_down_cleanup)
 
-    def tearDown(self):
+    def tear_down_cleanup(self):
         """Stop everything that was started."""
         if not self.skip_teardown_stop:
             self.hass.stop()
@@ -374,10 +375,7 @@ class TestPilightCallrateThrottler(unittest.TestCase):
     def setUp(self):  # pylint: disable=invalid-name
         """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_call_rate_delay_throttle_disabled(self):
         """Test that the limiter is a noop if no delay set."""

tests/components/proximity/test_init.py

@@ -25,10 +25,7 @@ class TestProximity(unittest.TestCase):
             "zoning",
             {"name": "work", "latitude": 2.3, "longitude": 1.3, "radius": 10},
         )
-
-    def tearDown(self):
-        """Stop everything that was started."""
-        self.hass.stop()
+        self.addCleanup(self.hass.stop)
 
     def test_proximities(self):
         """Test a list of proximities."""

tests/components/pushbullet/test_notify.py

@@ -18,8 +18,9 @@ class TestPushBullet(unittest.TestCase):
     def setUp(self):
         """Initialize values for this test case class."""
         self.hass = get_test_home_assistant()
+        self.addCleanup(self.tear_down_cleanup)
 
-    def tearDown(self):  # pylint: disable=invalid-name
+    def tear_down_cleanup(self):
         """Stop everything that we started."""
         self.hass.stop()
 

tests/components/radarr/test_sensor.py

@@ -208,8 +208,9 @@ class TestRadarrSetup(unittest.TestCase):
         self.DEVICES = []
         self.hass = get_test_home_assistant()
         self.hass.config.time_zone = "America/Los_Angeles"
+        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()
 

tests/components/recorder/test_init.py

@@ -28,8 +28,9 @@ class TestRecorder(unittest.TestCase):
         self.hass = get_test_home_assistant()
         init_recorder_component(self.hass)
         self.hass.start()
+        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()
 

tests/components/recorder/test_models.py

@@ -88,8 +88,9 @@ class TestRecorderRuns(unittest.TestCase):
         session.query(Events).delete()
         session.query(States).delete()
         session.query(RecorderRuns).delete()
+        self.addCleanup(self.tear_down_cleanup)
 
-    def tearDown(self):  # pylint: disable=invalid-name
+    def tear_down_cleanup(self):
         """Clean up."""
         self.session.rollback()
 

tests/components/recorder/test_purge.py

@@ -21,8 +21,9 @@ class TestRecorderPurge(unittest.TestCase):
         self.hass = get_test_home_assistant()
         init_recorder_component(self.hass)
         self.hass.start()
+        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()
 

tests/components/reddit/test_sensor.py

@@ -157,8 +157,9 @@ class TestRedditSetup(unittest.TestCase):
     def setUp(self):
         """Initialize values for this testcase class."""
         self.hass = get_test_home_assistant()
+        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()
 

tests/components/remember_the_milk/test_init.py

@@ -30,8 +30,9 @@ class TestConfiguration(unittest.TestCase):
                 }
             }
         )
+        self.addCleanup(self.tear_down_cleanup)
 
-    def tearDown(self):
+    def tear_down_cleanup(self):
         """Exit Home Assistant."""
         self.hass.stop()
 

tests/components/remote/test_init.py

@@ -28,11 +28,7 @@ class TestRemote(unittest.TestCase):
     def setUp(self):
         """Set up things to be run when tests are started."""
         self.hass = get_test_home_assistant()
-
-    # pylint: disable=invalid-name
-    def tearDown(self):
-        """Stop everything that was started."""
-        self.hass.stop()
+        self.addCleanup(self.hass.stop)
 
     def test_is_on(self):
         """Test is_on."""

tests/components/rest/test_binary_sensor.py

@@ -33,10 +33,7 @@ class TestRestBinarySensorSetup(unittest.TestCase):
         self.hass = get_test_home_assistant()
         # Reset for this test.
         self.DEVICES = []
-
-    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."""
@@ -162,10 +159,7 @@ class TestRestBinarySensor(unittest.TestCase):
         self.binary_sensor = rest.RestBinarySensor(
             self.hass, self.rest, self.name, self.device_class, self.value_template
         )
-
-    def tearDown(self):
-        """Stop everything that was started."""
-        self.hass.stop()
+        self.addCleanup(self.hass.stop)
 
     def update_side_effect(self, data):
         """Side effect function for mocking RestData.update()."""

tests/components/rest/test_sensor.py

@@ -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()."""

tests/components/rfxtrx/test_cover.py

@@ -18,8 +18,9 @@ class TestCoverRfxtrx(unittest.TestCase):
         """Set up things to be run when tests are started."""
         self.hass = get_test_home_assistant()
         mock_component("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 = {}

tests/components/rfxtrx/test_init.py

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

tests/components/rfxtrx/test_light.py

@@ -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 = {}

tests/components/rfxtrx/test_sensor.py

@@ -18,8 +18,9 @@ class TestSensorRfxtrx(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 = {}

tests/components/rfxtrx/test_switch.py

@@ -18,8 +18,9 @@ class TestSwitchRfxtrx(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 = {}

tests/components/ring/test_init.py

@@ -23,8 +23,9 @@ class TestRing(unittest.TestCase):
         """Initialize values for this test case class."""
         self.hass = get_test_home_assistant()
         self.config = VALID_CONFIG
+        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()
 

tests/components/scene/test_init.py

@@ -35,8 +35,9 @@ class TestScene(unittest.TestCase):
 
         assert not self.light_1.is_on
         assert not self.light_2.is_on
+        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()
 

tests/components/script/test_init.py

@@ -73,8 +73,9 @@ class TestScriptComponent(unittest.TestCase):
         """Set up things to be run when tests are started."""
         self.hass = get_test_home_assistant()
 
-    # pylint: disable=invalid-name
-    def tearDown(self):
+        self.addCleanup(self.tear_down_cleanup)
+
+    def tear_down_cleanup(self):
         """Stop down everything that was started."""
         self.hass.stop()
 

tests/components/season/test_sensor.py

@@ -45,10 +45,7 @@ class TestSeason(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_season_should_be_summer_northern_astronomical(self):
         """Test that season should be summer."""

tests/components/shell_command/test_init.py

@@ -40,8 +40,9 @@ class TestShellCommand(unittest.TestCase):
         """
         self.hass = get_test_home_assistant()
         asyncio.get_child_watcher().attach_loop(self.hass.loop)
+        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()
 

tests/components/sigfox/test_sensor.py

@@ -39,10 +39,7 @@ class TestSigfoxSensor(unittest.TestCase):
     def setUp(self):
         """Initialize values for this testcase class."""
         self.hass = get_test_home_assistant()
-
-    def tearDown(self):
-        """Stop everything that was started."""
-        self.hass.stop()
+        self.addCleanup(self.hass.stop)
 
     def test_invalid_credentials(self):
         """Test for invalid credentials."""

tests/components/sleepiq/test_binary_sensor.py

@@ -28,8 +28,9 @@ class TestSleepIQBinarySensorSetup(unittest.TestCase):
         self.password = "bar"
         self.config = {"username": self.username, "password": self.password}
         self.DEVICES = []
+        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()
 

tests/components/sleepiq/test_init.py

@@ -37,8 +37,9 @@ class TestSleepIQ(unittest.TestCase):
         self.config = {
             "sleepiq": {"username": self.username, "password": self.password}
         }
+        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()
 

tests/components/sleepiq/test_sensor.py

@@ -28,8 +28,9 @@ class TestSleepIQSensorSetup(unittest.TestCase):
         self.password = "bar"
         self.config = {"username": self.username, "password": self.password}
         self.DEVICES = []
+        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()
 

tests/components/smtp/test_notify.py

@@ -34,8 +34,9 @@ class TestNotifySmtp(unittest.TestCase):
             "Home Assistant",
             0,
         )
+        self.addCleanup(self.tear_down_cleanup)
 
-    def tearDown(self):  # pylint: disable=invalid-name
+    def tear_down_cleanup(self):
         """Stop down everything that was started."""
         self.hass.stop()
 

tests/components/splunk/test_init.py

@@ -19,8 +19,9 @@ class TestSplunk(unittest.TestCase):
     def setUp(self):  # pylint: disable=invalid-name
         """Set up things to be run when tests are started."""
         self.hass = get_test_home_assistant()
+        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()
 

tests/components/statsd/test_init.py

@@ -19,8 +19,9 @@ class TestStatsd(unittest.TestCase):
     def setUp(self):  # pylint: disable=invalid-name
         """Set up things to be run when tests are started."""
         self.hass = get_test_home_assistant()
+        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()
 

tests/components/switch/test_init.py

@@ -22,11 +22,7 @@ class TestSwitch(unittest.TestCase):
         platform.init()
         # Switch 1 is ON, switch 2 is OFF
         self.switch_1, self.switch_2, self.switch_3 = platform.ENTITIES
-
-    # pylint: disable=invalid-name
-    def tearDown(self):
-        """Stop everything that was started."""
-        self.hass.stop()
+        self.addCleanup(self.hass.stop)
 
     def test_methods(self):
         """Test is_on, turn_on, turn_off methods."""

tests/components/time_date/test_sensor.py

@@ -23,8 +23,9 @@ class TestTimeDateSensor(unittest.TestCase):
         """Initialize values for this testcase class."""
         self.hass = get_test_home_assistant()
         self.DEFAULT_TIME_ZONE = dt_util.DEFAULT_TIME_ZONE
+        self.addCleanup(self.tear_down_cleanup)
 
-    def tearDown(self):
+    def tear_down_cleanup(self):
         """Stop everything that was started."""
         dt_util.set_default_time_zone(self.DEFAULT_TIME_ZONE)
         self.hass.stop()

tests/components/transport_nsw/test_sensor.py

@@ -37,10 +37,7 @@ class TestRMVtransportSensor(unittest.TestCase):
     def setUp(self):
         """Set up things to run when tests begin."""
         self.hass = get_test_home_assistant()
-
-    def tearDown(self):
-        """Stop everything that was started."""
-        self.hass.stop()
+        self.addCleanup(self.hass.stop)
 
     @patch("TransportNSW.TransportNSW.get_departures", side_effect=get_departuresMock)
     def test_transportnsw_config(self, mock_get_departures):

tests/components/uk_transport/test_sensor.py

@@ -49,10 +49,7 @@ class TestUkTransportSensor(unittest.TestCase):
         """Initialize values for this testcase class."""
         self.hass = get_test_home_assistant()
         self.config = VALID_CONFIG
-
-    def tearDown(self):
-        """Stop everything that was started."""
-        self.hass.stop()
+        self.addCleanup(self.hass.stop)
 
     @requests_mock.Mocker()
     def test_bus(self, mock_req):

tests/components/universal/test_media_player.py

@@ -220,8 +220,9 @@ class TestMediaPlayer(unittest.TestCase):
                 "shuffle": self.mock_shuffle_switch_id,
             },
         }
+        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()
 

tests/components/uptime/test_sensor.py

@@ -16,10 +16,7 @@ class TestUptimeSensor(unittest.TestCase):
     def setUp(self):
         """Set up things to run when tests begin."""
         self.hass = get_test_home_assistant()
-
-    def tearDown(self):
-        """Stop everything that was started."""
-        self.hass.stop()
+        self.addCleanup(self.hass.stop)
 
     def test_uptime_min_config(self):
         """Test minimum uptime configuration."""

tests/components/uvc/test_camera.py

@@ -21,10 +21,7 @@ class TestUVCSetup(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)
 
     @mock.patch("uvcclient.nvr.UVCRemote")
     @mock.patch.object(uvc, "UnifiVideoCamera")

tests/components/vultr/test_binary_sensor.py

@@ -42,8 +42,9 @@ class TestVultrBinarySensorSetup(unittest.TestCase):
             {CONF_SUBSCRIPTION: "123456", CONF_NAME: "Failed Server"},
             {CONF_SUBSCRIPTION: "555555", CONF_NAME: vultr.DEFAULT_NAME},
         ]
+        self.addCleanup(self.tear_down_cleanup)
 
-    def tearDown(self):
+    def tear_down_cleanup(self):
         """Stop our started services."""
         self.hass.stop()
 

tests/components/vultr/test_init.py

@@ -21,8 +21,9 @@ class TestVultr(unittest.TestCase):
         """Initialize values for this test case class."""
         self.hass = get_test_home_assistant()
         self.config = VALID_CONFIG
+        self.addCleanup(self.tear_down_cleanup)
 
-    def tearDown(self):  # pylint: disable=invalid-name
+    def tear_down_cleanup(self):
         """Stop everything that we started."""
         self.hass.stop()
 

tests/components/vultr/test_sensor.py

@@ -51,10 +51,7 @@ class TestVultrSensorSetup(unittest.TestCase):
                 CONF_MONITORED_CONDITIONS: ["pending_charges"],
             },
         ]
-
-    def tearDown(self):
-        """Stop everything that was started."""
-        self.hass.stop()
+        self.addCleanup(self.hass.stop)
 
     @requests_mock.Mocker()
     def test_sensor(self, mock):

tests/components/vultr/test_switch.py

@@ -42,8 +42,9 @@ class TestVultrSwitchSetup(unittest.TestCase):
             {CONF_SUBSCRIPTION: "123456", CONF_NAME: "Failed Server"},
             {CONF_SUBSCRIPTION: "555555", CONF_NAME: vultr.DEFAULT_NAME},
         ]
+        self.addCleanup(self.tear_down_cleanup)
 
-    def tearDown(self):
+    def tear_down_cleanup(self):
         """Stop our started services."""
         self.hass.stop()
 

tests/components/wake_on_lan/test_switch.py

@@ -35,10 +35,7 @@ class TestWolSwitch(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)
 
     @patch("wakeonlan.send_magic_packet", new=send_magic_packet)
     @patch("subprocess.call", new=call)

tests/components/weather/test_weather.py

@@ -33,8 +33,9 @@ class TestWeather(unittest.TestCase):
             self.hass, weather.DOMAIN, {"weather": {"platform": "demo"}}
         )
         self.hass.block_till_done()
+        self.addCleanup(self.tear_down_cleanup)
 
-    def tearDown(self):
+    def tear_down_cleanup(self):
         """Stop down everything that was started."""
         self.hass.stop()