forked from LBRYCommunity/lbry-sdk
moved test_customLogger.py
This commit is contained in:
parent
a937aff80f
commit
855fd8bf9a
1 changed files with 13 additions and 13 deletions
|
@ -1,16 +1,16 @@
|
|||
from io import StringIO
|
||||
import StringIO
|
||||
import logging
|
||||
|
||||
import mock
|
||||
from unittest import skipIf
|
||||
import unittest
|
||||
from twisted.internet import defer
|
||||
from twisted.trial import unittest
|
||||
from twisted import trial
|
||||
|
||||
from lbrynet.core import log_support
|
||||
from tests.util import is_android
|
||||
from lbrynet import custom_logger
|
||||
from lbrynet.tests.util import is_android
|
||||
|
||||
|
||||
class TestLogger(unittest.TestCase):
|
||||
class TestLogger(trial.unittest.TestCase):
|
||||
def raiseError(self):
|
||||
raise Exception('terrible things happened')
|
||||
|
||||
|
@ -22,27 +22,27 @@ class TestLogger(unittest.TestCase):
|
|||
return d
|
||||
|
||||
def setUp(self):
|
||||
self.log = log_support.Logger('test')
|
||||
self.stream = StringIO()
|
||||
self.log = custom_logger.Logger('test')
|
||||
self.stream = StringIO.StringIO()
|
||||
handler = logging.StreamHandler(self.stream)
|
||||
handler.setFormatter(logging.Formatter("%(filename)s:%(lineno)d - %(message)s"))
|
||||
self.log.addHandler(handler)
|
||||
|
||||
@skipIf(is_android(),
|
||||
'Test cannot pass on Android because the tests package is compiled '
|
||||
'which results in a different method call stack')
|
||||
@unittest.skipIf(is_android(),
|
||||
'Test cannot pass on Android because the tests package is compiled '
|
||||
'which results in a different method call stack')
|
||||
def test_can_log_failure(self):
|
||||
def output_lines():
|
||||
return self.stream.getvalue().split('\n')
|
||||
|
||||
# the line number could change if this file gets refactored
|
||||
expected_first_line = 'test_log_support.py:20 - My message: terrible things happened'
|
||||
expected_first_line = 'test_customLogger.py:20 - My message: terrible things happened'
|
||||
|
||||
# testing the entirety of the message is futile as the
|
||||
# traceback will depend on the system the test is being run on
|
||||
# but hopefully these two tests are good enough
|
||||
d = self.triggerErrback()
|
||||
d.addCallback(lambda _: self.assertEqual(expected_first_line, output_lines()[0]))
|
||||
d.addCallback(lambda _: self.assertEquals(expected_first_line, output_lines()[0]))
|
||||
d.addCallback(lambda _: self.assertEqual(10, len(output_lines())))
|
||||
return d
|
||||
|
Loading…
Reference in a new issue