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()