/*
*
* 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 org.apache.qpid.AMQException;
import org.apache.qpid.client.AMQDestination;
import org.apache.qpid.client.AMQSession;
import org.apache.qpid.test.utils.QpidBrokerTestCase;
import javax.jms.Connection;
import javax.jms.JMSException;
import javax.jms.Message;
import javax.jms.MessageConsumer;
import javax.jms.MessageProducer;
import javax.jms.Queue;
import javax.jms.Session;
/**
* Test Case to ensure that messages are correctly returned.
* This includes checking:
* - The message is returned.
* - The broker doesn't leak memory.
* - The broker's state is correct after test.
*/
public class QueueDepthWithSelectorTest extends QpidBrokerTestCase
{
protected final String VHOST = "test";
protected final String QUEUE = this.getClass().getName();
protected Connection _clientConnection;
protected Connection _producerConnection;
private Session _clientSession;
protected Session _producerSession;
protected MessageProducer _producer;
private MessageConsumer _consumer;
protected static int MSG_COUNT = 50;
protected Message[] _messages = new Message[MSG_COUNT];
protected Queue _queue;
@Override
public void setUp() throws Exception
{
super.setUp();
_messages = new Message[MSG_COUNT];
_queue = getTestQueue();
//Create Producer
_producerConnection = getConnection();
_producerConnection.start();
_producerSession = _producerConnection.createSession(false, Session.AUTO_ACKNOWLEDGE);
_producer = _producerSession.createProducer(_queue);
// Create consumer
_clientConnection = getConnection();
_clientConnection.start();
_clientSession = _clientConnection.createSession(false, Session.AUTO_ACKNOWLEDGE);
_consumer = _clientSession.createConsumer(_queue, "key = 23");
}
public void test() throws Exception
{
//Send messages
_logger.info("Starting to send messages");
for (int msg = 0; msg < MSG_COUNT; msg++)
{
_producer.send(nextMessage(msg));
}
_logger.info("Closing connection");
//Close the connection.. .giving the broker time to clean up its state.
_producerConnection.close();
//Verify we get all the messages.
_logger.info("Verifying messages");
verifyAllMessagesRecevied(50);
verifyBrokerState(0);
//Close the connection.. .giving the broker time to clean up its state.
_clientConnection.close();
//Verify Broker state
_logger.info("Verifying broker state");
verifyBrokerState(0);
}
protected void verifyBrokerState(int expectedDepth)
{
try
{
Connection connection = getConnection();
AMQSession session = (AMQSession)connection.createSession(false, Session.AUTO_ACKNOWLEDGE);
long queueDepth = session.getQueueDepth((AMQDestination) _queue);
assertEquals("Session reports Queue depth not as expected", expectedDepth, queueDepth);
connection.close();
}
catch (AMQException e)
{
fail(e.getMessage());
}
catch (Exception e)
{
fail(e.getMessage());
}
}
protected void verifyAllMessagesRecevied(int expectedDepth) throws Exception
{
boolean[] msgIdRecevied = new boolean[MSG_COUNT];
for (int i = 0; i < expectedDepth; i++)
{
_messages[i] = _consumer.receive(1000);
assertNotNull("should have received a message but didn't", _messages[i]);
}
//Check received messages
int msgId = 0;
for (Message msg : _messages)
{
assertNotNull("Message should not be null", msg);
assertEquals("msgId was wrong", msgId, msg.getIntProperty("ID"));
assertFalse("Already received msg id " + msgId, msgIdRecevied[msgId]);
msgIdRecevied[msgId] = true;
msgId++;
}
//Check all received
for (msgId = 0; msgId < expectedDepth; msgId++)
{
assertTrue("Message " + msgId + " not received.", msgIdRecevied[msgId]);
}
//do a synchronous op to ensure the acks are processed
//on the broker before proceeding
((AMQSession)_clientSession).sync();
}
/**
* Get the next message putting the given count into the intProperties as ID.
*
* @param msgNo the message count to store as ID.
* @throws JMSException
*/
protected Message nextMessage(int msgNo) throws JMSException
{
Message send = _producerSession.createTextMessage("MessageReturnTest");
send.setIntProperty("ID", msgNo);
send.setIntProperty("key", 23);
return send;
}
}