def checkMessage(self, msg, **kwargs):
"""
Ensures that the supplied msg is formatted as a onStatus invoke.
@param msg: The L{message.IMessage} to check.
@param kwargs: Optional parameters to check. Valid keys are 'command',
'level, 'code', 'description' and 'extra'.
"""
sentinel = object()
def check(a, b):
c = kwargs.pop(b, sentinel)
if c is sentinel:
return
self.assertEqual(a, c)
self.assertTrue(message.typeByClass(msg), message.Invoke)
self.assertEqual(msg.name, 'onStatus')
self.assertEqual(msg.id, 0)
cmd, s = msg.argv
check(cmd, 'command')
check(s.level, 'level')
check(s.code, 'code')
check(s.description, 'description')
check(s.getExtraContext(), 'extra')
# sanity check
self.assertEqual(kwargs, {}, msg='improper use of checkMessage')
def checkMessage(self, msg, **kwargs):
"""
Ensures that the supplied msg is formatted as a onStatus invoke.
@param msg: The L{message.IMessage} to check.
@param kwargs: Optional parameters to check. Valid keys are 'command',
'level, 'code', 'description' and 'extra'.
"""
sentinel = object()
def check(a, b):
c = kwargs.pop(b, sentinel)
if c is sentinel:
return
self.assertEqual(a, c)
self.assertTrue(message.typeByClass(msg), message.Invoke)
self.assertEqual(msg.name, 'onStatus')
self.assertEqual(msg.id, 0)
cmd, s = msg.argv
check(cmd, 'command')
check(s.level, 'level')
check(s.code, 'code')
check(s.description, 'description')
check(s.getExtraContext(), 'extra')
# sanity check
self.assertEqual(kwargs, {}, msg='improper use of checkMessage')
def test_command_callback(self):
"""
Test a command result and the message that is generated.
"""
ret = yield self.makeCall('command_result')
self.assertEqual(ret, 'foo')
m = self.messages
self.assertEqual(len(m), 1)
msg = self.messages.pop()
self.assertTrue(message.typeByClass(msg), message.Invoke)
self.assertEqual(msg.name, '_result')
self.assertEqual(msg.argv, [{'one': 'two'}, 'foo'])
self.assertEqual(msg.id, 2)
def test_successful_call(self):
"""
Test a successful RPC call.
"""
ret = yield self.makeCall('known_return')
self.assertEqual(ret, 'foo')
m = self.messages
self.assertEqual(len(m), 1)
msg = self.messages.pop()
self.assertTrue(message.typeByClass(msg), message.Invoke)
self.assertEqual(msg.name, '_result')
self.assertEqual(msg.argv, [None, 'foo'])
self.assertEqual(msg.id, 2)
def test_successful_call(self):
"""
Test a successful RPC call.
"""
ret = yield self.makeCall('known_return')
self.assertEqual(ret, 'foo')
m = self.messages
self.assertEqual(len(m), 1)
msg = self.messages.pop()
self.assertTrue(message.typeByClass(msg), message.Invoke)
self.assertEqual(msg.name, '_result')
self.assertEqual(msg.argv, [None, 'foo'])
self.assertEqual(msg.id, 1)
def test_command_callback(self):
"""
Test a command result and the message that is generated.
"""
ret = yield self.makeCall('command_result')
self.assertEqual(ret, 'foo')
m = self.messages
self.assertEqual(len(m), 1)
msg = self.messages.pop()
self.assertTrue(message.typeByClass(msg), message.Invoke)
self.assertEqual(msg.name, '_result')
self.assertEqual(msg.argv, [{'one': 'two'}, 'foo'])
self.assertEqual(msg.id, 1)
def test_call_command(self):
"""
Ensure L{execute} accepts a C{command} kwarg and that it is
set on the message appropriately.
"""
cmd = {'foo': 'bar'}
i, m = self.invoker, self.messages
ret = i.call('remote_method', command=cmd)
self.assertEqual(ret, None)
self.assertEqual(len(m), 1)
msg = m.pop()
self.assertEqual(message.typeByClass(msg), message.INVOKE)
self.assertEqual(msg.id, 0)
self.assertEqual(msg.name, 'remote_method')
self.assertEqual(msg.argv, [cmd])
def test_call_message(self):
"""
Check the context of the message sent when L{callWithoutResult} is
executed.
"""
i = self.invoker
m = self.messages
ret = i.call('remote_method', 1, 2, 3, 'foo')
self.assertEqual(ret, None)
self.assertEqual(len(m), 1)
msg = m.pop()
self.assertEqual(message.typeByClass(msg), message.INVOKE)
self.assertEqual(msg.id, 0)
self.assertEqual(msg.name, 'remote_method')
self.assertEqual(msg.argv, [None, 1, 2, 3, 'foo'])
def test_call_command(self):
"""
Ensure L{execute} accepts a C{command} kwarg and that it is
set on the message appropriately.
"""
cmd = {'foo': 'bar'}
i, m = self.invoker, self.messages
ret = i.call('remote_method', command=cmd)
self.assertEqual(ret, None)
self.assertEqual(len(m), 1)
msg = m.pop()
self.assertEqual(message.typeByClass(msg), message.INVOKE)
self.assertEqual(msg.id, 0)
self.assertEqual(msg.name, 'remote_method')
self.assertEqual(msg.argv, [cmd])
def test_call_message(self):
"""
Check the context of the message sent when L{callWithoutResult} is
executed.
"""
i = self.invoker
m = self.messages
ret = i.call('remote_method', 1, 2, 3, 'foo')
self.assertEqual(ret, None)
self.assertEqual(len(m), 1)
msg = m.pop()
self.assertEqual(message.typeByClass(msg), message.INVOKE)
self.assertEqual(msg.id, 0)
self.assertEqual(msg.name, 'remote_method')
self.assertEqual(msg.argv, [None, 1, 2, 3, 'foo'])
def test_command(self):
"""
Ensure L{call} accepts a C{command} kwarg and that it
is set on the sent message appropriately.
"""
cmd = {'foo': 'bar'}
i, m = self.invoker, self.messages
d = i.call('remote_method', command=cmd, notify=True)
self.assertTrue(isinstance(d, defer.Deferred))
self.assertEqual(len(m), 1)
msg = m.pop()
self.assertEqual(message.typeByClass(msg), message.INVOKE)
self.assertEqual(msg.id, 1)
self.assertEqual(msg.name, 'remote_method')
self.assertEqual(msg.argv, [cmd])
callContext = i.getCallContext(msg.id)
self.assertEqual(callContext, (d, 'remote_method', (), cmd))
def test_command(self):
"""
Ensure L{call} accepts a C{command} kwarg and that it
is set on the sent message appropriately.
"""
cmd = {'foo': 'bar'}
i, m = self.invoker, self.messages
d = i.call('remote_method', command=cmd, notify=True)
self.assertTrue(isinstance(d, defer.Deferred))
self.assertEqual(len(m), 1)
msg = m.pop()
self.assertEqual(message.typeByClass(msg), message.INVOKE)
self.assertEqual(msg.id, 2)
self.assertEqual(msg.name, 'remote_method')
self.assertEqual(msg.argv, [cmd])
callContext = i.getCallContext(msg.id)
self.assertEqual(callContext,
(d, 'remote_method', (), cmd))
def test_call(self):
"""
Check the context of the message sent when L{call} is
executed.
"""
i = self.invoker
m = self.messages
d = i.call('remote_method', 1, 2, 3, 'foo', notify=True)
self.assertTrue(isinstance(d, defer.Deferred))
self.assertEqual(len(m), 1)
msg = m.pop()
self.assertEqual(message.typeByClass(msg), message.INVOKE)
self.assertEqual(msg.id, 2)
self.assertEqual(msg.name, 'remote_method')
self.assertEqual(msg.argv, [None, 1, 2, 3, 'foo'])
callContext = i.getCallContext(msg.id)
self.assertEqual(callContext,
(d, 'remote_method', (1, 2, 3, 'foo'), None))
def test_call(self):
"""
Check the context of the message sent when L{call} is
executed.
"""
i = self.invoker
m = self.messages
d = i.call('remote_method', 1, 2, 3, 'foo', notify=True)
self.assertTrue(isinstance(d, defer.Deferred))
self.assertEqual(len(m), 1)
msg = m.pop()
self.assertEqual(message.typeByClass(msg), message.INVOKE)
self.assertEqual(msg.id, 1)
self.assertEqual(msg.name, 'remote_method')
self.assertEqual(msg.argv, [None, 1, 2, 3, 'foo'])
callContext = i.getCallContext(msg.id)
self.assertEqual(callContext,
(d, 'remote_method', (1, 2, 3, 'foo'), None))
self.receiver.initiateCall(callId=callId)
try:
yield self.makeCall('foo', callId)
except exc.CallFailed, e:
pass
else:
self.fail('exc.CallFailed not raised')
self.assertEqual(str(e), 'Unable to initiate an already active call 2')
m = self.messages
self.assertEqual(len(m), 1)
msg = self.messages.pop()
self.assertTrue(message.typeByClass(msg), message.Invoke)
self.assertEqual(msg.name, '_error')
self.assertEqual(msg.argv, [None, {
'code': 'NetConnection.Call.Failed',
'description': 'Unable to initiate an already active call 2',
'level': 'error'
}])
self.assertEqual(msg.id, callId)
@defer.inlineCallbacks
def test_call_exposed(self):
"""
Call an exposed method on the faciliator.
"""
ret = yield self.receiver.callExposedMethod('exposed', 1, 2, 3)
self.receiver.initiateCall(callId=callId)
try:
yield self.makeCall('foo', callId)
except exc.CallFailed, e:
pass
else:
self.fail('exc.CallFailed not raised')
self.assertEqual(str(e), 'Unable to initiate an already active call 1')
m = self.messages
self.assertEqual(len(m), 1)
msg = self.messages.pop()
self.assertTrue(message.typeByClass(msg), message.Invoke)
self.assertEqual(msg.name, '_error')
self.assertEqual(msg.argv, [
None, {
'code': 'NetConnection.Call.Failed',
'description': 'Unable to initiate an already active call 1',
'level': 'error'
}
])
self.assertEqual(msg.id, callId)
@defer.inlineCallbacks
def test_call_exposed(self):
"""
Call an exposed method on the faciliator.
"""