anna/IM-server-new
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Minor, rewritten test to user Paho instead of fusesource mqtt client

Browse Source
This commit is contained in:
andsel
2016-03-06 19:50:47 +01:00
parent efbfab7914
commit 6cadaebb0c
3 changed files with 71 additions and 79 deletions
Download Patch File Download Diff File Expand all files Collapse all files
broker/src/test/java/io/moquette/server/ServerRestartIntegrationTest.java
+56 -62
View File
@@ -17,13 +17,16 @@ package io.moquette.server;
import io.moquette.server.config.IConfig;
import io.moquette.server.config.MemoryConfig;
import org.fusesource.mqtt.client.*;
import org.eclipse.paho.client.mqttv3.*;
import org.eclipse.paho.client.mqttv3.persist.MqttDefaultFilePersistence;
import org.junit.After;
import org.junit.Before;
import org.junit.BeforeClass;
import org.junit.Test;
import java.io.File;
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;
@@ -33,12 +36,16 @@ import static org.junit.Assert.assertNull;
*/
public class ServerRestartIntegrationTest {
static MqttClientPersistence s_dataStore;
static MqttClientPersistence s_pubDataStore;
static MqttConnectOptions CLEAN_SESSION_OPT = new MqttConnectOptions();
Server m_server;
MQTT m_mqtt;
BlockingConnection m_subscriber;
BlockingConnection m_publisher;
IMqttClient m_subscriber;
IMqttClient m_publisher;
IConfig m_config;
MessageCollector m_messageCollector;
protected void startServer() throws IOException {
m_server = new Server();
final Properties configProps = IntegrationUtils.prepareTestPropeties();
@@ -46,23 +53,34 @@ public class ServerRestartIntegrationTest {
m_server.startServer(m_config);
}
@BeforeClass
public static void beforeTests() {
String tmpDir = System.getProperty("java.io.tmpdir");
s_dataStore = new MqttDefaultFilePersistence(tmpDir);
s_pubDataStore = new MqttDefaultFilePersistence(tmpDir + File.separator + "publisher");
CLEAN_SESSION_OPT.setCleanSession(false);
}
@Before
public void setUp() throws Exception {
startServer();
String dbPath = IntegrationUtils.localMapDBPath();
IntegrationUtils.cleanPersistenceFile(dbPath);
m_mqtt = new MQTT();
m_subscriber = new MqttClient("tcp://localhost:1883", "Subscriber", s_dataStore);
m_messageCollector = new MessageCollector();
m_subscriber.setCallback(m_messageCollector);
m_mqtt.setHost("localhost", 1883);
m_mqtt.setCleanSession(false);
m_publisher = new MqttClient("tcp://localhost:1883", "Publisher", s_pubDataStore);
}
@After
public void tearDown() throws Exception {
if (m_subscriber != null) {
if (m_subscriber != null && m_subscriber.isConnected()) {
m_subscriber.disconnect();
}
if (m_publisher != null) {
if (m_publisher != null && m_publisher.isConnected()) {
m_publisher.disconnect();
}
@@ -73,12 +91,9 @@ public class ServerRestartIntegrationTest {
@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.connect(CLEAN_SESSION_OPT);
m_subscriber.subscribe("/topic", 1);
m_subscriber.disconnect();
//shutdown the server
@@ -89,32 +104,24 @@ public class ServerRestartIntegrationTest {
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);
m_subscriber.connect(CLEAN_SESSION_OPT);
m_subscriber.subscribe("/topic", 2);
//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);
m_publisher.connect(CLEAN_SESSION_OPT);
m_publisher.publish("/topic", "Hello world MQTT!!".getBytes(), 1, false);
//read the messages
Message msg = m_subscriber.receive();
msg.ack();
MqttMessage msg = m_messageCollector.getMessage(true);
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));
assertNull(m_messageCollector.getMessage(1));
}
@Test
public void checkDontPublishInactiveClientsAfterServerRestart() throws Exception {
m_mqtt.setClientId("SubPub");
BlockingConnection conn = subscribeAndPublish("/topic");
IMqttClient conn = subscribeAndPublish("/topic");
conn.disconnect();
//shutdown the server
@@ -123,25 +130,18 @@ public class ServerRestartIntegrationTest {
//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);
m_publisher.publish("/topic", "Hello world MQTT!!".getBytes(), 0, 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);
m_subscriber.subscribe("/topic", 0);
//unsubscribe from /topic
m_subscriber.unsubscribe(new String[]{"/topic"});
m_subscriber.unsubscribe("/topic");
m_subscriber.disconnect();
//shutdown the server
@@ -150,38 +150,32 @@ public class ServerRestartIntegrationTest {
//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 = new MqttClient("tcp://localhost:1883", "Subscriber", s_dataStore);
m_subscriber.setCallback(m_messageCollector);
m_subscriber.connect();
//publisher publishes on /topic
MQTT mqtt = new MQTT();
mqtt.setHost("localhost", 1883);
mqtt.setClientId("Publisher");
m_publisher = mqtt.blockingConnection();
m_publisher = new MqttClient("tcp://localhost:1883", "Publisher", s_pubDataStore);
m_publisher.connect();
m_publisher.publish("/topic", "Hello world MQTT!!".getBytes(), QoS.AT_LEAST_ONCE, false);
m_publisher.publish("/topic", "Hello world MQTT!!".getBytes(), 1, false);
//Expected
//the subscriber doesn't get notified (it's fully unsubscribed)
assertNull(m_subscriber.receive(1, TimeUnit.SECONDS));
assertNull(m_messageCollector.getMessage(1));
}
/**
* 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();
private IMqttClient subscribeAndPublish(String topic) throws Exception {
IMqttClient client = new MqttClient("tcp://localhost:1883", "SubPub");
MessageCollector collector = new MessageCollector();
client.setCallback(collector);
client.connect();
client.subscribe(topic, 1);
client.publish(topic, "Hello world MQTT!!".getBytes(), 0, false);
MqttMessage msg = collector.getMessage(true);
assertEquals("Hello world MQTT!!", new String(msg.getPayload()));
return conn;
return client;
}
}