From fec4966ee074ba558784d0d88035185e2bd84a55 Mon Sep 17 00:00:00 2001 From: Felipe Martins Diel <41558831+felipediel@users.noreply.github.com> Date: Wed, 19 Aug 2020 18:38:34 -0300 Subject: [PATCH] Clean up feedreader files at the end (#39064) --- tests/components/feedreader/test_init.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/tests/components/feedreader/test_init.py b/tests/components/feedreader/test_init.py index 1983cb5ab8d..b50ef5f6619 100644 --- a/tests/components/feedreader/test_init.py +++ b/tests/components/feedreader/test_init.py @@ -38,11 +38,14 @@ class TestFeedreaderComponent(unittest.TestCase): def setUp(self): """Initialize values for this testcase class.""" self.hass = get_test_home_assistant() - # Delete any previously stored data + self.addCleanup(self.tear_down_cleanup) + + def tear_down_cleanup(self): + """Clean up files and stop Home Assistant.""" data_file = self.hass.config.path(f"{feedreader.DOMAIN}.pickle") if exists(data_file): remove(data_file) - self.addCleanup(self.hass.stop) + self.hass.stop() def test_setup_one_feed(self): """Test the general setup of this component."""