Examples with Connection org.springframework.amqp.rabbit.connection.Connection used on opensource projects

Example 11 with Connection

use of org.springframework.amqp.rabbit.connection.Connection in project spring-integration by spring-projects.

the class InboundEndpointTests method testMessageConverterJsonHeadersHavePrecedenceOverMessageHeaders.

@Test
public void testMessageConverterJsonHeadersHavePrecedenceOverMessageHeaders() throws Exception {
    Connection connection = mock(Connection.class);
    doAnswer(invocation -> mock(Channel.class)).when(connection).createChannel(anyBoolean());
    ConnectionFactory connectionFactory = mock(ConnectionFactory.class);
    when(connectionFactory.createConnection()).thenReturn(connection);
    SimpleMessageListenerContainer container = new SimpleMessageListenerContainer();
    container.setConnectionFactory(connectionFactory);
    container.setAcknowledgeMode(AcknowledgeMode.MANUAL);
    DirectChannel channel = new DirectChannel();
    final Channel rabbitChannel = mock(Channel.class);
    channel.subscribe(new MessageTransformingHandler(message -> {
        assertSame(rabbitChannel, message.getHeaders().get(AmqpHeaders.CHANNEL));
        assertEquals(123L, message.getHeaders().get(AmqpHeaders.DELIVERY_TAG));
        return MessageBuilder.fromMessage(message).setHeader(JsonHeaders.TYPE_ID, "foo").setHeader(JsonHeaders.CONTENT_TYPE_ID, "bar").setHeader(JsonHeaders.KEY_TYPE_ID, "baz").build();
    }));
    RabbitTemplate rabbitTemplate = spy(new RabbitTemplate());
    rabbitTemplate.setMessageConverter(new Jackson2JsonMessageConverter());
    CountDownLatch sendLatch = new CountDownLatch(1);
    Mockito.doAnswer(invocation -> {
        org.springframework.amqp.core.Message message = invocation.getArgument(2);
        Map<String, Object> headers = message.getMessageProperties().getHeaders();
        assertTrue(headers.containsKey(JsonHeaders.TYPE_ID.replaceFirst(JsonHeaders.PREFIX, "")));
        assertNotEquals("foo", headers.get(JsonHeaders.TYPE_ID.replaceFirst(JsonHeaders.PREFIX, "")));
        assertFalse(headers.containsKey(JsonHeaders.CONTENT_TYPE_ID.replaceFirst(JsonHeaders.PREFIX, "")));
        assertFalse(headers.containsKey(JsonHeaders.KEY_TYPE_ID.replaceFirst(JsonHeaders.PREFIX, "")));
        assertFalse(headers.containsKey(JsonHeaders.TYPE_ID));
        assertFalse(headers.containsKey(JsonHeaders.KEY_TYPE_ID));
        assertFalse(headers.containsKey(JsonHeaders.CONTENT_TYPE_ID));
        sendLatch.countDown();
        return null;
    }).when(rabbitTemplate).send(anyString(), anyString(), any(org.springframework.amqp.core.Message.class), isNull());
    AmqpInboundGateway gateway = new AmqpInboundGateway(container, rabbitTemplate);
    gateway.setMessageConverter(new Jackson2JsonMessageConverter());
    gateway.setRequestChannel(channel);
    gateway.setBeanFactory(mock(BeanFactory.class));
    gateway.setDefaultReplyTo("foo");
    gateway.afterPropertiesSet();
    Object payload = new Foo("bar1");
    MessageProperties amqpMessageProperties = new MessageProperties();
    amqpMessageProperties.setDeliveryTag(123L);
    org.springframework.amqp.core.Message amqpMessage = new Jackson2JsonMessageConverter().toMessage(payload, amqpMessageProperties);
    ChannelAwareMessageListener listener = (ChannelAwareMessageListener) container.getMessageListener();
    listener.onMessage(amqpMessage, rabbitChannel);
    assertTrue(sendLatch.await(10, TimeUnit.SECONDS));
}

Example 12 with Connection

use of org.springframework.amqp.rabbit.connection.Connection in project spring-integration by spring-projects.

the class InboundEndpointTests method testAdapterConversionError.

@Test
public void testAdapterConversionError() throws Exception {
    Connection connection = mock(Connection.class);
    doAnswer(invocation -> mock(Channel.class)).when(connection).createChannel(anyBoolean());
    ConnectionFactory connectionFactory = mock(ConnectionFactory.class);
    when(connectionFactory.createConnection()).thenReturn(connection);
    SimpleMessageListenerContainer container = new SimpleMessageListenerContainer();
    container.setConnectionFactory(connectionFactory);
    AmqpInboundChannelAdapter adapter = new AmqpInboundChannelAdapter(container);
    QueueChannel outputChannel = new QueueChannel();
    adapter.setOutputChannel(outputChannel);
    QueueChannel errorChannel = new QueueChannel();
    adapter.setErrorChannel(errorChannel);
    adapter.setMessageConverter(new MessageConverter() {

        @Override
        public org.springframework.amqp.core.Message toMessage(Object object, MessageProperties messageProperties) throws MessageConversionException {
            throw new MessageConversionException("intended");
        }

        @Override
        public Object fromMessage(org.springframework.amqp.core.Message message) throws MessageConversionException {
            return null;
        }
    });
    adapter.afterPropertiesSet();
    ((ChannelAwareMessageListener) container.getMessageListener()).onMessage(null, null);
    assertNull(outputChannel.receive(0));
    assertNotNull(errorChannel.receive(0));
}