117 lines
3.8 KiB
Python
117 lines
3.8 KiB
Python
import unittest2
|
|
|
|
import mitogen.core
|
|
import mitogen.service
|
|
import testlib
|
|
|
|
|
|
class MyService(mitogen.service.Service):
|
|
def __init__(self, router):
|
|
super(MyService, self).__init__(router)
|
|
self._counter = 0
|
|
|
|
@mitogen.service.expose(policy=mitogen.service.AllowParents())
|
|
def get_id(self):
|
|
self._counter += 1
|
|
return self._counter, id(self)
|
|
|
|
@mitogen.service.expose(policy=mitogen.service.AllowParents())
|
|
def privileged_op(self):
|
|
return 'privileged!'
|
|
|
|
@mitogen.service.expose(policy=mitogen.service.AllowAny())
|
|
def unprivileged_op(self):
|
|
return 'unprivileged!'
|
|
|
|
|
|
|
|
class MyService2(MyService):
|
|
"""
|
|
A uniquely named service that lets us test framework activation and class
|
|
activation separately.
|
|
"""
|
|
|
|
|
|
def call_service_in(context, service_name, method_name):
|
|
return context.call_service(service_name, method_name)
|
|
|
|
|
|
class ActivationTest(testlib.RouterMixin, testlib.TestCase):
|
|
def test_parent_can_activate(self):
|
|
l1 = self.router.fork()
|
|
counter, id_ = l1.call_service(MyService, 'get_id')
|
|
self.assertEquals(1, counter)
|
|
self.assertTrue(isinstance(id_, int))
|
|
|
|
def test_sibling_cannot_activate_framework(self):
|
|
l1 = self.router.fork()
|
|
l2 = self.router.fork()
|
|
exc = self.assertRaises(mitogen.core.CallError,
|
|
lambda: l2.call(call_service_in, l1, MyService2.name(), 'get_id'))
|
|
self.assertTrue(mitogen.core.Router.refused_msg in exc.args[0])
|
|
|
|
def test_sibling_cannot_activate_service(self):
|
|
l1 = self.router.fork()
|
|
l2 = self.router.fork()
|
|
l1.call_service(MyService, 'get_id') # force framework activation
|
|
capture = testlib.LogCapturer()
|
|
capture.start()
|
|
try:
|
|
exc = self.assertRaises(mitogen.core.CallError,
|
|
lambda: l2.call(call_service_in, l1, MyService2.name(), 'get_id'))
|
|
finally:
|
|
capture.stop()
|
|
msg = mitogen.service.Activator.not_active_msg % (MyService2.name(),)
|
|
self.assertTrue(msg in exc.args[0])
|
|
|
|
def test_activates_only_once(self):
|
|
l1 = self.router.fork()
|
|
counter, id_ = l1.call_service(MyService, 'get_id')
|
|
counter2, id_2 = l1.call_service(MyService, 'get_id')
|
|
self.assertEquals(1, counter)
|
|
self.assertEquals(2, counter2)
|
|
self.assertEquals(id_, id_2)
|
|
|
|
|
|
class PermissionTest(testlib.RouterMixin, testlib.TestCase):
|
|
def test_sibling_unprivileged_ok(self):
|
|
l1 = self.router.fork()
|
|
l1.call_service(MyService, 'get_id')
|
|
l2 = self.router.fork()
|
|
self.assertEquals('unprivileged!',
|
|
l2.call(call_service_in, l1, MyService.name(), 'unprivileged_op'))
|
|
|
|
def test_sibling_privileged_bad(self):
|
|
l1 = self.router.fork()
|
|
l1.call_service(MyService, 'get_id')
|
|
l2 = self.router.fork()
|
|
capture = testlib.LogCapturer()
|
|
capture.start()
|
|
try:
|
|
exc = self.assertRaises(mitogen.core.CallError, lambda:
|
|
l2.call(call_service_in, l1, MyService.name(), 'privileged_op'))
|
|
finally:
|
|
capture.stop()
|
|
msg = mitogen.service.Invoker.unauthorized_msg % (
|
|
u'privileged_op',
|
|
MyService.name(),
|
|
)
|
|
self.assertTrue(msg in exc.args[0])
|
|
|
|
|
|
class CloseTest(testlib.RouterMixin, testlib.TestCase):
|
|
klass = mitogen.service.Pool
|
|
|
|
def test_receiver_closed(self):
|
|
pool = self.klass(router=self.router, services=[])
|
|
pool.stop()
|
|
self.assertEquals(None, pool._receiver.handle)
|
|
|
|
e = self.assertRaises(mitogen.core.ChannelError,
|
|
lambda: self.router.myself().call_service(MyService, 'foobar'))
|
|
self.assertEquals(e.args[0], self.router.invalid_handle_msg)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest2.main()
|