You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
27 lines
526 B
Python
27 lines
526 B
Python
6 years ago
|
import Queue
|
||
|
|
import subprocess
|
||
|
|
import time
|
||
|
|
|
||
|
|
import unittest2
|
||
|
|
|
||
|
|
import testlib
|
||
|
|
import mitogen.master
|
||
|
|
import mitogen.utils
|
||
|
|
|
||
|
|
mitogen.utils.log_to_file()
|
||
|
|
|
||
|
|
class AddHandlerTest(unittest2.TestCase):
|
||
|
|
klass = mitogen.master.Router
|
||
|
|
|
||
|
|
def test_invoked_at_shutdown(self):
|
||
|
|
router = self.klass()
|
||
|
|
queue = Queue.Queue()
|
||
|
|
handle = router.add_handler(queue.put)
|
||
|
|
router.broker.shutdown()
|
||
|
|
self.assertEquals(queue.get(timeout=5), mitogen.core._DEAD)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
unittest2.main()
|
||
|
|
|