Moved project package from org.eclipse.moquette to io.moquette

2015-11-20 21:30:59 +01:00
parent f438ef7f49
commit 371c740321
175 changed files with 629 additions and 669 deletions
@@ -0,0 +1,187 @@
+/*
+ * Copyright (c) 2012-2015 The original author or authors
+ * ------------------------------------------------------
+ * All rights reserved. This program and the accompanying materials
+ * are made available under the terms of the Eclipse Public License v1.0
+ * and Apache License v2.0 which accompanies this distribution.
+ *
+ * The Eclipse Public License is available at
+ * http://www.eclipse.org/legal/epl-v10.html
+ *
+ * The Apache License v2.0 is available at
+ * http://www.opensource.org/licenses/apache2.0.php
+ *
+ * You may elect to redistribute this code under either of these licenses.
+ */
+package io.moquette.server;
+
+import io.moquette.server.config.IConfig;
+import io.moquette.server.config.MemoryConfig;
+import org.fusesource.mqtt.client.*;
+import org.junit.After;
+import org.junit.Before;
+import org.junit.Test;
+import java.io.IOException;
+import java.util.Properties;
+import java.util.concurrent.TimeUnit;
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertNull;
+
+/**
+ *
+ * @author andrea
+ */
+public class ServerRestartIntegrationTest {
+
+    Server m_server;
+    MQTT m_mqtt;
+    BlockingConnection m_subscriber;
+    BlockingConnection m_publisher;
+    IConfig m_config;
+    
+    protected void startServer() throws IOException {
+        m_server = new Server();
+        final Properties configProps = IntegrationUtils.prepareTestPropeties();
+        m_config = new MemoryConfig(configProps);
+        m_server.startServer(m_config);
+    }
+
+    @Before
+    public void setUp() throws Exception {
+        startServer();
+
+        m_mqtt = new MQTT();
+
+        m_mqtt.setHost("localhost", 1883);
+        m_mqtt.setCleanSession(false);
+    }
+
+    @After
+    public void tearDown() throws Exception {
+        if (m_subscriber != null) {
+            m_subscriber.disconnect();
+        }
+        
+        if (m_publisher != null) {
+            m_publisher.disconnect();
+        }
+
+        m_server.stopServer();
+        IntegrationUtils.cleanPersistenceFile(m_config);
+    }
+    
+    
+    @Test
+    public void checkRestartCleanSubscriptionTree() throws Exception {
+        m_mqtt.setClientId("Subscriber");
+        //subscribe to /topic
+        m_subscriber = m_mqtt.blockingConnection();
+        m_subscriber.connect();
+        Topic[] topics = new Topic[]{new Topic("/topic", QoS.AT_LEAST_ONCE)};
+        m_subscriber.subscribe(topics);
+        m_subscriber.disconnect();
+        
+        //shutdown the server
+        m_server.stopServer();
+        IntegrationUtils.cleanPersistenceFile(m_config);
+
+        //restart the server
+        m_server.startServer(IntegrationUtils.prepareTestPropeties());
+        
+        //reconnect the Subscriber subscribing to the same /topic but different QoS
+        m_subscriber = m_mqtt.blockingConnection();
+        m_subscriber.connect();
+        topics = new Topic[]{new Topic("/topic", QoS.EXACTLY_ONCE)};
+        m_subscriber.subscribe(topics);
+        
+        //should be just one registration so a publisher receive one notification
+        MQTT mqtt = new MQTT();
+        mqtt.setHost("localhost", 1883); 
+        mqtt.setClientId("Publisher");
+        m_publisher = mqtt.blockingConnection();
+        m_publisher.connect();
+        m_publisher.publish("/topic", "Hello world MQTT!!".getBytes(), QoS.AT_LEAST_ONCE, false);
+        
+        //read the messages
+        Message msg = m_subscriber.receive();
+        msg.ack();
+        assertEquals("Hello world MQTT!!", new String(msg.getPayload()));
+        //no more messages on the same topic will be received
+        assertNull(m_subscriber.receive(1, TimeUnit.SECONDS));
+    }
+
+
+    @Test
+    public void checkDontPublishInactiveClientsAfterServerRestart() throws Exception {
+        m_mqtt.setClientId("SubPub");
+        BlockingConnection conn = subscribeAndPublish("/topic");
+        conn.disconnect();
+
+        //shutdown the server
+        m_server.stopServer();
+
+        //restart the server
+        m_server.startServer(IntegrationUtils.prepareTestPropeties());
+
+        MQTT mqtt = new MQTT();
+        mqtt.setHost("localhost", 1883);
+        mqtt.setClientId("Publisher");
+        m_publisher = mqtt.blockingConnection();
+        m_publisher.connect();
+        m_publisher.publish("/topic", "Hello world MQTT!!".getBytes(), QoS.AT_MOST_ONCE, false);
+    }
+
+    @Test
+    public void testClientDoesntRemainSubscribedAfterASubscriptionAndServerRestart() throws Exception {
+        m_mqtt.setClientId("Subscriber");
+        //subscribe to /topic
+        m_subscriber = m_mqtt.blockingConnection();
+        m_subscriber.connect();
+        Topic[] topics = new Topic[]{new Topic("/topic", QoS.AT_MOST_ONCE)};
+        //subscribe /topic
+        m_subscriber.subscribe(topics);
+        //unsubscribe from /topic
+        m_subscriber.unsubscribe(new String[]{"/topic"});
+        m_subscriber.disconnect();
+
+        //shutdown the server
+        m_server.stopServer();
+
+        //restart the server
+        m_server.startServer(IntegrationUtils.prepareTestPropeties());
+        //subscriber reconnects
+        MQTT mqttSub = new MQTT();
+        mqttSub.setHost("localhost", 1883);
+        mqttSub.setClientId("Subscriber");
+        m_subscriber = mqttSub.blockingConnection();
+        m_subscriber.connect();
+
+        //publisher publishes on /topic
+        MQTT mqtt = new MQTT();
+        mqtt.setHost("localhost", 1883);
+        mqtt.setClientId("Publisher");
+        m_publisher = mqtt.blockingConnection();
+        m_publisher.connect();
+        m_publisher.publish("/topic", "Hello world MQTT!!".getBytes(), QoS.AT_LEAST_ONCE, false);
+
+        //Expected
+        //the subscriber doesn't get notified (it's fully unsubscribed)
+        assertNull(m_subscriber.receive(1, TimeUnit.SECONDS));
+    }
+
+    /**
+     * Connect subscribe to topic and publish on the same topic
+     * */
+    private BlockingConnection subscribeAndPublish(String topic) throws Exception {
+        BlockingConnection conn = m_mqtt.blockingConnection();
+        conn.connect();
+        Topic[] topics = new Topic[]{new Topic(topic, QoS.AT_MOST_ONCE)};
+        conn.subscribe(topics);
+        conn.publish(topic, "Hello world MQTT!!".getBytes(), QoS.AT_MOST_ONCE, false);
+        //read the message
+        Message msg = conn.receive();
+        msg.ack();
+        assertEquals("Hello world MQTT!!", new String(msg.getPayload()));
+        return conn;
+    }
+}