Entry 4724

import logging

class LogMemoryHandler(logging.Handler):
    '''appends log traces into memory 
    for easy asserting in unittests'''
    def __init__(self):
        self.reset()
        logging.Handler.__init__(self)

    def emit(self,record):
        self.messages[record.levelname.lower()].append(record.getMessage())

    def reset(self):
        self.messages = {'debug':[],
                         'info':[],
                         'warning':[],
                         'error':[],
                         'critical':[]
                         }

    def contains(self,logtrace,level):
        for trace in self.messages[level]:
            if trace.find(logtrace) >= 0:
                return True
        return False