def test_connect(self):
handler = MongoHandler(host='localhost',
database_name=self.database_name,
collection=self.collection_name)
self.assertTrue(isinstance(handler, MongoHandler))
self.handler.connection.drop_database(self.database_name)
handler.close()
def test_connect_failed_silent(self):
handler = MongoHandler(host='unknow_host',
database_name=self.database_name,
collection=self.collection_name,
fail_silently=True)
self.assertTrue(isinstance(handler, MongoHandler))
self.handler.connection.drop_database(self.database_name)
handler.close()
def test_connect_failed_silent(self):
log4mongo.handlers._connection = None
kwargs = {'connectTimeoutMS': 2000, 'serverselectiontimeoutms': 2000}
handler = MongoHandler(host='unknow_host',
database_name=self.database_name,
collection=self.collection_name,
fail_silently=True,
**kwargs)
self.assertTrue(isinstance(handler, MongoHandler))
self.handler.connection.drop_database(self.database_name)
handler.close()
class TestMongoHandler(unittest.TestCase):
host_name = 'localhost'
database_name = 'log4mongo_test'
collection_name = 'logs_test'
def setUp(self):
self.handler = MongoHandler(
host=self.host_name,
database_name=self.database_name,
collection=self.collection_name,
)
self.log = logging.getLogger('testLogger')
self.log.setLevel(logging.DEBUG)
self.log.addHandler(self.handler)
self.old_stderr = sys.stdout
sys.stderr = StringIO()
def tearDown(self):
self.handler.connection.drop_database(self.database_name)
self.handler.close()
self.log.removeHandler(self.handler)
self.log = None
self.handler = None
sys.stderr.close()
sys.stderr = self.old_stderr
def test_connect(self):
handler = MongoHandler(host='localhost',
database_name=self.database_name,
collection=self.collection_name)
self.assertTrue(isinstance(handler, MongoHandler))
self.handler.connection.drop_database(self.database_name)
handler.close()
def test_1_connect_failed(self):
log4mongo.handlers._connection = None
kwargs = {'connectTimeoutMS': 2000, 'serverselectiontimeoutms': 2000}
if pymongo.version_tuple[0] < 3:
with self.assertRaises(PyMongoError):
MongoHandler(host='unknow_host',
database_name=self.database_name,
collection=self.collection_name,
**kwargs)
else:
with self.assertRaises(ServerSelectionTimeoutError):
MongoHandler(host='unknow_host',
database_name=self.database_name,
collection=self.collection_name,
**kwargs)
def test_connect_failed_silent(self):
log4mongo.handlers._connection = None
kwargs = {'connectTimeoutMS': 2000, 'serverselectiontimeoutms': 2000}
handler = MongoHandler(host='unknow_host',
database_name=self.database_name,
collection=self.collection_name,
fail_silently=True,
**kwargs)
self.assertTrue(isinstance(handler, MongoHandler))
self.handler.connection.drop_database(self.database_name)
handler.close()
def test_emit(self):
self.log.warning('test message')
document = self.handler.collection.find_one({
'message': 'test message',
'level': 'WARNING'
})
self.assertEqual(document['message'], 'test message')
self.assertEqual(document['level'], 'WARNING')
def test_emit_exception(self):
try:
raise Exception('exc1')
except:
self.log.exception('test message')
document = self.handler.collection.find_one({
'message': 'test message',
'level': 'ERROR'
})
self.assertEqual(document['message'], 'test message')
self.assertEqual(document['level'], 'ERROR')
self.assertEqual(document['exception']['message'], 'exc1')
def test_emit_fail(self):
self.handler.collection = ''
self.log.warning('test warning')
val = sys.stderr.getvalue()
self.assertRegexpMatches(
val, r"AttributeError: 'str' object has no attribute '{}'".format(
write_method))
def test_email_fail_silent(self):
self.handler.fail_silently = True
self.handler.collection = ''
self.log.warning('test warming')
self.assertEqual(sys.stderr.getvalue(), '')
def test_contextual_info(self):
self.log.info('test message with contextual info',
extra={
'ip': '127.0.0.1',
'host': 'localhost'
})
document = self.handler.collection.find_one({
'message': 'test message with contextual info',
'level': 'INFO'
})
self.assertEqual(document['message'],
'test message with contextual info')
self.assertEqual(document['level'], 'INFO')
self.assertEqual(document['ip'], '127.0.0.1')
self.assertEqual(document['host'], 'localhost')
def test_contextual_info_adapter(self):
adapter = logging.LoggerAdapter(self.log, {
'ip': '127.0.0.1',
'host': 'localhost'
})
adapter.info('test message with contextual info')
document = self.handler.collection.find_one({
'message': 'test message with contextual info',
'level': 'INFO'
})
self.assertEqual(document['message'],
'test message with contextual info')
self.assertEqual(document['level'], 'INFO')
self.assertEqual(document['ip'], '127.0.0.1')
self.assertEqual(document['host'], 'localhost')
class TestMongoHandler(unittest.TestCase):
host_name = 'localhost'
database_name = 'log4mongo_test'
collection_name = 'logs_test'
def setUp(self):
self.handler = MongoHandler(host=self.host_name,
database_name=self.database_name,
collection=self.collection_name)
self.log = logging.getLogger('testLogger')
self.log.setLevel(logging.DEBUG)
self.log.addHandler(self.handler)
self.old_stderr = sys.stdout
sys.stderr = StringIO()
def tearDown(self):
self.handler.connection.drop_database(self.database_name)
self.handler.close()
self.log.removeHandler(self.handler)
self.log = None
self.handler = None
sys.stderr.close()
sys.stderr = self.old_stderr
def test_connect(self):
handler = MongoHandler(host='localhost',
database_name=self.database_name,
collection=self.collection_name)
self.assertTrue(isinstance(handler, MongoHandler))
self.handler.connection.drop_database(self.database_name)
handler.close()
def test_connect_failed(self):
with self.assertRaises(AutoReconnect):
MongoHandler(host='unknow_host',
database_name=self.database_name,
collection=self.collection_name)
def test_connect_failed_silent(self):
handler = MongoHandler(host='unknow_host',
database_name=self.database_name,
collection=self.collection_name,
fail_silently=True)
self.assertTrue(isinstance(handler, MongoHandler))
self.handler.connection.drop_database(self.database_name)
handler.close()
def test_emit(self):
self.log.warning('test message')
document = self.handler.collection.find_one({
'message': 'test message',
'level': 'WARNING'
})
self.assertEqual(document['message'], 'test message')
self.assertEqual(document['level'], 'WARNING')
def test_emit_exception(self):
try:
raise Exception('exc1')
except:
self.log.exception('test message')
document = self.handler.collection.find_one({
'message': 'test message',
'level': 'ERROR'
})
self.assertEqual(document['message'], 'test message')
self.assertEqual(document['level'], 'ERROR')
self.assertEqual(document['exception']['message'], 'exc1')
def test_emit_fail(self):
self.handler.collection = ''
self.log.warn('test warning')
self.assertRegexpMatches(
sys.stderr.getvalue(),
r"AttributeError: 'str' object has no attribute 'save'")
def test_email_fail_silent(self):
self.handler.fail_silently = True
self.handler.collection = ''
self.log.warn('test warming')
self.assertEqual(sys.stderr.getvalue(), '')
def test_contextual_info(self):
self.log.info('test message with contextual info',
extra={
'ip': '127.0.0.1',
'host': 'localhost'
})
document = self.handler.collection.find_one({
'message': 'test message with contextual info',
'level': 'INFO'
})
self.assertEqual(document['message'],
'test message with contextual info')
self.assertEqual(document['level'], 'INFO')
self.assertEqual(document['ip'], '127.0.0.1')
self.assertEqual(document['host'], 'localhost')
def test_contextual_info_adapter(self):
adapter = logging.LoggerAdapter(self.log, {
'ip': '127.0.0.1',
'host': 'localhost'
})
adapter.info('test message with contextual info')
document = self.handler.collection.find_one({
'message': 'test message with contextual info',
'level': 'INFO'
})
self.assertEqual(document['message'],
'test message with contextual info')
self.assertEqual(document['level'], 'INFO')
self.assertEqual(document['ip'], '127.0.0.1')
self.assertEqual(document['host'], 'localhost')
