| /* |
| * |
| * 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.server.queue; |
| |
| import junit.framework.TestCase; |
| import junit.framework.Assert; |
| import org.apache.log4j.Logger; |
| import org.apache.qpid.client.transport.TransportConnection; |
| import org.apache.qpid.client.AMQConnection; |
| import org.apache.qpid.client.AMQSession; |
| import org.apache.qpid.client.AMQQueue; |
| import org.apache.qpid.client.AMQDestination; |
| import org.apache.qpid.jndi.PropertiesFileInitialContextFactory; |
| import org.apache.qpid.url.URLSyntaxException; |
| import org.apache.qpid.AMQException; |
| import org.apache.qpid.framing.AMQShortString; |
| import org.apache.qpid.framing.FieldTable; |
| |
| import javax.jms.*; |
| import javax.naming.NamingException; |
| import javax.naming.Context; |
| import javax.naming.spi.InitialContextFactory; |
| import java.util.Hashtable; |
| import java.util.HashMap; |
| import java.util.Map; |
| |
| /** Test Case provided by client Non-functional Test NF101: heap exhaustion behaviour */ |
| public class PriorityTest extends TestCase |
| { |
| private static final int TIMEOUT = 1500; |
| |
| |
| private static final Logger _logger = Logger.getLogger(PriorityTest.class); |
| |
| |
| protected final String BROKER = "vm://:1"; |
| protected final String VHOST = "/test"; |
| protected final String QUEUE = "PriorityQueue"; |
| |
| private static final int MSG_COUNT = 50; |
| |
| private Context context = null; |
| private Connection producerConnection; |
| private MessageProducer producer; |
| private Session producerSession; |
| private Queue queue; |
| private Connection consumerConnection; |
| private Session consumerSession; |
| |
| |
| private MessageConsumer consumer; |
| |
| protected void setUp() throws Exception |
| { |
| super.setUp(); |
| |
| if (usingInVMBroker()) |
| { |
| TransportConnection.createVMBroker(1); |
| } |
| |
| InitialContextFactory factory = new PropertiesFileInitialContextFactory(); |
| Hashtable<String, String> env = new Hashtable<String, String>(); |
| |
| env.put("connectionfactory.connection", "amqp://guest:guest@PRIORITY_TEST_ID" + VHOST + "?brokerlist='" + BROKER + "'"); |
| env.put("queue.queue", QUEUE); |
| |
| context = factory.getInitialContext(env); |
| producerConnection = ((ConnectionFactory) context.lookup("connection")).createConnection(); |
| producerSession = producerConnection.createSession(true, Session.AUTO_ACKNOWLEDGE); |
| |
| producerConnection.start(); |
| |
| consumerConnection = ((ConnectionFactory) context.lookup("connection")).createConnection(); |
| consumerSession = consumerConnection.createSession(false, Session.AUTO_ACKNOWLEDGE); |
| |
| } |
| |
| private boolean usingInVMBroker() |
| { |
| return BROKER.startsWith("vm://"); |
| } |
| |
| protected void tearDown() throws Exception |
| { |
| producerConnection.close(); |
| consumerConnection.close(); |
| if (usingInVMBroker()) |
| { |
| TransportConnection.killAllVMBrokers(); |
| } |
| super.tearDown(); |
| } |
| |
| public void testPriority() throws JMSException, NamingException, AMQException |
| { |
| final Map<String,Object> arguments = new HashMap<String, Object>(); |
| arguments.put("x-qpid-priorities",10); |
| ((AMQSession) producerSession).createQueue(new AMQShortString(QUEUE), true, false, false, arguments); |
| queue = new AMQQueue("amq.direct",QUEUE); |
| ((AMQSession) producerSession).declareAndBind((AMQDestination)queue); |
| producer = producerSession.createProducer(queue); |
| |
| for (int msg = 0; msg < MSG_COUNT; msg++) |
| { |
| producer.setPriority(msg % 10); |
| producer.send(nextMessage(msg, false, producerSession, producer)); |
| } |
| producerSession.commit(); |
| producer.close(); |
| producerSession.close(); |
| producerConnection.close(); |
| |
| consumer = consumerSession.createConsumer(queue); |
| consumerConnection.start(); |
| Message received; |
| int receivedCount = 0; |
| Message previous = null; |
| int messageCount = 0; |
| while((received = consumer.receive(1000))!=null) |
| { |
| messageCount++; |
| if(previous != null) |
| { |
| assertTrue("Messages arrived in unexpected order " + messageCount + " " + previous.getIntProperty("msg") + " " + received.getIntProperty("msg") + " " + previous.getJMSPriority() + " " + received.getJMSPriority(), (previous.getJMSPriority() > received.getJMSPriority()) || ((previous.getJMSPriority() == received.getJMSPriority()) && previous.getIntProperty("msg") < received.getIntProperty("msg")) ); |
| } |
| |
| previous = received; |
| receivedCount++; |
| } |
| |
| assertEquals("Incorrect number of message received", 50, receivedCount); |
| } |
| |
| public void testOddOrdering() throws AMQException, JMSException |
| { |
| final Map<String,Object> arguments = new HashMap<String, Object>(); |
| arguments.put("x-qpid-priorities",3); |
| ((AMQSession) producerSession).createQueue(new AMQShortString(QUEUE), true, false, false, arguments); |
| queue = new AMQQueue("amq.direct",QUEUE); |
| ((AMQSession) producerSession).declareAndBind((AMQDestination)queue); |
| producer = producerSession.createProducer(queue); |
| |
| // In order ABC |
| producer.setPriority(9); |
| producer.send(nextMessage(1, false, producerSession, producer)); |
| producer.setPriority(4); |
| producer.send(nextMessage(2, false, producerSession, producer)); |
| producer.setPriority(1); |
| producer.send(nextMessage(3, false, producerSession, producer)); |
| |
| // Out of order BAC |
| producer.setPriority(4); |
| producer.send(nextMessage(4, false, producerSession, producer)); |
| producer.setPriority(9); |
| producer.send(nextMessage(5, false, producerSession, producer)); |
| producer.setPriority(1); |
| producer.send(nextMessage(6, false, producerSession, producer)); |
| |
| // Out of order BCA |
| producer.setPriority(4); |
| producer.send(nextMessage(7, false, producerSession, producer)); |
| producer.setPriority(1); |
| producer.send(nextMessage(8, false, producerSession, producer)); |
| producer.setPriority(9); |
| producer.send(nextMessage(9, false, producerSession, producer)); |
| |
| // Reverse order CBA |
| producer.setPriority(1); |
| producer.send(nextMessage(10, false, producerSession, producer)); |
| producer.setPriority(4); |
| producer.send(nextMessage(11, false, producerSession, producer)); |
| producer.setPriority(9); |
| producer.send(nextMessage(12, false, producerSession, producer)); |
| producerSession.commit(); |
| |
| consumer = consumerSession.createConsumer(queue); |
| consumerConnection.start(); |
| |
| Message msg = consumer.receive(TIMEOUT); |
| assertEquals(1, msg.getIntProperty("msg")); |
| msg = consumer.receive(TIMEOUT); |
| assertEquals(5, msg.getIntProperty("msg")); |
| msg = consumer.receive(TIMEOUT); |
| assertEquals(9, msg.getIntProperty("msg")); |
| msg = consumer.receive(TIMEOUT); |
| assertEquals(12, msg.getIntProperty("msg")); |
| |
| msg = consumer.receive(TIMEOUT); |
| assertEquals(2, msg.getIntProperty("msg")); |
| msg = consumer.receive(TIMEOUT); |
| assertEquals(4, msg.getIntProperty("msg")); |
| msg = consumer.receive(TIMEOUT); |
| assertEquals(7, msg.getIntProperty("msg")); |
| msg = consumer.receive(TIMEOUT); |
| assertEquals(11, msg.getIntProperty("msg")); |
| |
| msg = consumer.receive(TIMEOUT); |
| assertEquals(3, msg.getIntProperty("msg")); |
| msg = consumer.receive(TIMEOUT); |
| assertEquals(6, msg.getIntProperty("msg")); |
| msg = consumer.receive(TIMEOUT); |
| assertEquals(8, msg.getIntProperty("msg")); |
| msg = consumer.receive(TIMEOUT); |
| assertEquals(10, msg.getIntProperty("msg")); |
| } |
| |
| private Message nextMessage(int msg, boolean first, Session producerSession, MessageProducer producer) throws JMSException |
| { |
| Message send = producerSession.createTextMessage("Message: " + msg); |
| send.setIntProperty("msg", msg); |
| |
| return send; |
| } |
| |
| |
| } |