def test_get_log_level_code(self):
self.assertEquals(syslog.LOG_CRIT,
LogLevel.find_by_level_name('CRITICAL'))
self.assertEquals(syslog.LOG_WARNING,
LogLevel.find_by_level_name('WARNING'))
self.assertEquals(syslog.LOG_DEBUG,
LogLevel.find_by_level_name('DEBUG'))
self.assertEquals(syslog.LOG_INFO, LogLevel.find_by_level_name('INFO'))
self.assertEquals(syslog.LOG_ERR, LogLevel.find_by_level_name('ERROR'))
self.assertEquals(syslog.LOG_NOTICE,
LogLevel.find_by_level_name('NOTICE'))
self.assertIsNone(LogLevel.find_by_level_name('UNKNOWN'))
def replace_log_level(query):
log_level_regex = 'level\:\s*[a-zA-Z]+\s*'
match = re.search(log_level_regex, query)
if match:
log_level_name = match.group(0).split(':')[1].strip().upper()
log_level_code = LogLevel.find_by_level_name(log_level_name)
if not log_level_code:
message = "The given log level({}) is invalid. Use one of the following: {}"
cli_error(
message.format(log_level_name, LogLevel.list_levels()))
exit()
return re.sub(log_level_regex, 'level:%s ' % log_level_code, query)
else:
return query