source: sasview/src/sas/qtgui/Utilities/UnitTesting/SasviewLoggerTest.py @ 200ce9c

import sys
import unittest
import logging

from PyQt5.QtGui import *
from PyQt5.QtCore import *
from PyQt5.QtWidgets import *

# set up import paths
import sas.qtgui.path_prepare

# Local
from sas.qtgui.Utilities.SasviewLogger import QtHandler

if not QApplication.instance():
    app = QApplication(sys.argv)

class SasviewLoggerTest(unittest.TestCase):
    def setUp(self):
        """
        Prepare the logger
        """
        self.logger = logging.getLogger(__name__)
        self.handler = QtHandler()
        self.handler.setFormatter(logging.Formatter("%(levelname)s: %(message)s"))
        self.logger.addHandler(self.handler)
        self.logger.setLevel(logging.DEBUG)

        self.outHandlerGui=QTextBrowser()


    def testQtHandler(self):
        """
        Test redirection of all levels of logging
        """
        # Attach the listener
        self.handler.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')

        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)

if __name__ == "__main__":
    unittest.main()