def test_AcknowledgeMessage(self):
m = messaging.AcknowledgeMessage()
m.correlationId = '1234'
self.assertEqual(pyamf.encode(m).getvalue(),
'\n\x81\x0bUflex.messaging.messages.AcknowledgeMessage\tbody'
'\x11clientId\x17destination\x0fheaders\x13messageId\x13timestamp'
'\x15timeToLive\x1bcorrelationId\x01\x01\x01\n\x0b\x01\x01\x01\x01'
'\x01\x06\t1234\x01')
def test_getmessage(self):
"""
Tests for `getSmallMessage`
"""
for cls in ['AbstractMessage', 'ErrorMessage', 'RemotingMessage']:
cls = getattr(messaging, cls)
self.assertRaises(NotImplementedError, cls().getSmallMessage)
kwargs = {
'body': {
'foo': 'bar'
},
'clientId': 'spam',
'destination': 'eggs',
'headers': {
'blarg': 'whoop'
},
'messageId': 'baz',
'timestamp': 1234,
'timeToLive': 99
}
# test async
a = messaging.AsyncMessage(correlationId='yay', **kwargs)
m = a.getSmallMessage()
k = kwargs.copy()
k.update({'correlationId': 'yay'})
self.assertTrue(isinstance(m, messaging.AsyncMessageExt))
self.assertEquals(m.__dict__, k)
# test command
a = messaging.CommandMessage(operation='yay', **kwargs)
m = a.getSmallMessage()
k = kwargs.copy()
k.update({
'operation': 'yay',
'correlationId': None,
'messageRefType': None
})
self.assertTrue(isinstance(m, messaging.CommandMessageExt))
self.assertEquals(m.__dict__, k)
# test ack
a = messaging.AcknowledgeMessage(**kwargs)
m = a.getSmallMessage()
k = kwargs.copy()
k.update({'correlationId': None})
self.assertTrue(isinstance(m, messaging.AcknowledgeMessageExt))
self.assertEquals(m.__dict__, k)
def generate_acknowledgement(request=None):
ack = messaging.AcknowledgeMessage()
ack.messageId = generate_random_id()
ack.clientId = generate_random_id()
ack.timestamp = calendar.timegm(time.gmtime())
if request:
ack.correlationId = request.messageId
return ack