Removed dependency on Mina IO framework
This commit is contained in:
andrea
+232
-235
@@ -1,14 +1,12 @@
|
||||
package org.dna.mqtt.moquette.messaging.spi.impl;
|
||||
|
||||
import java.util.Arrays;
|
||||
import java.util.HashMap;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
import org.apache.mina.core.filterchain.IoFilter;
|
||||
import org.apache.mina.core.filterchain.IoFilterAdapter;
|
||||
import org.apache.mina.core.session.DummySession;
|
||||
import org.apache.mina.core.session.IoSession;
|
||||
import org.apache.mina.core.write.WriteRequest;
|
||||
//import org.apache.mina.core.filterchain.IoFilter;
|
||||
//import org.apache.mina.core.filterchain.IoFilterAdapter;
|
||||
//import org.apache.mina.core.session.DummySession;
|
||||
//import org.apache.mina.core.session.IoSession;
|
||||
//import org.apache.mina.core.write.WriteRequest;
|
||||
import org.dna.mqtt.moquette.messaging.spi.IStorageService;
|
||||
import org.dna.mqtt.moquette.messaging.spi.impl.events.PublishEvent;
|
||||
import org.dna.mqtt.moquette.messaging.spi.impl.subscriptions.Subscription;
|
||||
@@ -20,9 +18,8 @@ import org.dna.mqtt.moquette.proto.messages.ConnectMessage;
|
||||
import org.dna.mqtt.moquette.proto.messages.PublishMessage;
|
||||
import org.dna.mqtt.moquette.proto.messages.SubAckMessage;
|
||||
import org.dna.mqtt.moquette.proto.messages.SubscribeMessage;
|
||||
import org.dna.mqtt.moquette.server.ConnectionDescriptor;
|
||||
import org.dna.mqtt.moquette.server.ServerChannel;
|
||||
import org.dna.mqtt.moquette.server.mina.MinaChannel;
|
||||
//import org.dna.mqtt.moquette.server.mina.MinaChannel;
|
||||
import static org.junit.Assert.assertEquals;
|
||||
import static org.junit.Assert.assertNotNull;
|
||||
import static org.junit.Assert.assertTrue;
|
||||
@@ -34,231 +31,231 @@ import org.junit.Test;
|
||||
* @author andrea
|
||||
*/
|
||||
public class ProtocolProcessorTest {
|
||||
final static String FAKE_CLIENT_ID = "FAKE_123";
|
||||
final static String FAKE_TOPIC = "/news";
|
||||
|
||||
ServerChannel m_session;
|
||||
byte m_returnCode;
|
||||
ConnectMessage connMsg;
|
||||
ProtocolProcessor m_processor;
|
||||
|
||||
IStorageService m_storageService;
|
||||
SubscriptionsStore subscriptions;
|
||||
AbstractMessage m_receivedMessage;
|
||||
|
||||
IoSession m_minaSession;
|
||||
|
||||
|
||||
@Before
|
||||
public void setUp() throws InterruptedException {
|
||||
connMsg = new ConnectMessage();
|
||||
connMsg.setProcotolVersion((byte) 0x03);
|
||||
|
||||
m_minaSession = new DummySession();
|
||||
|
||||
m_session = new MinaChannel(m_minaSession);
|
||||
|
||||
m_minaSession.getFilterChain().addFirst("MessageCatcher", new IoFilterAdapter() {
|
||||
|
||||
@Override
|
||||
public void filterWrite(IoFilter.NextFilter nextFilter, IoSession session,
|
||||
WriteRequest writeRequest) throws Exception {
|
||||
try {
|
||||
m_receivedMessage = (AbstractMessage) writeRequest.getMessage();
|
||||
if (m_receivedMessage instanceof ConnAckMessage) {
|
||||
ConnAckMessage buf = (ConnAckMessage) m_receivedMessage;
|
||||
m_returnCode = buf.getReturnCode();
|
||||
}
|
||||
} catch (Exception ex) {
|
||||
throw new AssertionError("Wrong return code");
|
||||
}
|
||||
}
|
||||
});
|
||||
|
||||
//sleep to let the messaging batch processor to process the initEvent
|
||||
Thread.sleep(300);
|
||||
|
||||
m_storageService = new MemoryStorageService();
|
||||
//m_storageService.initStore();
|
||||
|
||||
subscriptions = new SubscriptionsStore();
|
||||
subscriptions.init(m_storageService);
|
||||
m_processor = new ProtocolProcessor();
|
||||
m_processor.init(subscriptions, m_storageService);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testHandleConnect_BadProtocol() {
|
||||
connMsg.setProcotolVersion((byte) 0x02);
|
||||
|
||||
//Exercise
|
||||
m_processor.processConnect(m_session, connMsg);
|
||||
|
||||
//Verify
|
||||
assertEquals(ConnAckMessage.UNNACEPTABLE_PROTOCOL_VERSION, m_returnCode);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testConnect_badClientID() {
|
||||
connMsg.setClientID("extremely_long_clientID_greater_than_23");
|
||||
|
||||
//Exercise
|
||||
m_processor.processConnect(m_session, connMsg);
|
||||
|
||||
//Verify
|
||||
assertEquals(ConnAckMessage.IDENTIFIER_REJECTED, m_returnCode);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testWill() {
|
||||
connMsg.setClientID("123");
|
||||
connMsg.setWillFlag(true);
|
||||
connMsg.setWillTopic("topic");
|
||||
connMsg.setWillMessage("Topic message");
|
||||
|
||||
|
||||
//Exercise
|
||||
//m_handler.setMessaging(mockedMessaging);
|
||||
m_processor.processConnect(m_session, connMsg);
|
||||
|
||||
//Verify
|
||||
assertEquals(ConnAckMessage.CONNECTION_ACCEPTED, m_returnCode);
|
||||
//TODO verify the call
|
||||
/*verify(mockedMessaging).publish(eq("topic"), eq("Topic message".getBytes()),
|
||||
any(AbstractMessage.QOSType.class), anyBoolean(), eq("123"), any(IoSession.class));*/
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testPublish() throws InterruptedException {
|
||||
final Subscription subscription = new Subscription(FAKE_CLIENT_ID,
|
||||
FAKE_TOPIC, AbstractMessage.QOSType.MOST_ONE, true);
|
||||
|
||||
//subscriptions.matches(topic) redefine the method to return true
|
||||
SubscriptionsStore subs = new SubscriptionsStore() {
|
||||
@Override
|
||||
public List<Subscription> matches(String topic) {
|
||||
if (topic.equals(FAKE_TOPIC)) {
|
||||
return Arrays.asList(subscription);
|
||||
} else {
|
||||
throw new IllegalArgumentException("Expected " + FAKE_TOPIC + " buf found " + topic);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
};
|
||||
|
||||
//simulate a connect that register a clientID to an IoSession
|
||||
subs.init(m_storageService);
|
||||
m_processor.init(subs, m_storageService);
|
||||
ConnectMessage connectMessage = new ConnectMessage();
|
||||
connectMessage.setProcotolVersion((byte)3);
|
||||
connectMessage.setClientID(FAKE_CLIENT_ID);
|
||||
connectMessage.setCleanSession(subscription.isCleanSession());
|
||||
m_processor.processConnect(m_session, connectMessage);
|
||||
|
||||
|
||||
//Exercise
|
||||
PublishEvent pubEvt = new PublishEvent(FAKE_TOPIC, AbstractMessage.QOSType.MOST_ONE, "Hello".getBytes(), false, "FakeCLI", null);
|
||||
m_processor.processPublish(pubEvt);
|
||||
|
||||
//Verify
|
||||
assertNotNull(m_receivedMessage);
|
||||
//TODO check received message attributes
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testSubscribe() {
|
||||
//Exercise
|
||||
SubscribeMessage msg = new SubscribeMessage();
|
||||
msg.addSubscription(new SubscribeMessage.Couple((byte)AbstractMessage.QOSType.MOST_ONE.ordinal(), FAKE_TOPIC));
|
||||
m_processor.processSubscribe(m_session, msg, FAKE_CLIENT_ID, false);
|
||||
|
||||
//Verify
|
||||
assertTrue(m_receivedMessage instanceof SubAckMessage);
|
||||
Subscription expectedSubscription = new Subscription(FAKE_CLIENT_ID, FAKE_TOPIC, AbstractMessage.QOSType.MOST_ONE, false);
|
||||
assertTrue(subscriptions.contains(expectedSubscription));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testDoubleSubscribe() {
|
||||
SubscribeMessage msg = new SubscribeMessage();
|
||||
msg.addSubscription(new SubscribeMessage.Couple((byte)AbstractMessage.QOSType.MOST_ONE.ordinal(), FAKE_TOPIC));
|
||||
|
||||
subscriptions.clearAllSubscriptions();
|
||||
assertEquals(0, subscriptions.size());
|
||||
|
||||
m_processor.processSubscribe(m_session, msg, FAKE_CLIENT_ID, false);
|
||||
|
||||
//Exercise
|
||||
m_processor.processSubscribe(m_session, msg, FAKE_CLIENT_ID, false);
|
||||
|
||||
//Verify
|
||||
assertEquals(1, subscriptions.size());
|
||||
Subscription expectedSubscription = new Subscription(FAKE_CLIENT_ID, FAKE_TOPIC, AbstractMessage.QOSType.MOST_ONE, false);
|
||||
assertTrue(subscriptions.contains(expectedSubscription));
|
||||
}
|
||||
|
||||
|
||||
@Test
|
||||
public void testPublishOfRetainedMessage_afterNewSubscription() throws Exception {
|
||||
m_minaSession.getFilterChain().remove("MessageCatcher");
|
||||
m_minaSession.getFilterChain().addFirst("MessageCatcher", new IoFilterAdapter() {
|
||||
|
||||
@Override
|
||||
public void filterWrite(IoFilter.NextFilter nextFilter, IoSession session,
|
||||
WriteRequest writeRequest) throws Exception {
|
||||
try {
|
||||
System.out.println("filterReceived class " + writeRequest.getMessage().getClass().getName());
|
||||
if (writeRequest.getMessage() instanceof PublishMessage) {
|
||||
m_receivedMessage = (AbstractMessage) writeRequest.getMessage();
|
||||
}
|
||||
|
||||
if (m_receivedMessage instanceof ConnAckMessage) {
|
||||
ConnAckMessage buf = (ConnAckMessage) m_receivedMessage;
|
||||
m_returnCode = buf.getReturnCode();
|
||||
}
|
||||
} catch (Exception ex) {
|
||||
throw new AssertionError("Wrong return code");
|
||||
}
|
||||
}
|
||||
});
|
||||
|
||||
// final static String FAKE_CLIENT_ID = "FAKE_123";
|
||||
// final static String FAKE_TOPIC = "/news";
|
||||
//
|
||||
// ServerChannel m_session;
|
||||
// byte m_returnCode;
|
||||
// ConnectMessage connMsg;
|
||||
// ProtocolProcessor m_processor;
|
||||
//
|
||||
// IStorageService m_storageService;
|
||||
// SubscriptionsStore subscriptions;
|
||||
// AbstractMessage m_receivedMessage;
|
||||
//
|
||||
// IoSession m_minaSession;
|
||||
//
|
||||
//
|
||||
// @Before
|
||||
// public void setUp() throws InterruptedException {
|
||||
// connMsg = new ConnectMessage();
|
||||
// connMsg.setProcotolVersion((byte) 0x03);
|
||||
//
|
||||
// m_minaSession = new DummySession();
|
||||
//
|
||||
// m_session = new MinaChannel(m_minaSession);
|
||||
//
|
||||
// m_minaSession.getFilterChain().addFirst("MessageCatcher", new IoFilterAdapter() {
|
||||
//
|
||||
// @Override
|
||||
// public void filterWrite(IoFilter.NextFilter nextFilter, IoSession session,
|
||||
// WriteRequest writeRequest) throws Exception {
|
||||
// try {
|
||||
// m_receivedMessage = (AbstractMessage) writeRequest.getMessage();
|
||||
// if (m_receivedMessage instanceof ConnAckMessage) {
|
||||
// ConnAckMessage buf = (ConnAckMessage) m_receivedMessage;
|
||||
// m_returnCode = buf.getReturnCode();
|
||||
// }
|
||||
// } catch (Exception ex) {
|
||||
// throw new AssertionError("Wrong return code");
|
||||
// }
|
||||
// }
|
||||
// });
|
||||
//
|
||||
// //sleep to let the messaging batch processor to process the initEvent
|
||||
// Thread.sleep(300);
|
||||
//
|
||||
// m_storageService = new MemoryStorageService();
|
||||
// //m_storageService.initStore();
|
||||
//
|
||||
// subscriptions = new SubscriptionsStore();
|
||||
// subscriptions.init(m_storageService);
|
||||
// m_processor = new ProtocolProcessor();
|
||||
// m_processor.init(subscriptions, m_storageService);
|
||||
// }
|
||||
//
|
||||
// @Test
|
||||
// public void testHandleConnect_BadProtocol() {
|
||||
// connMsg.setProcotolVersion((byte) 0x02);
|
||||
//
|
||||
// //Exercise
|
||||
// m_processor.processConnect(m_session, connMsg);
|
||||
//
|
||||
// //Verify
|
||||
// assertEquals(ConnAckMessage.UNNACEPTABLE_PROTOCOL_VERSION, m_returnCode);
|
||||
// }
|
||||
//
|
||||
// @Test
|
||||
// public void testConnect_badClientID() {
|
||||
// connMsg.setClientID("extremely_long_clientID_greater_than_23");
|
||||
//
|
||||
// //Exercise
|
||||
// m_processor.processConnect(m_session, connMsg);
|
||||
//
|
||||
// //Verify
|
||||
// assertEquals(ConnAckMessage.IDENTIFIER_REJECTED, m_returnCode);
|
||||
// }
|
||||
//
|
||||
// @Test
|
||||
// public void testWill() {
|
||||
// connMsg.setClientID("123");
|
||||
// connMsg.setWillFlag(true);
|
||||
// connMsg.setWillTopic("topic");
|
||||
// connMsg.setWillMessage("Topic message");
|
||||
//
|
||||
//
|
||||
// //Exercise
|
||||
// //m_handler.setMessaging(mockedMessaging);
|
||||
// m_processor.processConnect(m_session, connMsg);
|
||||
//
|
||||
// //Verify
|
||||
// assertEquals(ConnAckMessage.CONNECTION_ACCEPTED, m_returnCode);
|
||||
// //TODO verify the call
|
||||
// /*verify(mockedMessaging).publish(eq("topic"), eq("Topic message".getBytes()),
|
||||
// any(AbstractMessage.QOSType.class), anyBoolean(), eq("123"), any(IoSession.class));*/
|
||||
// }
|
||||
//
|
||||
// @Test
|
||||
// public void testPublish() throws InterruptedException {
|
||||
// final Subscription subscription = new Subscription(FAKE_CLIENT_ID,
|
||||
// FAKE_TOPIC, AbstractMessage.QOSType.MOST_ONE, true);
|
||||
//
|
||||
// //subscriptions.matches(topic) redefine the method to return true
|
||||
// SubscriptionsStore subs = new SubscriptionsStore() {
|
||||
// @Override
|
||||
// public List<Subscription> matches(String topic) {
|
||||
// if (topic.equals(FAKE_TOPIC)) {
|
||||
// return Arrays.asList(subscription);
|
||||
// } else {
|
||||
// throw new IllegalArgumentException("Expected " + FAKE_TOPIC + " buf found " + topic);
|
||||
// }
|
||||
// }
|
||||
//
|
||||
//
|
||||
// };
|
||||
//
|
||||
// //simulate a connect that register a clientID to an IoSession
|
||||
final Subscription subscription = new Subscription(FAKE_CLIENT_ID,
|
||||
FAKE_TOPIC, AbstractMessage.QOSType.MOST_ONE, true);
|
||||
|
||||
//subscriptions.matches(topic) redefine the method to return true
|
||||
SubscriptionsStore subs = new SubscriptionsStore() {
|
||||
@Override
|
||||
public List<Subscription> matches(String topic) {
|
||||
if (topic.equals(FAKE_TOPIC)) {
|
||||
return Arrays.asList(subscription);
|
||||
} else {
|
||||
throw new IllegalArgumentException("Expected " + FAKE_TOPIC + " buf found " + topic);
|
||||
}
|
||||
}
|
||||
};
|
||||
subs.init(new MemoryStorageService());
|
||||
|
||||
//simulate a connect that register a clientID to an IoSession
|
||||
m_processor.init(subs, m_storageService);
|
||||
ConnectMessage connectMessage = new ConnectMessage();
|
||||
connectMessage.setClientID(FAKE_CLIENT_ID);
|
||||
connectMessage.setProcotolVersion((byte)3);
|
||||
connectMessage.setCleanSession(subscription.isCleanSession());
|
||||
m_processor.processConnect(m_session, connectMessage);
|
||||
PublishEvent pubEvt = new PublishEvent(FAKE_TOPIC, AbstractMessage.QOSType.MOST_ONE, "Hello".getBytes(), true, "FakeCLI", null);
|
||||
m_processor.processPublish(pubEvt);
|
||||
|
||||
//Exercise
|
||||
SubscribeMessage msg = new SubscribeMessage();
|
||||
msg.addSubscription(new SubscribeMessage.Couple((byte)QOSType.MOST_ONE.ordinal(), "#"));
|
||||
m_processor.processSubscribe(m_session, msg, FAKE_CLIENT_ID, false);
|
||||
|
||||
//Verify
|
||||
assertNotNull(m_receivedMessage);
|
||||
assertTrue(m_receivedMessage instanceof PublishMessage);
|
||||
PublishMessage pubMessage = (PublishMessage) m_receivedMessage;
|
||||
assertEquals(FAKE_TOPIC, pubMessage.getTopicName());
|
||||
}
|
||||
// subs.init(m_storageService);
|
||||
// m_processor.init(subs, m_storageService);
|
||||
// ConnectMessage connectMessage = new ConnectMessage();
|
||||
// connectMessage.setProcotolVersion((byte)3);
|
||||
// connectMessage.setClientID(FAKE_CLIENT_ID);
|
||||
// connectMessage.setCleanSession(subscription.isCleanSession());
|
||||
// m_processor.processConnect(m_session, connectMessage);
|
||||
//
|
||||
//
|
||||
// //Exercise
|
||||
// PublishEvent pubEvt = new PublishEvent(FAKE_TOPIC, AbstractMessage.QOSType.MOST_ONE, "Hello".getBytes(), false, "FakeCLI", null);
|
||||
// m_processor.processPublish(pubEvt);
|
||||
//
|
||||
// //Verify
|
||||
// assertNotNull(m_receivedMessage);
|
||||
// //TODO check received message attributes
|
||||
// }
|
||||
//
|
||||
// @Test
|
||||
// public void testSubscribe() {
|
||||
// //Exercise
|
||||
// SubscribeMessage msg = new SubscribeMessage();
|
||||
// msg.addSubscription(new SubscribeMessage.Couple((byte)AbstractMessage.QOSType.MOST_ONE.ordinal(), FAKE_TOPIC));
|
||||
// m_processor.processSubscribe(m_session, msg, FAKE_CLIENT_ID, false);
|
||||
//
|
||||
// //Verify
|
||||
// assertTrue(m_receivedMessage instanceof SubAckMessage);
|
||||
// Subscription expectedSubscription = new Subscription(FAKE_CLIENT_ID, FAKE_TOPIC, AbstractMessage.QOSType.MOST_ONE, false);
|
||||
// assertTrue(subscriptions.contains(expectedSubscription));
|
||||
// }
|
||||
//
|
||||
// @Test
|
||||
// public void testDoubleSubscribe() {
|
||||
// SubscribeMessage msg = new SubscribeMessage();
|
||||
// msg.addSubscription(new SubscribeMessage.Couple((byte)AbstractMessage.QOSType.MOST_ONE.ordinal(), FAKE_TOPIC));
|
||||
//
|
||||
// subscriptions.clearAllSubscriptions();
|
||||
// assertEquals(0, subscriptions.size());
|
||||
//
|
||||
// m_processor.processSubscribe(m_session, msg, FAKE_CLIENT_ID, false);
|
||||
//
|
||||
// //Exercise
|
||||
// m_processor.processSubscribe(m_session, msg, FAKE_CLIENT_ID, false);
|
||||
//
|
||||
// //Verify
|
||||
// assertEquals(1, subscriptions.size());
|
||||
// Subscription expectedSubscription = new Subscription(FAKE_CLIENT_ID, FAKE_TOPIC, AbstractMessage.QOSType.MOST_ONE, false);
|
||||
// assertTrue(subscriptions.contains(expectedSubscription));
|
||||
// }
|
||||
//
|
||||
//
|
||||
// @Test
|
||||
// public void testPublishOfRetainedMessage_afterNewSubscription() throws Exception {
|
||||
// m_minaSession.getFilterChain().remove("MessageCatcher");
|
||||
// m_minaSession.getFilterChain().addFirst("MessageCatcher", new IoFilterAdapter() {
|
||||
//
|
||||
// @Override
|
||||
// public void filterWrite(IoFilter.NextFilter nextFilter, IoSession session,
|
||||
// WriteRequest writeRequest) throws Exception {
|
||||
// try {
|
||||
// System.out.println("filterReceived class " + writeRequest.getMessage().getClass().getName());
|
||||
// if (writeRequest.getMessage() instanceof PublishMessage) {
|
||||
// m_receivedMessage = (AbstractMessage) writeRequest.getMessage();
|
||||
// }
|
||||
//
|
||||
// if (m_receivedMessage instanceof ConnAckMessage) {
|
||||
// ConnAckMessage buf = (ConnAckMessage) m_receivedMessage;
|
||||
// m_returnCode = buf.getReturnCode();
|
||||
// }
|
||||
// } catch (Exception ex) {
|
||||
// throw new AssertionError("Wrong return code");
|
||||
// }
|
||||
// }
|
||||
// });
|
||||
//
|
||||
//// //simulate a connect that register a clientID to an IoSession
|
||||
// final Subscription subscription = new Subscription(FAKE_CLIENT_ID,
|
||||
// FAKE_TOPIC, AbstractMessage.QOSType.MOST_ONE, true);
|
||||
//
|
||||
// //subscriptions.matches(topic) redefine the method to return true
|
||||
// SubscriptionsStore subs = new SubscriptionsStore() {
|
||||
// @Override
|
||||
// public List<Subscription> matches(String topic) {
|
||||
// if (topic.equals(FAKE_TOPIC)) {
|
||||
// return Arrays.asList(subscription);
|
||||
// } else {
|
||||
// throw new IllegalArgumentException("Expected " + FAKE_TOPIC + " buf found " + topic);
|
||||
// }
|
||||
// }
|
||||
// };
|
||||
// subs.init(new MemoryStorageService());
|
||||
//
|
||||
// //simulate a connect that register a clientID to an IoSession
|
||||
// m_processor.init(subs, m_storageService);
|
||||
// ConnectMessage connectMessage = new ConnectMessage();
|
||||
// connectMessage.setClientID(FAKE_CLIENT_ID);
|
||||
// connectMessage.setProcotolVersion((byte)3);
|
||||
// connectMessage.setCleanSession(subscription.isCleanSession());
|
||||
// m_processor.processConnect(m_session, connectMessage);
|
||||
// PublishEvent pubEvt = new PublishEvent(FAKE_TOPIC, AbstractMessage.QOSType.MOST_ONE, "Hello".getBytes(), true, "FakeCLI", null);
|
||||
// m_processor.processPublish(pubEvt);
|
||||
//
|
||||
// //Exercise
|
||||
// SubscribeMessage msg = new SubscribeMessage();
|
||||
// msg.addSubscription(new SubscribeMessage.Couple((byte)QOSType.MOST_ONE.ordinal(), "#"));
|
||||
// m_processor.processSubscribe(m_session, msg, FAKE_CLIENT_ID, false);
|
||||
//
|
||||
// //Verify
|
||||
// assertNotNull(m_receivedMessage);
|
||||
// assertTrue(m_receivedMessage instanceof PublishMessage);
|
||||
// PublishMessage pubMessage = (PublishMessage) m_receivedMessage;
|
||||
// assertEquals(FAKE_TOPIC, pubMessage.getTopicName());
|
||||
// }
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user