diff options
Diffstat (limited to 'tests/unit/test_log.py')
-rw-r--r-- | tests/unit/test_log.py | 7 |
1 files changed, 2 insertions, 5 deletions
diff --git a/tests/unit/test_log.py b/tests/unit/test_log.py index 451fb4f..3ea5943 100644 --- a/tests/unit/test_log.py +++ b/tests/unit/test_log.py @@ -201,13 +201,10 @@ class LegacyFormatterTestCase(test_utils.BaseTestCase): self.handler = logging.StreamHandler(self.stream) self.handler.setFormatter(log.LegacyFormatter()) self.log.logger.addHandler(self.handler) + self.addCleanup(self.log.logger.removeHandler, self.handler) self.level = self.log.logger.getEffectiveLevel() self.log.logger.setLevel(logging.DEBUG) - - def tearDown(self): - self.log.logger.setLevel(self.level) - self.log.logger.removeHandler(self.handler) - super(LegacyFormatterTestCase, self).tearDown() + self.addCleanup(self.log.logger.setLevel, self.level) def test_uncontextualized_log(self): self.log.info("foo") |