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