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.
mitogen/tests/broker_test.py

68 lines
1.7 KiB
Python

try:
from unittest import mock
except ImportError:
import mock
import testlib
import mitogen.core
class ShutdownTest(testlib.TestCase):
klass = mitogen.core.Broker
def test_poller_closed(self):
broker = self.klass()
actual_close = broker.poller.close
broker.poller.close = mock.Mock()
broker.shutdown()
broker.join()
self.assertEqual(1, len(broker.poller.close.mock_calls))
actual_close()
class DeferTest(testlib.TestCase):
klass = mitogen.core.Broker
def test_defer(self):
latch = mitogen.core.Latch()
broker = self.klass()
try:
broker.defer(lambda: latch.put(123))
self.assertEqual(123, latch.get())
finally:
broker.shutdown()
broker.join()
def test_defer_after_shutdown(self):
latch = mitogen.core.Latch()
broker = self.klass()
broker.shutdown()
broker.join()
e = self.assertRaises(mitogen.core.Error,
lambda: broker.defer(lambda: latch.put(123)))
self.assertEqual(e.args[0], mitogen.core.Waker.broker_shutdown_msg)
class DeferSyncTest(testlib.TestCase):
klass = mitogen.core.Broker
def test_okay(self):
broker = self.klass()
try:
th = broker.defer_sync(lambda: mitogen.core.threading__current_thread())
self.assertEqual(th, broker._thread)
finally:
broker.shutdown()
broker.join()
def test_exception(self):
broker = self.klass()
try:
self.assertRaises(ValueError,
broker.defer_sync, lambda: int('dave'))
finally:
broker.shutdown()
broker.join()