Examples with AmqpInboundGateway org.springframework.integration.amqp.inbound.AmqpInboundGateway used on opensource projects

Example 1 with AmqpInboundGateway

use of org.springframework.integration.amqp.inbound.AmqpInboundGateway in project spring-integration by spring-projects.

the class AmqpInboundGatewayParserTests method verifyUsageWithHeaderMapper.

@Test
public void verifyUsageWithHeaderMapper() throws Exception {
    DirectChannel requestChannel = context.getBean("requestChannel", DirectChannel.class);
    requestChannel.subscribe(siMessage -> {
        org.springframework.messaging.Message<?> replyMessage = MessageBuilder.fromMessage(siMessage).setHeader("bar", "bar").build();
        MessageChannel replyChannel = (MessageChannel) siMessage.getHeaders().getReplyChannel();
        replyChannel.send(replyMessage);
    });
    final AmqpInboundGateway gateway = context.getBean("withHeaderMapper", AmqpInboundGateway.class);
    Field amqpTemplateField = ReflectionUtils.findField(AmqpInboundGateway.class, "amqpTemplate");
    amqpTemplateField.setAccessible(true);
    RabbitTemplate amqpTemplate = TestUtils.getPropertyValue(gateway, "amqpTemplate", RabbitTemplate.class);
    amqpTemplate = Mockito.spy(amqpTemplate);
    Mockito.doAnswer(invocation -> {
        Object[] args = invocation.getArguments();
        Message amqpReplyMessage = (Message) args[2];
        MessageProperties properties = amqpReplyMessage.getMessageProperties();
        assertEquals("bar", properties.getHeaders().get("bar"));
        return null;
    }).when(amqpTemplate).send(Mockito.any(String.class), Mockito.any(String.class), Mockito.any(Message.class), isNull());
    ReflectionUtils.setField(amqpTemplateField, gateway, amqpTemplate);
    AbstractMessageListenerContainer mlc = TestUtils.getPropertyValue(gateway, "messageListenerContainer", AbstractMessageListenerContainer.class);
    ChannelAwareMessageListener listener = TestUtils.getPropertyValue(mlc, "messageListener", ChannelAwareMessageListener.class);
    MessageProperties amqpProperties = new MessageProperties();
    amqpProperties.setAppId("test.appId");
    amqpProperties.setClusterId("test.clusterId");
    amqpProperties.setContentEncoding("test.contentEncoding");
    amqpProperties.setContentLength(99L);
    amqpProperties.setReplyTo("oleg");
    amqpProperties.setContentType("test.contentType");
    amqpProperties.setHeader("foo", "foo");
    amqpProperties.setHeader("bar", "bar");
    Message amqpMessage = new Message("hello".getBytes(), amqpProperties);
    listener.onMessage(amqpMessage, null);
    Mockito.verify(amqpTemplate, Mockito.times(1)).send(Mockito.any(String.class), Mockito.any(String.class), Mockito.any(Message.class), isNull());
}