def test_pp_dis_enabled(self, pformat): pformat.return_value = 'something pprinted' logger = mock.MagicMock() logger.isEnabledFor.return_value = False lg = getLogger() lg.pprint('object', logger=logger) self.assertFalse(logger._log.called)
def test_pp_enabled(self, pformat): pformat.return_value = 'something pprinted' logger = mock.MagicMock() logger.isEnabledFor.return_value = True lg = getLogger() lg.pprint('object', logger=logger) logger._log.assert_called_with(logging.DEBUG, '\nsomething pprinted', [], {})
def test_logger_before_queries(self, connection): connection.queries = [1, 2] logger = mock.MagicMock() logger.isEnabledFor.return_value = True lg = getLogger() lg.before_queries(txt='object', logger=logger) logger._log.assert_called_with(logging.DEBUG, 'object', [], {}) self.assertEqual(logger._num_queries, 2)
def test_logger_before_queries_notxt(self, connection): connection.queries = [1, 2] logger = mock.MagicMock() logger.isEnabledFor.return_value = True lg = getLogger() lg.before_queries(logger=logger) self.assertFalse(logger._log.called) self.assertEqual(logger._num_queries, 2)
def test_logger(self): func = mock.MagicMock(return_value=None, name='logger', spec=['func_code']) func_code = mock.MagicMock(name='func_code') func.func_code = func_code func_code.co_filename = 'file name' func_code.co_firstlineno = 23 func_code.co_name = 'name' logger = mock.MagicMock() lg = getLogger() lg.callable(func, logger=logger) logger._log.assert_called_with(logging.DEBUG, 'file name:23 name', [], {})
def test_logger_alert(self): logger = mock.MagicMock() lg = getLogger() lg.alert('object', marker='1', logger=logger) bannerEdge = '1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1' calls = [ mock.call(logging.DEBUG, bannerEdge, [], {}), mock.call(logging.DEBUG, bannerEdge, [], {}), mock.call(logging.DEBUG, ' object', [], {}), mock.call(logging.DEBUG, bannerEdge, [], {}), mock.call(logging.DEBUG, bannerEdge, [], {}), ] logger._log.assert_has_calls(calls)
def test_logger_after_queries_enabled(self, connection): logger = mock.MagicMock() logger._num_queries = 1 qry1 = {'name': 'qry1', 'time': '1', 'sql': 'select'} qry2 = {'name': 'qry2', 'time': '2', 'sql': 'insert'} qry3 = {'name': 'qry3', 'time': '3', 'sql': 'update'} connection.queries = [qry1, qry2, qry3] logger.isEnabledFor.return_value = True lg = getLogger() lg.after_queries(obj='object', logger=logger) calls = [ mock.call(logging.DEBUG, 'Database hits: 2', [], {}), mock.call(logging.DEBUG, 'Database hit, 2 seconds\ninsert', [], {}), mock.call(logging.DEBUG, 'Database hit, 3 seconds\nupdate', [], {}), ] logger._log.assert_has_calls(calls)
def test_logger_callable(self): with self.assertRaises(AssertionError): logger = getLogger() logger.callable('some var', logger=None)
def test_logger_before_disabled(self): logger = mock.MagicMock() logger.isEnabledFor.return_value = False lg = getLogger() lg.before_queries(txt='object', logger=logger) self.assertFalse(logger._log.called)