org.apache.activemq.artemis.junit.ActiveMQConsumerResource Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of artemis-junit Show documentation
The newest version!
/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You 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.apache.activemq.artemis.junit;

import java.util.Map;

import org.apache.activemq.artemis.api.core.SimpleString;
import org.apache.activemq.artemis.api.core.client.ClientMessage;
import org.apache.activemq.artemis.api.core.client.ServerLocator;
import org.junit.rules.ExternalResource;

/**
 * A JUnit Rule that embeds an ActiveMQ Artemis ClientConsumer into a test. This JUnit Rule is
 * designed to simplify using ActiveMQ Artemis clients in unit tests. Adding the rule to a test will
 * startup a ClientConsumer, which can then be used to consume messages from an ActiveMQ Artemis
 * server.
 *
 *  * 
 * public class SimpleTest {
 *     @Rule
 *     public ActiveMQConsumerResource client = new ActiveMQProducerResource( "vm://0", "test.queue" );
 *
 *     @Test
 *     public void testSomething() throws Exception {
 *         // Use the embedded client here
 *         ClientMessage message = client.receiveMessage();
 *     }
 * }
 * 
 * 
 */
public class ActiveMQConsumerResource extends ExternalResource implements ActiveMQConsumerOperations {

   private final ActiveMQConsumerDelegate activeMQConsumer;

   public ActiveMQConsumerResource(String url, String queueName) {
      this.activeMQConsumer = new ActiveMQConsumerDelegate(url, queueName);
   }

   public ActiveMQConsumerResource(String url, String queueName, String username, String password) {
      this.activeMQConsumer =
               new ActiveMQConsumerDelegate(url, SimpleString.of(queueName), username, password);
   }

   public ActiveMQConsumerResource(String url, SimpleString queueName, String username, String password) {
      this.activeMQConsumer = new ActiveMQConsumerDelegate(url, queueName, username, password);
   }

   public ActiveMQConsumerResource(String url, SimpleString queueName) {
      this.activeMQConsumer = new ActiveMQConsumerDelegate(url, queueName, null, null);
   }

   public ActiveMQConsumerResource(ServerLocator serverLocator, String queueName, String username, String password) {
      this(serverLocator, SimpleString.of(queueName), username, password);
   }

   public ActiveMQConsumerResource(ServerLocator serverLocator, String queueName) {
      this(serverLocator, SimpleString.of(queueName), null, null);
   }

   public ActiveMQConsumerResource(ServerLocator serverLocator, SimpleString queueName, String username,
                                   String password) {
      this.activeMQConsumer = new ActiveMQConsumerDelegate(serverLocator, queueName, username, password);
   }

   public ActiveMQConsumerResource(ServerLocator serverLocator, SimpleString queueName) {
      this(serverLocator, queueName, null, null);
   }

   protected void createClient() {
      activeMQConsumer.createClient();
   }

   protected void stopClient() {
      activeMQConsumer.stopClient();
   }

   @Override
   public boolean isAutoCreateQueue() {
      return activeMQConsumer.isAutoCreateQueue();
   }

   @Override
   public void setAutoCreateQueue(boolean autoCreateQueue) {
      activeMQConsumer.setAutoCreateQueue(autoCreateQueue);
   }

   @Override
   public ClientMessage receiveMessage() {
      return activeMQConsumer.receiveMessage();
   }

   @Override
   public ClientMessage receiveMessage(long timeout) {
      return activeMQConsumer.receiveMessage(timeout);
   }

   @Override
   protected void before() throws Throwable {
      super.before();
      activeMQConsumer.start();
   }

   @Override
   protected void after() {
      activeMQConsumer.stop();
      super.after();
   }

   public static void addMessageProperties(ClientMessage message, Map properties) {
      AbstractActiveMQClientDelegate.addMessageProperties(message, properties);
   }

   @Override
   public long getDefaultReceiveTimeout() {
      return activeMQConsumer.getDefaultReceiveTimeout();
   }

   @Override
   public void setDefaultReceiveTimeout(long defaultReceiveTimeout) {
      activeMQConsumer.setDefaultReceiveTimeout(defaultReceiveTimeout);
   }

}