class TestBiiOutputStream(TestCase): def setUp(self): _, self.log_file = tempfile.mkstemp() self.outstring = StringIO() self.out = OutputStreamWrapper(stream=self.outstring, log_file_name=self.log_file) def test_write(self): self.out.write('hello') self.assertEquals('hello', self.outstring.getvalue()) with open(self.log_file) as log: regex = '.* INFO: hello\n' a = re.compile(regex) content = log.read() self.assertTrue(a.match(content), '%s does not match regex' % content) def test_log(self): self.out.log('debug') # Debug logs do not show on normal output self.out.log('info', INFO) self.out.log('warn', WARN) self.out.log('error', ERROR) self.assertEquals('INFO: info\nWARN: warn\nERROR: error\n', self.outstring.getvalue()) with open(self.log_file) as log: regex = '.* DEBUG: debug\n.* INFO: info\n.* WARNING: warn\n.* ERROR: error' a = re.compile(regex) content = log.read() print content self.assertTrue(a.match(content), '%s does not match regex' % content)
def setUp(self): _, self.log_file = tempfile.mkstemp() self.outstring = StringIO() self.out = OutputStreamWrapper(stream=self.outstring, log_file_name=self.log_file)