X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/d17049c214dda2a37a928b6ac1a4a348c8d3c9e2..ee9d1e39b5d469a827be5a719c9c0860914ab2a8:/sdk/python/tests/test_api.py

diff --git a/sdk/python/tests/test_api.py b/sdk/python/tests/test_api.py
index 65d804ebb1..0f85e5520c 100644
--- a/sdk/python/tests/test_api.py
+++ b/sdk/python/tests/test_api.py
@@ -402,13 +402,11 @@ class ArvadosApiTest(run_test_server.TestCaseWithServers):
             pass
         real_logger = logging.getLogger('googleapiclient.http')
         mock_logger = mock.Mock(wraps=real_logger)
-        mock_logger.hasHandlers.return_value = False
+        mock_logger.handlers = logging.getLogger('googleapiclient').handlers
         mock_logger.level = logging.NOTSET
-        with (
-                mock.patch('logging.getLogger', return_value=mock_logger),
-                mock.patch('time.sleep'),
-                self.assertLogs(real_logger, 'INFO') as actual_logs,
-        ):
+        with mock.patch('logging.getLogger', return_value=mock_logger), \
+             mock.patch('time.sleep'), \
+             self.assertLogs(real_logger, 'INFO') as actual_logs:
             try:
                 api_client('v1', 'https://test.invalid/', 'NoToken', num_retries=1)
             except httplib2.error.ServerNotFoundError:
@@ -423,11 +421,9 @@ class ArvadosApiTest(run_test_server.TestCaseWithServers):
     def test_configured_logger_untouched(self):
         real_logger = logging.getLogger('googleapiclient.http')
         mock_logger = mock.Mock(wraps=real_logger)
-        mock_logger.hasHandlers.return_value = True
-        with (
-                mock.patch('logging.getLogger', return_value=mock_logger),
-                mock.patch('time.sleep'),
-        ):
+        mock_logger.handlers = logging.getLogger().handlers
+        with mock.patch('logging.getLogger', return_value=mock_logger), \
+             mock.patch('time.sleep'):
             try:
                 api_client('v1', 'https://test.invalid/', 'NoToken', num_retries=1)
             except httplib2.error.ServerNotFoundError: