From fc30855953f3dfd46a80e65e17283102085392e5 Mon Sep 17 00:00:00 2001 From: Dana Powers Date: Tue, 9 Jun 2015 20:25:13 -0700 Subject: Reset kafka.conn logger level in tearDown --- test/test_conn.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) (limited to 'test/test_conn.py') diff --git a/test/test_conn.py b/test/test_conn.py index 6e47cc8..2b70344 100644 --- a/test/test_conn.py +++ b/test/test_conn.py @@ -50,6 +50,11 @@ class ConnTest(unittest.TestCase): # Reset any mock counts caused by __init__ self.MockCreateConn.reset_mock() + def tearDown(self): + # Return connection logging to INFO + logging.getLogger('kafka.conn').setLevel(logging.INFO) + + def test_collect_hosts__happy_path(self): hosts = "localhost:1234,localhost" results = collect_hosts(hosts) @@ -172,6 +177,14 @@ class ConnTest(unittest.TestCase): class TestKafkaConnection(unittest.TestCase): + def setUp(self): + # kafka.conn debug logging is verbose, so only enable in conn tests + logging.getLogger('kafka.conn').setLevel(logging.DEBUG) + + def tearDown(self): + # Return connection logging to INFO + logging.getLogger('kafka.conn').setLevel(logging.INFO) + @mock.patch('socket.create_connection') def test_copy(self, socket): """KafkaConnection copies work as expected""" -- cgit v1.2.1