@Test public void testIsForMessage() throws Exception { DisconnectMessage message = new DisconnectMessage(); assertFalse(disconnect.isForMessage(message)); message.getHeader().setReceipt("blah"); assertTrue(disconnect.isForMessage(message)); ReceiptMessage receipt = new ReceiptMessage("blah"); assertFalse(disconnect.isForMessage(receipt)); disconnect.interceptMessage(message, hostPort); assertTrue(disconnect.isForMessage(receipt)); }
@Test(expected = MessageListenerHaltException.class) public void testMessageReceived() throws Exception { disconnect.setGateway(clientGateway); disconnect.setReceiptId("test"); ReceiptMessage message = new ReceiptMessage("blah"); disconnect.setCloseOnDisconnectMessage(false); disconnect.messageReceived(message, hostPort); disconnect.setReceiptId("test"); disconnect.setCloseOnDisconnectMessage(true); disconnect.messageReceived(message, hostPort); }
@After public void after() throws Exception { disconnect.setReceiptId(null); }