RabbitSingleNodeStreamDeploymentIntegrationTests.java

/*
 * Copyright 2013-2015 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.xd.dirt.stream;

import static org.hamcrest.Matchers.instanceOf;
import static org.junit.Assert.*;

import java.util.Collections;
import java.util.List;
import java.util.Map;

import org.junit.BeforeClass;
import org.junit.ClassRule;
import org.junit.Test;
import org.junit.rules.ExternalResource;

import org.springframework.amqp.core.MessageDeliveryMode;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.amqp.utils.test.TestUtils;
import org.springframework.messaging.MessageChannel;
import org.springframework.xd.dirt.integration.bus.Binding;
import org.springframework.xd.dirt.integration.bus.MessageBus;
import org.springframework.xd.dirt.integration.bus.rabbit.RabbitTestMessageBus;
import org.springframework.xd.dirt.test.sink.NamedChannelSink;
import org.springframework.xd.dirt.test.sink.SingleNodeNamedChannelSinkFactory;
import org.springframework.xd.dirt.test.source.NamedChannelSource;
import org.springframework.xd.dirt.test.source.SingleNodeNamedChannelSourceFactory;
import org.springframework.xd.test.mqtt.MqttTestSupport;
import org.springframework.xd.test.rabbit.RabbitTestSupport;

/**
 * @author Mark Fisher
 * @author Gary Russell
 */
public class RabbitSingleNodeStreamDeploymentIntegrationTests extends
		AbstractDistributedTransportSingleNodeStreamDeploymentIntegrationTests {

	@ClassRule
	public static RabbitTestSupport rabbitAvailableRule = new RabbitTestSupport();

	@ClassRule
	public static MqttTestSupport mqttAvailableRule = new MqttTestSupport();

	@BeforeClass
	public static void setUp() {
		setUp("rabbit");
	}

	@ClassRule
	public static ExternalResource initializeRabbitTestMessageBus = new ExternalResource() {

		@Override
		protected void before() {
			if (testMessageBus == null || !(testMessageBus instanceof RabbitTestMessageBus)) {
				testMessageBus = new RabbitTestMessageBus(rabbitAvailableRule.getResource());
			}
		}
	};

	@Test
	public void mqttSourceStreamReceivesMqttSinkStreamOutput() throws Exception {
		StreamDefinition mqtt1 = new StreamDefinition("mqtt1", "queue:mqttsource > mqtt --topic=foo --async=true");
		StreamDefinition mqtt2 = new StreamDefinition("mqtt2", "mqtt --topics=foo --charset=UTF-8 > queue:mqttsink");
		integrationSupport.createAndDeployStream(mqtt1);
		integrationSupport.createAndDeployStream(mqtt2);

		NamedChannelSource source = new SingleNodeNamedChannelSourceFactory(integrationSupport.messageBus()).createNamedChannelSource("queue:mqttsource");
		NamedChannelSink sink = new SingleNodeNamedChannelSinkFactory(integrationSupport.messageBus()).createNamedChannelSink("queue:mqttsink");

		Thread.sleep(1000);
		source.sendPayload("hello");
		Object result = sink.receivePayload(1000);

		assertEquals("hello", result);
		source.unbind();
		sink.unbind();
	}

	@Test
	public void mqttSourceStreamReceivesMqttSinkStreamOutputBinary() throws Exception {
		StreamDefinition mqtt3 = new StreamDefinition("mqtt3", "queue:mqttsource2 > mqtt --topic=foo2");
		StreamDefinition mqtt4 = new StreamDefinition("mqtt4", "mqtt --topics=foo2 --binary=true > queue:mqttsink2");
		integrationSupport.createAndDeployStream(mqtt3);
		integrationSupport.createAndDeployStream(mqtt4);

		NamedChannelSource source = new SingleNodeNamedChannelSourceFactory(integrationSupport.messageBus()).createNamedChannelSource("queue:mqttsource2");
		NamedChannelSink sink = new SingleNodeNamedChannelSinkFactory(integrationSupport.messageBus()).createNamedChannelSink("queue:mqttsink2");

		Thread.sleep(1000);
		source.sendPayload("hello");
		Object result = sink.receivePayload(1000);

		assertThat(result, instanceOf(byte[].class));
		assertEquals("hello", new String((byte[]) result));
		source.unbind();
		sink.unbind();
	}

	@Override
	protected Map<String, String> onDemandProperties() {
		return Collections.singletonMap("module.router.producer.deliveryMode", "NON_PERSISTENT");
	}

	@Override
	protected void verifyOnDemandQueues(MessageChannel y3, MessageChannel z3, Map<String, Object> initialTransportState) {
		RabbitTemplate template = new RabbitTemplate(rabbitAvailableRule.getResource());
		Object y = template.receiveAndConvert("xdbus.queue:y");
		assertNotNull(y);
		assertEquals("y", y);
		Object z = template.receiveAndConvert("xdbus.queue:z");
		assertNotNull(z);
		assertEquals("z", z);
	}

	@Override
	protected void verifyDynamicProperties(MessageBus bus, String type) {
		@SuppressWarnings("unchecked")
		List<Binding> bindings = TestUtils.getPropertyValue(bus, "messageBus.bindings", List.class);
		for (Binding binding : bindings) {
			if (binding.getEndpoint().getComponentName().equals("outbound." + type + ":x")) {
				assertEquals(MessageDeliveryMode.PERSISTENT, TestUtils.getPropertyValue(binding.getEndpoint(),
						"handler.delegate.defaultDeliveryMode", MessageDeliveryMode.class));
			}
			else if (binding.getEndpoint().getComponentName().matches("outbound." + type + ":(y|z)")) {
				assertEquals(MessageDeliveryMode.NON_PERSISTENT, TestUtils.getPropertyValue(binding.getEndpoint(),
						"handler.delegate.defaultDeliveryMode", MessageDeliveryMode.class));
			}
		}
	}

}