| /* |
| * 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.qpid.jms.integration; |
| |
| import static org.hamcrest.Matchers.equalTo; |
| import static org.junit.Assert.assertEquals; |
| import static org.junit.Assert.assertFalse; |
| import static org.junit.Assert.assertNotNull; |
| import static org.junit.Assert.assertNull; |
| import static org.junit.Assert.assertTrue; |
| |
| import java.util.Date; |
| import java.util.concurrent.CountDownLatch; |
| import java.util.concurrent.TimeUnit; |
| import java.util.concurrent.atomic.AtomicBoolean; |
| |
| import javax.jms.Connection; |
| import javax.jms.Message; |
| import javax.jms.MessageConsumer; |
| import javax.jms.MessageListener; |
| import javax.jms.Queue; |
| import javax.jms.Session; |
| import javax.jms.TextMessage; |
| |
| import org.apache.qpid.jms.policy.JmsDefaultPrefetchPolicy; |
| import org.apache.qpid.jms.test.QpidJmsTestCase; |
| import org.apache.qpid.jms.test.testpeer.TestAmqpPeer; |
| import org.apache.qpid.jms.test.testpeer.describedtypes.sections.AmqpValueDescribedType; |
| import org.apache.qpid.jms.test.testpeer.describedtypes.sections.PropertiesDescribedType; |
| import org.apache.qpid.jms.test.testpeer.matchers.AcceptedMatcher; |
| import org.apache.qpid.jms.test.testpeer.matchers.ModifiedMatcher; |
| import org.apache.qpid.proton.amqp.UnsignedInteger; |
| import org.junit.Test; |
| import org.slf4j.Logger; |
| import org.slf4j.LoggerFactory; |
| |
| public class MessageExpirationIntegrationTest extends QpidJmsTestCase { |
| private static final Logger LOG = LoggerFactory.getLogger(MessageExpirationIntegrationTest.class); |
| |
| private final IntegrationTestFixture testFixture = new IntegrationTestFixture(); |
| |
| @Test(timeout=20000) |
| public void testIncomingExpiredMessageGetsFiltered() throws Exception { |
| try (TestAmqpPeer testPeer = new TestAmqpPeer();) { |
| Connection connection = testFixture.establishConnecton(testPeer); |
| connection.start(); |
| |
| testPeer.expectBegin(); |
| |
| Session session = connection.createSession(false, Session.AUTO_ACKNOWLEDGE); |
| Queue queue = session.createQueue("myQueue"); |
| |
| // Expected the consumer to attach and send credit, then send it an |
| // already-expired message followed by a live message. |
| testPeer.expectReceiverAttach(); |
| |
| PropertiesDescribedType props = new PropertiesDescribedType(); |
| props.setAbsoluteExpiryTime(new Date(System.currentTimeMillis() - 100)); |
| testPeer.expectLinkFlowRespondWithTransfer(null, null, props, null, new AmqpValueDescribedType("already-expired")); |
| |
| String liveMsgContent = "valid"; |
| testPeer.sendTransferToLastOpenedLinkOnLastOpenedSession(null, null, null, null, new AmqpValueDescribedType(liveMsgContent), 2); |
| |
| final MessageConsumer consumer = session.createConsumer(queue); |
| |
| // Call receive, expect the first message to be filtered due to expiry, |
| // and the second message to be given to the test app and accepted. |
| ModifiedMatcher modified = new ModifiedMatcher(); |
| modified.withDeliveryFailed(equalTo(true)); |
| modified.withUndeliverableHere(equalTo(true)); |
| |
| testPeer.expectDisposition(true, modified, 1, 1); |
| testPeer.expectDisposition(true, new AcceptedMatcher(), 2, 2); |
| |
| Message m = consumer.receive(3000); |
| assertNotNull("Message should have been received", m); |
| assertTrue(m instanceof TextMessage); |
| assertEquals("Unexpected message content", liveMsgContent, ((TextMessage)m).getText()); |
| |
| // Verify the other message is not there. Will drain to be sure there are no messages. |
| testPeer.expectLinkFlow(true, true, equalTo(UnsignedInteger.valueOf(JmsDefaultPrefetchPolicy.DEFAULT_QUEUE_PREFETCH - 2))); |
| // Then reopen the credit window afterwards |
| testPeer.expectLinkFlow(false, false, equalTo(UnsignedInteger.valueOf(JmsDefaultPrefetchPolicy.DEFAULT_QUEUE_PREFETCH))); |
| |
| m = consumer.receive(10); |
| assertNull("Message should not have been received", m); |
| |
| testPeer.expectClose(); |
| connection.close(); |
| |
| testPeer.waitForAllHandlersToComplete(3000); |
| } |
| } |
| |
| @Test(timeout=20000) |
| public void testIncomingExpiredMessageGetsConsumedWhenFilterDisabled() throws Exception { |
| try (TestAmqpPeer testPeer = new TestAmqpPeer();) { |
| Connection connection = testFixture.establishConnecton(testPeer, "?jms.localMessageExpiry=false"); |
| connection.start(); |
| |
| testPeer.expectBegin(); |
| |
| Session session = connection.createSession(false, Session.AUTO_ACKNOWLEDGE); |
| Queue queue = session.createQueue("myQueue"); |
| |
| // Expected the consumer to attach and send credit, then send it an |
| // already-expired message followed by a live message. |
| testPeer.expectReceiverAttach(); |
| |
| String expiredMsgContent = "already-expired"; |
| PropertiesDescribedType props = new PropertiesDescribedType(); |
| props.setAbsoluteExpiryTime(new Date(System.currentTimeMillis() - 100)); |
| testPeer.expectLinkFlowRespondWithTransfer(null, null, props, null, new AmqpValueDescribedType(expiredMsgContent)); |
| |
| String liveMsgContent = "valid"; |
| testPeer.sendTransferToLastOpenedLinkOnLastOpenedSession(null, null, null, null, new AmqpValueDescribedType(liveMsgContent), 2); |
| |
| final MessageConsumer consumer = session.createConsumer(queue); |
| |
| // Call receive, expect the expired message as we disabled local expiry. |
| testPeer.expectDisposition(true, new AcceptedMatcher(), 1, 1); |
| |
| Message m = consumer.receive(3000); |
| assertNotNull("Message should have been received", m); |
| assertTrue(m instanceof TextMessage); |
| assertEquals("Unexpected message content", expiredMsgContent, ((TextMessage)m).getText()); |
| |
| // Verify the other message is there |
| testPeer.expectDisposition(true, new AcceptedMatcher(), 2, 2); |
| |
| m = consumer.receive(3000); |
| assertNotNull("Message should have been received", m); |
| assertTrue(m instanceof TextMessage); |
| assertEquals("Unexpected message content", liveMsgContent, ((TextMessage)m).getText()); |
| |
| testPeer.expectClose(); |
| connection.close(); |
| |
| testPeer.waitForAllHandlersToComplete(3000); |
| } |
| } |
| |
| @Test(timeout=20000) |
| public void testIncomingExpiredMessageGetsFilteredAsync() throws Exception { |
| try (TestAmqpPeer testPeer = new TestAmqpPeer();) { |
| Connection connection = testFixture.establishConnecton(testPeer); |
| connection.start(); |
| |
| testPeer.expectBegin(); |
| |
| Session session = connection.createSession(false, Session.AUTO_ACKNOWLEDGE); |
| Queue queue = session.createQueue("myQueue"); |
| |
| // Expected the consumer to attach and send credit, then send it an |
| // already-expired message followed by a live message. |
| testPeer.expectReceiverAttach(); |
| |
| String expiredMessageContent = "already-expired"; |
| PropertiesDescribedType props = new PropertiesDescribedType(); |
| props.setAbsoluteExpiryTime(new Date(System.currentTimeMillis() - 100)); |
| testPeer.expectLinkFlowRespondWithTransfer(null, null, props, null, new AmqpValueDescribedType(expiredMessageContent)); |
| |
| final String liveMsgContent = "valid"; |
| testPeer.sendTransferToLastOpenedLinkOnLastOpenedSession(null, null, null, null, new AmqpValueDescribedType(liveMsgContent), 2); |
| |
| final MessageConsumer consumer = session.createConsumer(queue); |
| |
| // Add message listener, expect the first message to be filtered due to expiry, |
| // and the second message to be given to the test app and accepted. |
| ModifiedMatcher modified = new ModifiedMatcher(); |
| modified.withDeliveryFailed(equalTo(true)); |
| modified.withUndeliverableHere(equalTo(true)); |
| |
| testPeer.expectDisposition(true, modified, 1, 1); |
| testPeer.expectDisposition(true, new AcceptedMatcher(), 2, 2); |
| |
| final CountDownLatch success = new CountDownLatch(1); |
| final AtomicBoolean listenerFailure = new AtomicBoolean(); |
| consumer.setMessageListener(new MessageListener() { |
| |
| @Override |
| public void onMessage(Message incoming) { |
| try { |
| TextMessage textMessage = (TextMessage) incoming; |
| if (liveMsgContent.equals(textMessage.getText())) { |
| success.countDown(); |
| } else { |
| listenerFailure.set(true); |
| LOG.error("Received unexpected message:" + incoming); |
| } |
| } catch (Exception e) { |
| listenerFailure.set(true); |
| LOG.error("Exception in listener", e); |
| } |
| } |
| }); |
| |
| assertTrue("didn't get expected message", success.await(5, TimeUnit.SECONDS)); |
| assertFalse("There was a failure in the listener, see logs", listenerFailure.get()); |
| |
| testPeer.waitForAllHandlersToComplete(3000); |
| |
| testPeer.expectClose(); |
| connection.close(); |
| |
| testPeer.waitForAllHandlersToComplete(3000); |
| } |
| } |
| |
| @Test(timeout=20000) |
| public void testIncomingExpiredMessageGetsConsumedWhenFilterDisabledAsync() throws Exception { |
| try (TestAmqpPeer testPeer = new TestAmqpPeer();) { |
| Connection connection = testFixture.establishConnecton(testPeer, "?jms.localMessageExpiry=false"); |
| connection.start(); |
| |
| testPeer.expectBegin(); |
| |
| Session session = connection.createSession(false, Session.AUTO_ACKNOWLEDGE); |
| Queue queue = session.createQueue("myQueue"); |
| |
| // Expected the consumer to attach and send credit, then send it an |
| // already-expired message followed by a live message. |
| testPeer.expectReceiverAttach(); |
| |
| final String expiredMessageContent = "already-expired"; |
| PropertiesDescribedType props = new PropertiesDescribedType(); |
| props.setAbsoluteExpiryTime(new Date(System.currentTimeMillis() - 100)); |
| testPeer.expectLinkFlowRespondWithTransfer(null, null, props, null, new AmqpValueDescribedType(expiredMessageContent)); |
| |
| final String liveMsgContent = "valid"; |
| testPeer.sendTransferToLastOpenedLinkOnLastOpenedSession(null, null, null, null, new AmqpValueDescribedType(liveMsgContent), 2); |
| |
| final MessageConsumer consumer = session.createConsumer(queue); |
| |
| // Add message listener, expect both messages as the filter is disabled |
| testPeer.expectDisposition(true, new AcceptedMatcher(), 1, 1); |
| testPeer.expectDisposition(true, new AcceptedMatcher(), 2, 2); |
| |
| final CountDownLatch success = new CountDownLatch(2); |
| final AtomicBoolean listenerFailure = new AtomicBoolean(); |
| consumer.setMessageListener(new MessageListener() { |
| |
| @Override |
| public void onMessage(Message incoming) { |
| try { |
| TextMessage textMessage = (TextMessage) incoming; |
| if (expiredMessageContent.equals(textMessage.getText()) || liveMsgContent.equals(textMessage.getText())) { |
| success.countDown(); |
| } else { |
| listenerFailure.set(true); |
| LOG.error("Received unexpected message:" + incoming); |
| } |
| } catch (Exception e) { |
| listenerFailure.set(true); |
| LOG.error("Exception in listener", e); |
| } |
| } |
| }); |
| |
| assertTrue("didn't get expected messages", success.await(5, TimeUnit.SECONDS)); |
| assertFalse("There was a failure in the listener, see logs", listenerFailure.get()); |
| |
| testPeer.waitForAllHandlersToComplete(3000); |
| |
| testPeer.expectClose(); |
| connection.close(); |
| |
| testPeer.waitForAllHandlersToComplete(3000); |
| } |
| } |
| } |