Skip to content

Commit 00329a3

Browse files
committed
integration-tests: fix undefiend logger variable
e6abdf1 seem to have broke some of the test teardown while trying to cleanup log handlers and failing like the following: ``` self = <tests.integration.standard.test_query.PreparedStatementArgTest testMethod=test_prepare_batch_statement> def tearDown(self): > logger.removeHandler(self.mock_handler) E NameError: name 'logger' is not defined tests/integration/standard/test_query.py:513: NameError ```
1 parent b9f295e commit 00329a3

File tree

2 files changed

+6
-6
lines changed

2 files changed

+6
-6
lines changed

tests/integration/standard/test_query.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -506,11 +506,11 @@ class PreparedStatementArgTest(unittest.TestCase):
506506

507507
def setUp(self):
508508
self.mock_handler = MockLoggingHandler()
509-
logger = logging.getLogger(cluster.__name__)
510-
logger.addHandler(self.mock_handler)
509+
self.logger = logging.getLogger(cluster.__name__)
510+
self.logger.addHandler(self.mock_handler)
511511

512512
def tearDown(self):
513-
logger.removeHandler(self.mock_handler)
513+
self.logger.removeHandler(self.mock_handler)
514514

515515
def test_prepare_on_all_hosts(self):
516516
"""

tests/integration/upgrade/__init__.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -76,12 +76,12 @@ class UpgradeBase(unittest.TestCase):
7676
@classmethod
7777
def setUpClass(cls):
7878
cls.logger_handler = MockLoggingHandler()
79-
logger = logging.getLogger(cluster.__name__)
80-
logger.addHandler(cls.logger_handler)
79+
cls.logger = logging.getLogger(cluster.__name__)
80+
cls.logger.addHandler(cls.logger_handler)
8181

8282
@classmethod
8383
def tearDownClass(cls):
84-
logger.removeHandler(cls.logger_handler)
84+
cls.logger.removeHandler(cls.logger_handler)
8585

8686
def _upgrade_step_setup(self):
8787
"""

0 commit comments

Comments
 (0)