Ejemplo n.º 1
0
    def test_ack(self):
        self.assertEquals(commands.ack(StompFrame(StompSpec.MESSAGE, {StompSpec.MESSAGE_ID_HEADER: 'hi'})), StompFrame(command=StompSpec.ACK, headers={StompSpec.MESSAGE_ID_HEADER: 'hi'}))
        self.assertEquals(commands.ack(StompFrame(StompSpec.MESSAGE, {StompSpec.MESSAGE_ID_HEADER: 'hi', StompSpec.SUBSCRIPTION_HEADER: 'there'})), StompFrame(command=StompSpec.ACK, headers={StompSpec.MESSAGE_ID_HEADER: 'hi', StompSpec.SUBSCRIPTION_HEADER: 'there'}))
        self.assertEquals(commands.ack(StompFrame(StompSpec.MESSAGE, {StompSpec.MESSAGE_ID_HEADER: 'hi', StompSpec.SUBSCRIPTION_HEADER: 'there', StompSpec.TRANSACTION_HEADER: 'man'}), ['man', 'woman']), StompFrame(command=StompSpec.ACK, headers={StompSpec.MESSAGE_ID_HEADER: 'hi', StompSpec.SUBSCRIPTION_HEADER: 'there', StompSpec.TRANSACTION_HEADER: 'man'}))
        self.assertEquals(commands.ack(StompFrame(StompSpec.MESSAGE, {StompSpec.MESSAGE_ID_HEADER: 'hi', StompSpec.SUBSCRIPTION_HEADER: 'there', StompSpec.TRANSACTION_HEADER: 'man'}), transactions=['woman']), StompFrame(command=StompSpec.ACK, headers={StompSpec.MESSAGE_ID_HEADER: 'hi', StompSpec.SUBSCRIPTION_HEADER: 'there'}))
        self.assertEquals(commands.ack(StompFrame(StompSpec.MESSAGE, {StompSpec.MESSAGE_ID_HEADER: 'hi', StompSpec.SUBSCRIPTION_HEADER: 'there', StompSpec.TRANSACTION_HEADER: 'man'})), StompFrame(command=StompSpec.ACK, headers={StompSpec.MESSAGE_ID_HEADER: 'hi', StompSpec.SUBSCRIPTION_HEADER: 'there'}))
        self.assertEquals(commands.ack(StompFrame(StompSpec.MESSAGE, {StompSpec.MESSAGE_ID_HEADER: 'hi'}), receipt='4711'), StompFrame(command=StompSpec.ACK, headers={StompSpec.MESSAGE_ID_HEADER: 'hi', StompSpec.RECEIPT_HEADER: '4711'}))
        self.assertEquals(commands.ack(StompFrame(StompSpec.MESSAGE, {StompSpec.MESSAGE_ID_HEADER: 'hi', StompSpec.SUBSCRIPTION_HEADER: 'there', '4711': '0815', StompSpec.TRANSACTION_HEADER: 'man'})), StompFrame(command=StompSpec.ACK, headers={StompSpec.MESSAGE_ID_HEADER: 'hi', StompSpec.SUBSCRIPTION_HEADER: 'there'}))
        self.assertEquals(commands.ack(StompFrame(StompSpec.MESSAGE, {StompSpec.MESSAGE_ID_HEADER: 'hi', StompSpec.SUBSCRIPTION_HEADER: 'there', '4711': '0815', StompSpec.TRANSACTION_HEADER: 'man'}), ['man'], '4711'), StompFrame(command=StompSpec.ACK, headers={StompSpec.MESSAGE_ID_HEADER: 'hi', StompSpec.SUBSCRIPTION_HEADER: 'there', StompSpec.TRANSACTION_HEADER: 'man', StompSpec.RECEIPT_HEADER: '4711'}))
        self.assertEquals(commands.ack(StompFrame(StompSpec.MESSAGE, {StompSpec.MESSAGE_ID_HEADER: 'hi', StompSpec.SUBSCRIPTION_HEADER: 'there', StompSpec.TRANSACTION_HEADER: 'man'}, version=StompSpec.VERSION_1_1)), StompFrame(command=StompSpec.ACK, headers={StompSpec.MESSAGE_ID_HEADER: 'hi', StompSpec.SUBSCRIPTION_HEADER: 'there'}))
        self.assertEquals(commands.ack(StompFrame(StompSpec.MESSAGE, {StompSpec.MESSAGE_ID_HEADER: 'hi', StompSpec.SUBSCRIPTION_HEADER: 'there', '4711': '0815', StompSpec.TRANSACTION_HEADER: 'man'}, version=StompSpec.VERSION_1_1), transactions=set(['man', 'woman'])), StompFrame(command=StompSpec.ACK, headers={StompSpec.MESSAGE_ID_HEADER: 'hi', StompSpec.SUBSCRIPTION_HEADER: 'there', StompSpec.TRANSACTION_HEADER: 'man'}))
        self.assertRaises(StompProtocolError, commands.ack, StompFrame(StompSpec.CONNECTED, {StompSpec.MESSAGE_ID_HEADER: 'hi'}, version=StompSpec.VERSION_1_0))
        self.assertRaises(StompProtocolError, commands.ack, StompFrame(StompSpec.CONNECTED, {StompSpec.MESSAGE_ID_HEADER: 'hi'}, version=StompSpec.VERSION_1_1))
        self.assertRaises(StompProtocolError, commands.ack, StompFrame(StompSpec.MESSAGE, {StompSpec.SUBSCRIPTION_HEADER: 'hi'}))
        self.assertRaises(StompProtocolError, commands.ack, StompFrame(StompSpec.MESSAGE, {StompSpec.SUBSCRIPTION_HEADER: 'hi'}, version=StompSpec.VERSION_1_1))
        self.assertRaises(StompProtocolError, commands.ack, StompFrame(StompSpec.MESSAGE, {StompSpec.MESSAGE_ID_HEADER: 'hi'}, version=StompSpec.VERSION_1_1))

        self.assertEquals(commands.ack(StompFrame(StompSpec.MESSAGE, {StompSpec.MESSAGE_ID_HEADER: 'hi', StompSpec.SUBSCRIPTION_HEADER: 'there', StompSpec.ACK_HEADER: '4711'}, version=StompSpec.VERSION_1_2)), StompFrame(command=StompSpec.ACK, headers={'id': '4711'}))
        self.assertRaises(StompProtocolError, commands.ack, StompFrame(StompSpec.MESSAGE, {StompSpec.MESSAGE_ID_HEADER: 'hi'}, version=StompSpec.VERSION_1_2))
        self.assertRaises(StompProtocolError, commands.ack, StompFrame(StompSpec.MESSAGE, {StompSpec.MESSAGE_ID_HEADER: 'hi', StompSpec.SUBSCRIPTION_HEADER: 'there'}, version=StompSpec.VERSION_1_2))
Ejemplo n.º 2
0
    def test_ack(self):
        self.assertEqual(
            commands.ack(
                StompFrame(StompSpec.MESSAGE,
                           {StompSpec.MESSAGE_ID_HEADER: 'hi'})),
            StompFrame(command=StompSpec.ACK,
                       headers={StompSpec.MESSAGE_ID_HEADER: 'hi'}))
        self.assertEqual(
            commands.ack(
                StompFrame(
                    StompSpec.MESSAGE, {
                        StompSpec.MESSAGE_ID_HEADER: 'hi',
                        StompSpec.SUBSCRIPTION_HEADER: 'there'
                    })),
            StompFrame(command=StompSpec.ACK,
                       headers={
                           StompSpec.MESSAGE_ID_HEADER: 'hi',
                           StompSpec.SUBSCRIPTION_HEADER: 'there'
                       }))
        self.assertEqual(
            commands.ack(
                StompFrame(
                    StompSpec.MESSAGE, {
                        StompSpec.MESSAGE_ID_HEADER: 'hi',
                        StompSpec.SUBSCRIPTION_HEADER: 'there',
                        StompSpec.TRANSACTION_HEADER: 'man'
                    }), ['man', 'woman']),
            StompFrame(command=StompSpec.ACK,
                       headers={
                           StompSpec.MESSAGE_ID_HEADER: 'hi',
                           StompSpec.SUBSCRIPTION_HEADER: 'there',
                           StompSpec.TRANSACTION_HEADER: 'man'
                       }))
        self.assertEqual(
            commands.ack(StompFrame(
                StompSpec.MESSAGE, {
                    StompSpec.MESSAGE_ID_HEADER: 'hi',
                    StompSpec.SUBSCRIPTION_HEADER: 'there',
                    StompSpec.TRANSACTION_HEADER: 'man'
                }),
                         transactions=['woman']),
            StompFrame(command=StompSpec.ACK,
                       headers={
                           StompSpec.MESSAGE_ID_HEADER: 'hi',
                           StompSpec.SUBSCRIPTION_HEADER: 'there'
                       }))
        self.assertEqual(
            commands.ack(
                StompFrame(
                    StompSpec.MESSAGE, {
                        StompSpec.MESSAGE_ID_HEADER: 'hi',
                        StompSpec.SUBSCRIPTION_HEADER: 'there',
                        StompSpec.TRANSACTION_HEADER: 'man'
                    })),
            StompFrame(command=StompSpec.ACK,
                       headers={
                           StompSpec.MESSAGE_ID_HEADER: 'hi',
                           StompSpec.SUBSCRIPTION_HEADER: 'there'
                       }))
        self.assertEqual(
            commands.ack(StompFrame(StompSpec.MESSAGE,
                                    {StompSpec.MESSAGE_ID_HEADER: 'hi'}),
                         receipt='4711'),
            StompFrame(command=StompSpec.ACK,
                       headers={
                           StompSpec.MESSAGE_ID_HEADER: 'hi',
                           StompSpec.RECEIPT_HEADER: '4711'
                       }))
        self.assertEqual(
            commands.ack(
                StompFrame(
                    StompSpec.MESSAGE, {
                        StompSpec.MESSAGE_ID_HEADER: 'hi',
                        StompSpec.SUBSCRIPTION_HEADER: 'there',
                        '4711': '0815',
                        StompSpec.TRANSACTION_HEADER: 'man'
                    })),
            StompFrame(command=StompSpec.ACK,
                       headers={
                           StompSpec.MESSAGE_ID_HEADER: 'hi',
                           StompSpec.SUBSCRIPTION_HEADER: 'there'
                       }))
        self.assertEqual(
            commands.ack(
                StompFrame(
                    StompSpec.MESSAGE, {
                        StompSpec.MESSAGE_ID_HEADER: 'hi',
                        StompSpec.SUBSCRIPTION_HEADER: 'there',
                        '4711': '0815',
                        StompSpec.TRANSACTION_HEADER: 'man'
                    }), ['man'], '4711'),
            StompFrame(command=StompSpec.ACK,
                       headers={
                           StompSpec.MESSAGE_ID_HEADER: 'hi',
                           StompSpec.SUBSCRIPTION_HEADER: 'there',
                           StompSpec.TRANSACTION_HEADER: 'man',
                           StompSpec.RECEIPT_HEADER: '4711'
                       }))
        self.assertEqual(
            commands.ack(
                StompFrame(StompSpec.MESSAGE, {
                    StompSpec.MESSAGE_ID_HEADER: 'hi',
                    StompSpec.SUBSCRIPTION_HEADER: 'there',
                    StompSpec.TRANSACTION_HEADER: 'man'
                },
                           version=StompSpec.VERSION_1_1)),
            StompFrame(command=StompSpec.ACK,
                       headers={
                           StompSpec.MESSAGE_ID_HEADER: 'hi',
                           StompSpec.SUBSCRIPTION_HEADER: 'there'
                       }))
        self.assertEqual(
            commands.ack(StompFrame(StompSpec.MESSAGE, {
                StompSpec.MESSAGE_ID_HEADER: 'hi',
                StompSpec.SUBSCRIPTION_HEADER: 'there',
                '4711': '0815',
                StompSpec.TRANSACTION_HEADER: 'man'
            },
                                    version=StompSpec.VERSION_1_1),
                         transactions=set(['man', 'woman'])),
            StompFrame(command=StompSpec.ACK,
                       headers={
                           StompSpec.MESSAGE_ID_HEADER: 'hi',
                           StompSpec.SUBSCRIPTION_HEADER: 'there',
                           StompSpec.TRANSACTION_HEADER: 'man'
                       }))
        self.assertRaises(
            StompProtocolError, commands.ack,
            StompFrame(StompSpec.CONNECTED,
                       {StompSpec.MESSAGE_ID_HEADER: 'hi'},
                       version=StompSpec.VERSION_1_0))
        self.assertRaises(
            StompProtocolError, commands.ack,
            StompFrame(StompSpec.CONNECTED,
                       {StompSpec.MESSAGE_ID_HEADER: 'hi'},
                       version=StompSpec.VERSION_1_1))
        self.assertRaises(
            StompProtocolError, commands.ack,
            StompFrame(StompSpec.MESSAGE,
                       {StompSpec.SUBSCRIPTION_HEADER: 'hi'}))
        self.assertRaises(
            StompProtocolError, commands.ack,
            StompFrame(StompSpec.MESSAGE,
                       {StompSpec.SUBSCRIPTION_HEADER: 'hi'},
                       version=StompSpec.VERSION_1_1))
        self.assertRaises(
            StompProtocolError, commands.ack,
            StompFrame(StompSpec.MESSAGE, {StompSpec.MESSAGE_ID_HEADER: 'hi'},
                       version=StompSpec.VERSION_1_1))

        self.assertEqual(
            commands.ack(
                StompFrame(StompSpec.MESSAGE, {
                    StompSpec.MESSAGE_ID_HEADER: 'hi',
                    StompSpec.SUBSCRIPTION_HEADER: 'there',
                    StompSpec.ACK_HEADER: '4711'
                },
                           version=StompSpec.VERSION_1_2)),
            StompFrame(command=StompSpec.ACK, headers={'id': '4711'}))
        self.assertRaises(
            StompProtocolError, commands.ack,
            StompFrame(StompSpec.MESSAGE, {StompSpec.MESSAGE_ID_HEADER: 'hi'},
                       version=StompSpec.VERSION_1_2))
        self.assertRaises(
            StompProtocolError, commands.ack,
            StompFrame(StompSpec.MESSAGE, {
                StompSpec.MESSAGE_ID_HEADER: 'hi',
                StompSpec.SUBSCRIPTION_HEADER: 'there'
            },
                       version=StompSpec.VERSION_1_2))