org.springframework.cloud.stream.config.MessageChannelConfigurerTests.java Source code

Introduction

Here is the source code for org.springframework.cloud.stream.config.MessageChannelConfigurerTests.java

Source

/*
 * Copyright 2015-2017 the original author or authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.springframework.cloud.stream.config;

import java.util.concurrent.CountDownLatch;
import java.util.concurrent.TimeUnit;

import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.databind.SerializationFeature;
import org.junit.Test;
import org.junit.runner.RunWith;

import org.springframework.beans.DirectFieldAccessor;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.autoconfigure.EnableAutoConfiguration;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.cloud.stream.annotation.EnableBinding;
import org.springframework.cloud.stream.binder.BinderHeaders;
import org.springframework.cloud.stream.converter.CompositeMessageConverterFactory;
import org.springframework.cloud.stream.messaging.Sink;
import org.springframework.cloud.stream.messaging.Source;
import org.springframework.cloud.stream.test.binder.MessageCollector;
import org.springframework.context.annotation.PropertySource;
import org.springframework.integration.support.MessageBuilder;
import org.springframework.messaging.Message;
import org.springframework.messaging.MessageHandler;
import org.springframework.messaging.MessagingException;
import org.springframework.messaging.converter.CompositeMessageConverter;
import org.springframework.messaging.converter.MessageConverter;
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
import org.springframework.tuple.Tuple;
import org.springframework.util.MimeTypeUtils;

import static org.assertj.core.api.Assertions.assertThat;
import static org.junit.Assert.assertNull;

/**
 * @author Ilayaperumal Gopinathan
 * @author Gary Russell
 */
@RunWith(SpringJUnit4ClassRunner.class)
@SpringBootTest(classes = { MessageChannelConfigurerTests.TestSink.class,
        MessageChannelConfigurerTests.TestSource.class })
public class MessageChannelConfigurerTests {

    @Autowired
    private Sink testSink;

    @Autowired
    private Source testSource;

    @Autowired
    private CompositeMessageConverterFactory messageConverterFactory;

    @Autowired
    private ObjectMapper objectMapper;

    @Autowired
    private MessageCollector messageCollector;

    @Test
    public void testMessageConverterConfigurer() throws Exception {
        final CountDownLatch latch = new CountDownLatch(1);
        MessageHandler messageHandler = new MessageHandler() {
            @Override
            public void handleMessage(Message<?> message) throws MessagingException {
                assertThat(message.getPayload()).isInstanceOf(Tuple.class);
                assertThat(((Tuple) message.getPayload()).getFieldNames().get(0)).isEqualTo("message");
                assertThat(((Tuple) message.getPayload()).getValue(0)).isEqualTo("Hi");
                latch.countDown();
            }
        };
        testSink.input().subscribe(messageHandler);
        testSink.input().send(MessageBuilder.withPayload("{\"message\":\"Hi\"}").build());
        assertThat(latch.await(10, TimeUnit.SECONDS)).isTrue();
        testSink.input().unsubscribe(messageHandler);
    }

    @Test
    public void testObjectMapperConfig() throws Exception {
        CompositeMessageConverter converters = (CompositeMessageConverter) messageConverterFactory
                .getMessageConverterForType(MimeTypeUtils.APPLICATION_JSON);
        for (MessageConverter converter : converters.getConverters()) {
            DirectFieldAccessor converterAccessor = new DirectFieldAccessor(converter);
            ObjectMapper objectMapper = (ObjectMapper) converterAccessor.getPropertyValue("objectMapper");
            // assert that the ObjectMapper used by the converters is compliant with the
            // Boot configuration
            assertThat(!objectMapper.getSerializationConfig()
                    .isEnabled(SerializationFeature.WRITE_DATES_AS_TIMESTAMPS))
                            .withFailMessage("SerializationFeature 'WRITE_DATES_AS_TIMESTAMPS' should be disabled");
            // assert that the globally set bean is used by the converters
            assertThat(objectMapper).isSameAs(this.objectMapper);
        }
    }

    @Test
    public void testPartitionHeader() throws Exception {
        this.testSource.output().send(MessageBuilder.withPayload("{\"message\":\"Hi\"}").build());
        Message<?> message = this.messageCollector.forChannel(testSource.output()).poll(1, TimeUnit.SECONDS);
        assertThat(message.getHeaders().get(BinderHeaders.PARTITION_HEADER).equals(0));
        assertNull(message.getHeaders().get(BinderHeaders.PARTITION_OVERRIDE));
    }

    @Test
    public void testPartitionHeaderWithPartitionOverride() throws Exception {
        this.testSource.output().send(MessageBuilder.withPayload("{\"message\":\"Hi\"}")
                .setHeader(BinderHeaders.PARTITION_OVERRIDE, 123).build());
        Message<?> message = this.messageCollector.forChannel(testSource.output()).poll(1, TimeUnit.SECONDS);
        assertThat(message.getHeaders().get(BinderHeaders.PARTITION_HEADER).equals(123));
        assertNull(message.getHeaders().get(BinderHeaders.PARTITION_OVERRIDE));
    }

    @EnableBinding(Sink.class)
    @EnableAutoConfiguration
    @PropertySource("classpath:/org/springframework/cloud/stream/config/channel/sink-channel-configurers.properties")
    public static class TestSink {

    }

    @EnableBinding(Source.class)
    @EnableAutoConfiguration
    @PropertySource("classpath:/org/springframework/cloud/stream/config/channel/partitioned-configurers.properties")
    public static class TestSource {

    }
}