def testAddLogEntry(self): addLogEntry(TEST_DB, TEST_USER_ID, TEST_MSG, LOG_LVL_INFO) obj = list(db[LOG].find().sort("_id", pymongo.DESCENDING).limit(1))[0] self.assertEqual(obj[TEST_MSG_FIELD], TEST_MSG) self.assertEqual(obj[TEST_USER_ID_FIELD], TEST_USER_ID) self.assertEqual(obj[TEST_LEVEL_FIELD], TEST_LOG_LEVEL_INFO) self.assertTrue(TEST_SERVICE_FIELD not in obj) addLogEntry(TEST_DB_MASTER, TEST_USER_ID, TEST_MSG, TEST_LOG_LEVEL_CRITICAL, TEST_SERVICE) obj = list(db_master[LOG].find().sort("_id", pymongo.DESCENDING).limit(1))[0] self.assertEqual(obj[TEST_MSG_FIELD], TEST_MSG) self.assertTrue(TEST_SERVICE_FIELD in obj) self.assertEqual(obj[TEST_SERVICE_FIELD], TEST_SERVICE) self.assertEqual(obj[TEST_LEVEL_FIELD], TEST_LOG_LEVEL_CRITICAL)
def testAddLogEntry(self): addLogEntry(TEST_DB, TEST_USER_ID, TEST_MSG, LOG_LVL_INFO) obj = list(db[LOG].find().sort('_id', pymongo.DESCENDING).limit(1))[0] self.assertEqual(obj[TEST_MSG_FIELD], TEST_MSG) self.assertEqual(obj[TEST_USER_ID_FIELD], TEST_USER_ID) self.assertEqual(obj[TEST_LEVEL_FIELD], TEST_LOG_LEVEL_INFO) self.assertTrue(TEST_SERVICE_FIELD not in obj) addLogEntry(TEST_DB_MASTER, TEST_USER_ID, TEST_MSG, TEST_LOG_LEVEL_CRITICAL, TEST_SERVICE) obj = list(db_master[LOG].find().sort('_id', pymongo.DESCENDING).limit(1))[0] self.assertEqual(obj[TEST_MSG_FIELD], TEST_MSG) self.assertTrue(TEST_SERVICE_FIELD in obj) self.assertEqual(obj[TEST_SERVICE_FIELD], TEST_SERVICE) self.assertEqual(obj[TEST_LEVEL_FIELD], TEST_LOG_LEVEL_CRITICAL)
def testSearchLogForSubstr(self): addLogEntry(TEST_DB, TEST_USER_ID, TEST_MSG, LOG_LVL_INFO) result = list(getLog(TEST_DB, 1, 0, None, None, SUBSTRING))[0] self.assertEqual(result[TEST_MSG_FIELD], TEST_MSG)
def testFixForAddLogEntry(self): addLogEntry(TEST_DB, TEST_USER_ID, TEST_MSG, LOG_LVL_INFO) objects = list(db[LOG].find()) obj = objects[len(objects) - 1] self.assertEqual(type(obj[DATE]), datetime)
def writeServiceLog(serviceName, userId, message, level): addLogEntry(serviceName, userId, message, level)
def writeInstanceLog(userId, message, level, service='instance'): addLogEntry(getDbName(), userId, message, level, service)