def testProcessPacket(self):
outter_msg = remoting.Message(target='null', response='/1')
inner_msg = messaging.CommandMessage()
inner_msg.destination = self.service_name
inner_msg.operation = messaging.CommandMessage.CLIENT_PING_OPERATION
inner_msg.headers = {'DSEndpoint': 'amf'}
inner_msg.body = (None, )
inner_msg.messageId = '123'
outter_msg.body = (inner_msg, )
packet = remoting.Packet(messages=[outter_msg])
encoded_packet = self.channel.encode(packet)
decoded_packet = self.channel.decode(encoded_packet)
response = self.channel.invoke(decoded_packet)
encoded_response = self.channel.encode(response)
response = self.channel.endpoint.decodePacket(encoded_response)
# Check outer msg
self.assertEquals(1, len(response.messages))
self.assertEquals(True,
response.messages[0].target.endswith('onResult'))
self.assertEquals('', response.messages[0].response)
# Check inner msg
self.assertEquals(messaging.AcknowledgeMessage,
response.messages[0].body.__class__)
self.assertEquals('123', response.messages[0].body.correlationId)
def testOldStyleTarget(self):
qualified_name = self.service_name + '.' + self.target_name
message = remoting.Message(target=qualified_name,
response='/1',
body=(self.arg, ))
packet = remoting.Packet(messages=[message])
packet.channel = self.channel
response = packet.invoke()
self.assertEquals(1, len(response.messages))
self.assertEquals(True,
response.messages[0].target.endswith('onResult'))
self.assertEquals('', response.messages[0].response)
self.assertEquals(True, response.messages[0].body)
def testOldStyleTargetFault(self):
message = remoting.Message(target='bad_target',
response='/1',
body=(self.arg, ))
packet = remoting.Packet(messages=[message])
packet.channel = self.channel
response = packet.invoke()
self.assertEquals(1, len(response.messages))
self.assertEquals(True,
response.messages[0].target.endswith('onStatus'))
self.assertEquals('', response.messages[0].response)
self.assertEquals(remoting.AsError,
response.messages[0].body.__class__)
def message_to_amfast(name, msg):
"""Converts a PyAmf Request to an AmFast Message
arguments:
===========
* name - string, response
* msg - pyamf.remoting.Request
Returns amfast.remoting.Message
"""
if hasattr(msg, 'target'):
target = msg.target
else:
target = ''
if hasattr(msg, 'status'):
response = name + msg.status
else:
response = name
return amfast_remoting.Message(target=target, response=response, body=msg.body)
def testRoTarget(self):
outter_msg = remoting.Message(target='null', response='/1')
inner_msg = messaging.RemotingMessage()
inner_msg.destination = self.service_name
inner_msg.operation = self.target_name
inner_msg.headers = {self.target_name: self.arg, 'DSEndpoint': 'amf'}
inner_msg.body = (self.arg, )
inner_msg.messageId = '123'
outter_msg.body = (inner_msg, )
packet = remoting.Packet(messages=[outter_msg])
packet.channel = self.channel
response = packet.invoke()
# Check outer msg
self.assertEquals(1, len(response.messages))
self.assertEquals(True,
response.messages[0].target.endswith('onResult'))
self.assertEquals('', response.messages[0].response)
# Check inner msg
self.assertEquals(True, response.messages[0].body.body)
self.assertEquals('123', response.messages[0].body.correlationId)