content: /*
   * 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.transport.amqp;
  
  import static org.junit.Assert.assertNotNull;
  import static org.junit.Assert.assertTrue;
  
  import java.io.UnsupportedEncodingException;
  import java.util.Arrays;
  
  import jakarta.jms.BytesMessage;
  import jakarta.jms.Connection;
  import jakarta.jms.ExceptionListener;
  import jakarta.jms.JMSException;
  import jakarta.jms.Message;
  import jakarta.jms.MessageConsumer;
  import jakarta.jms.Session;
  
  import org.apache.activemq.broker.BrokerService;
  import org.apache.activemq.broker.TransportConnector;
  import org.apache.qpid.jms.JmsConnectionFactory;
  import org.fusesource.mqtt.client.BlockingConnection;
  import org.fusesource.mqtt.client.MQTT;
  import org.fusesource.mqtt.client.QoS;
  import org.junit.After;
  import org.junit.Before;
  import org.junit.Test;
  
  public class AmqpAndMqttTest {
  
      protected BrokerService broker;
      private TransportConnector amqpConnector;
      private TransportConnector mqttConnector;
  
      @Before
      public void setUp() throws Exception {
          broker = createBroker();
          broker.start();
          broker.waitUntilStarted();
      }
  
      @After
      public void tearDown() throws Exception {
          if (broker != null) {
              broker.stop();
              broker.waitUntilStopped();
              broker = null;
          }
      }
  
      protected BrokerService createBroker() throws Exception {
          BrokerService broker = new BrokerService();
          broker.setPersistent(false);
          broker.setUseJmx(false);
          broker.setAdvisorySupport(false);
          broker.setSchedulerSupport(false);
  
          amqpConnector = broker.addConnector("amqp://0.0.0.0:0");
          mqttConnector = broker.addConnector("mqtt://0.0.0.0:0");
  
          return broker;
      }
  
      @Test(timeout = 60000)
      public void testFromMqttToAmqp() throws Exception {
          Connection amqp = createAmqpConnection();
          Session session = amqp.createSession(false, Session.AUTO_ACKNOWLEDGE);
          MessageConsumer consumer = session.createConsumer(session.createTopic("topic://FOO"));
  
          final BlockingConnection mqtt = createMQTTConnection().blockingConnection();
          mqtt.connect();
          byte[] payload = bytes("Hello World");
          mqtt.publish("FOO", payload, QoS.AT_LEAST_ONCE, false);
          mqtt.disconnect();
  
          Message msg = consumer.receive(1000 * 5);
          assertNotNull(msg);
          assertTrue(msg instanceof BytesMessage);
  
          BytesMessage bmsg = (BytesMessage) msg;
          byte[] actual = new byte[(int) bmsg.getBodyLength()];
          bmsg.readBytes(actual);
          assertTrue(Arrays.equals(actual, payload));
          amqp.close();
      }
  
      private byte[] bytes(String value) {
          try {
              return value.getBytes("UTF-8");
          } catch (UnsupportedEncodingException e) {
              throw new RuntimeException(e);
          }
      }
  
      protected MQTT createMQTTConnection() throws Exception {
          MQTT mqtt = new MQTT();
          mqtt.setConnectAttemptsMax(1);
          mqtt.setReconnectAttemptsMax(0);
          mqtt.setHost("localhost", mqttConnector.getConnectUri().getPort());
          return mqtt;
      }
  
      public Connection createAmqpConnection() throws Exception {
  
          String amqpURI = "amqp://localhost:" + amqpConnector.getConnectUri().getPort();
  
          final JmsConnectionFactory factory = new JmsConnectionFactory(amqpURI);
  
          factory.setUsername("admin");
          factory.setPassword("password");
  
          final Connection connection = factory.createConnection();
          connection.setExceptionListener(new ExceptionListener() {
              @Override
              public void onException(JMSException exception) {
                  exception.printStackTrace();
              }
          });
  
          connection.start();
          return connection;
      }
  }
  