11import logging
22
33import pytest
4- from mock import call , patch , ANY
4+ from mock import call , patch
55from rethinkdb .logger import DriverLogger
66
77
88@pytest .mark .unit
99class TestDriverLogger (object ):
1010 driver_logger = DriverLogger (logging .DEBUG )
11- logger = logging .getLogger ()
11+ logger = logging .getLogger ("rethinkdb.logger" )
1212
1313 def test_converter (self ):
1414 expected_message = 'converted message'
@@ -53,31 +53,31 @@ def test_log_debug(self):
5353 with patch .object (self .logger , 'log' ) as mock_log :
5454 self .driver_logger .debug (expected_message )
5555
56- mock_log .assert_called_once_with (logging .DEBUG , expected_message , ANY , ANY )
56+ mock_log .assert_called_once_with (logging .DEBUG , expected_message )
5757
5858 def test_log_info (self ):
5959 expected_message = 'info message'
6060
6161 with patch .object (self .logger , 'log' ) as mock_log :
6262 self .driver_logger .info (expected_message )
6363
64- mock_log .assert_called_once_with (logging .INFO , expected_message , ANY , ANY )
64+ mock_log .assert_called_once_with (logging .INFO , expected_message )
6565
6666 def test_log_warning (self ):
6767 expected_message = 'warning message'
6868
6969 with patch .object (self .logger , 'log' ) as mock_log :
7070 self .driver_logger .warning (expected_message )
7171
72- mock_log .assert_called_once_with (logging .WARNING , expected_message , ANY , ANY )
72+ mock_log .assert_called_once_with (logging .WARNING , expected_message )
7373
7474 def test_log_error (self ):
7575 expected_message = 'error message'
7676
7777 with patch .object (self .logger , 'log' ) as mock_log :
7878 self .driver_logger .error (expected_message )
7979
80- mock_log .assert_called_once_with (logging .ERROR , expected_message , ANY , ANY )
80+ mock_log .assert_called_once_with (logging .ERROR , expected_message )
8181
8282 @patch ('rethinkdb.logger.DriverLogger._convert_message' )
8383 def test_log_exception (self , mock_converter ):
@@ -92,7 +92,7 @@ def test_log_exception(self, mock_converter):
9292 self .driver_logger .exception (exc )
9393
9494 mock_converter .assert_called_once_with (expected_exception )
95- mock_log .assert_called_once_with (logging .ERROR , expected_message , ANY , { ' exc_info' : 1 } )
95+ mock_log .assert_called_once_with (logging .ERROR , expected_message , exc_info = 1 )
9696
9797 @patch ('rethinkdb.logger.DriverLogger._convert_message' )
9898 def test_log_exception_and_raise (self , mock_converter ):
@@ -105,4 +105,4 @@ def test_log_exception_and_raise(self, mock_converter):
105105 self .driver_logger .exception (expected_exception , with_raise = True )
106106
107107 mock_converter .assert_called_once_with (expected_exception )
108- mock_log .assert_called_once_with (logging .ERROR , expected_message , ANY , { ' exc_info' : 1 } )
108+ mock_log .assert_called_once_with (logging .ERROR , expected_message , exc_info = 1 )
0 commit comments