Ejemplo n.º 1
0
    def test_send(self, _find, auth, uuid4, document):
        _impl = Mock()
        _impl.send.return_value = '456'
        plugin = Mock()
        plugin.Sender.return_value = _impl
        _find.return_value = plugin
        uuid4.return_value = '<uuid>'
        address = 'amq.direct/bar'
        ttl = 234
        body = {'A': 1, 'B': 2}

        # test
        producer = Producer(TEST_URL)
        producer.authenticator = Mock()
        sn = producer.send(address, ttl=ttl, **body)

        # validation
        document.assert_called_once_with(sn=str(uuid4.return_value),
                                         version=VERSION,
                                         routing=(None, address))
        unsigned = document.return_value
        auth.sign.assert_called_once_with(
            producer.authenticator,
            unsigned.__iadd__.return_value.dump.return_value)
        _impl.send.assert_called_once_with(address, auth.sign.return_value,
                                           ttl)
        self.assertEqual(sn, uuid4.return_value)
Ejemplo n.º 2
0
    def test_send(self, _find, auth, uuid4, document):
        _impl = Mock()
        _impl.send.return_value = '456'
        plugin = Mock()
        plugin.Sender.return_value = _impl
        _find.return_value = plugin
        uuid4.return_value = '<uuid>'
        address = 'amq.direct/bar'
        ttl = 234
        body = {'A': 1, 'B': 2}

        # test
        producer = Producer(TEST_URL)
        producer.authenticator = Mock()
        sn = producer.send(address, ttl=ttl, **body)

        # validation
        document.assert_called_once_with(
            sn=str(uuid4.return_value),
            version=VERSION,
            routing=(None, address)
        )
        unsigned = document.return_value
        auth.sign.assert_called_once_with(
            producer.authenticator, unsigned.__iadd__.return_value.dump.return_value)
        _impl.send.assert_called_once_with(address, auth.sign.return_value, ttl)
        self.assertEqual(sn, uuid4.return_value)