91 lines
2.2 KiB
Python
91 lines
2.2 KiB
Python
|
|
import logging
|
|
import mock
|
|
|
|
import unittest2
|
|
import testlib
|
|
import mitogen.core
|
|
import mitogen.master
|
|
import mitogen.parent
|
|
import mitogen.utils
|
|
from mitogen.core import b
|
|
|
|
|
|
def ping():
|
|
pass
|
|
|
|
|
|
class BufferingTest(testlib.TestCase):
|
|
klass = mitogen.core.LogHandler
|
|
|
|
def record(self):
|
|
return logging.LogRecord(
|
|
name='name',
|
|
level=99,
|
|
pathname='pathname',
|
|
lineno=123,
|
|
msg='msg',
|
|
args=(),
|
|
exc_info=None,
|
|
)
|
|
|
|
def build(self):
|
|
context = mock.Mock()
|
|
return context, self.klass(context)
|
|
|
|
def test_initially_buffered(self):
|
|
context, handler = self.build()
|
|
rec = self.record()
|
|
handler.emit(rec)
|
|
self.assertEquals(0, context.send.call_count)
|
|
self.assertEquals(1, len(handler._buffer))
|
|
|
|
def test_uncork(self):
|
|
context, handler = self.build()
|
|
rec = self.record()
|
|
handler.emit(rec)
|
|
handler.uncork()
|
|
|
|
self.assertEquals(1, context.send.call_count)
|
|
self.assertEquals(None, handler._buffer)
|
|
|
|
_, args, _ = context.send.mock_calls[0]
|
|
msg, = args
|
|
|
|
self.assertEquals(mitogen.core.FORWARD_LOG, msg.handle)
|
|
self.assertEquals(b('name\x0099\x00msg'), msg.data)
|
|
|
|
|
|
class StartupTest(testlib.RouterMixin, testlib.TestCase):
|
|
def test_earliest_messages_logged(self):
|
|
log = testlib.LogCapturer()
|
|
log.start()
|
|
|
|
c1 = self.router.local()
|
|
c1.shutdown(wait=True)
|
|
|
|
logs = log.stop()
|
|
self.assertTrue('Python version is' in logs)
|
|
self.assertTrue('Parent is context 0 (master)' in logs)
|
|
|
|
def test_earliest_messages_logged_via(self):
|
|
c1 = self.router.local(name='c1')
|
|
# ensure any c1-related msgs are processed before beginning capture.
|
|
c1.call(ping)
|
|
|
|
log = testlib.LogCapturer()
|
|
log.start()
|
|
|
|
c2 = self.router.local(via=c1, name='c2', debug=True)
|
|
c2.shutdown(wait=True)
|
|
|
|
logs = log.stop()
|
|
self.assertTrue('Python version is' in logs)
|
|
|
|
expect = 'Parent is context %s (%s)' % (c1.context_id, 'parent')
|
|
self.assertTrue(expect in logs)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest2.main()
|