import sys import unittest import logging from PyQt4.QtGui import * from PyQt4.QtCore import * # set up import paths import path_prepare # Local from SasviewLogger import XStream from SasviewLogger import QtHandler app = QApplication(sys.argv) class SasviewLoggerTest(unittest.TestCase): def setUp(self): """ Prepare the logger """ self.logger = logging.getLogger(__name__) handler = QtHandler() handler.setFormatter(logging.Formatter("%(levelname)s: %(message)s")) self.logger.addHandler(handler) self.logger.setLevel(logging.DEBUG) self.outHandlerGui=QTextBrowser() def testQtHandler(self): """ Test redirection of all levels of logging """ # Attach the listeners XStream.stderr().messageWritten.connect( self.outHandlerGui.insertPlainText ) XStream.stdout().messageWritten.connect( self.outHandlerGui.insertPlainText ) # Send the signals self.logger.debug('debug message') self.logger.info('info message') self.logger.warning('warning message') self.logger.error('error message') sys.stdout.write('with stdout') sys.stderr.write('with stderr') out=self.outHandlerGui.toPlainText() # Assure everything got logged self.assertIn('DEBUG: debug message', out) self.assertIn('INFO: info message', out) self.assertIn('WARNING: warning message', out) self.assertIn('ERROR: error message', out) self.assertIn('with stdout', out) self.assertIn('with stderr', out) if __name__ == "__main__": unittest.main()