78 lines
1.9 KiB
Python
78 lines
1.9 KiB
Python
|
|
import os
|
|
import signal
|
|
import sys
|
|
import tempfile
|
|
import threading
|
|
import time
|
|
|
|
import unittest2
|
|
import testlib
|
|
|
|
import mitogen.core
|
|
import mitogen.parent
|
|
|
|
|
|
class ConnectionTest(testlib.RouterMixin, testlib.TestCase):
|
|
def test_broker_shutdown_while_connect_in_progress(self):
|
|
# if Broker.shutdown() is called while a connection attempt is in
|
|
# progress, the connection should be torn down.
|
|
|
|
path = tempfile.mktemp(prefix='broker_shutdown_sem_')
|
|
open(path, 'wb').close()
|
|
|
|
os.environ['BROKER_SHUTDOWN_SEMAPHORE'] = path
|
|
result = []
|
|
|
|
def thread():
|
|
python_path = testlib.data_path('broker_shutdown_test_python.py')
|
|
try:
|
|
result.append(self.router.local(python_path=python_path))
|
|
except Exception:
|
|
result.append(sys.exc_info()[1])
|
|
|
|
th = threading.Thread(target=thread)
|
|
th.start()
|
|
|
|
while os.path.exists(path):
|
|
time.sleep(0.05)
|
|
|
|
self.broker.shutdown()
|
|
th.join()
|
|
|
|
exc, = result
|
|
self.assertTrue(isinstance(exc, mitogen.parent.CancelledError))
|
|
self.assertEquals(mitogen.parent.BROKER_SHUTDOWN_MSG, exc.args[0])
|
|
|
|
|
|
@mitogen.core.takes_econtext
|
|
def do_detach(econtext):
|
|
econtext.detach()
|
|
while 1:
|
|
time.sleep(1)
|
|
logging.getLogger('mitogen').error('hi')
|
|
|
|
|
|
class DetachReapTest(testlib.RouterMixin, testlib.TestCase):
|
|
def test_subprocess_preserved_on_shutdown(self):
|
|
c1 = self.router.local()
|
|
pid = c1.call(os.getpid)
|
|
|
|
l = mitogen.core.Latch()
|
|
mitogen.core.listen(c1, 'disconnect', l.put)
|
|
c1.call_no_reply(do_detach)
|
|
l.get()
|
|
|
|
self.broker.shutdown()
|
|
self.broker.join()
|
|
|
|
os.kill(pid, 0) # succeeds if process still alive
|
|
|
|
# now clean up
|
|
os.kill(pid, signal.SIGTERM)
|
|
os.waitpid(pid, 0)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest2.main()
|