def test_CommandMessage(self): m = messaging.CommandMessage(operation='foo.bar') self.assertEqual(pyamf.encode(m).getvalue(), '\n\x81\x1bMflex.messaging.messages.CommandMessage\x1bcorrelationId' '\tbody\x11clientId\x17destination\x0fheaders\x13messageId\x13' 'timestamp\x15timeToLive\x13operation\x01\x01\x01\x01\n\x0b\x01\x01' '\x01\x01\x01\x06\x0ffoo.bar\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 test_disconnect(self): message = messaging.CommandMessage(operation=12) rp = amf3.RequestProcessor(None) request = remoting.Request('null', body=[message]) response = rp(request) ack = response.body self.assertTrue(isinstance(response, remoting.Response)) self.assertEqual(response.status, remoting.STATUS_OK) self.assertTrue(isinstance(ack, messaging.AcknowledgeMessage))