Source code: org/activemq/test/QueueMessageStoreTestSupport.java
1 /**
2 *
3 * Copyright 2004 Protique Ltd
4 *
5 * Licensed under the Apache License, Version 2.0 (the "License");
6 * you may not use this file except in compliance with the License.
7 * You may obtain a copy of the License at
8 *
9 * http://www.apache.org/licenses/LICENSE-2.0
10 *
11 * Unless required by applicable law or agreed to in writing, software
12 * distributed under the License is distributed on an "AS IS" BASIS,
13 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 * See the License for the specific language governing permissions and
15 * limitations under the License.
16 *
17 **/
18 package org.activemq.test;
19
20 import java.util.ArrayList;
21 import java.util.List;
22
23 import javax.jms.Destination;
24 import javax.jms.JMSException;
25
26 import org.activemq.message.ActiveMQDestination;
27 import org.activemq.message.ActiveMQMessage;
28 import org.activemq.message.ActiveMQQueue;
29 import org.activemq.message.MessageAck;
30 import org.activemq.service.MessageIdentity;
31 import org.activemq.service.QueueMessageContainer;
32
33 /**
34 * @version $Revision: 1.1.1.1 $
35 */
36 public abstract class QueueMessageStoreTestSupport extends MessageStoreTestSupport {
37
38 protected void acknowledgeMessage(int i) throws JMSException {
39 MessageIdentity identity = getMessage(i).getJMSMessageIdentity();
40 getQueueMessageContainer().delete(identity, createMessageAck(identity, getMessage(i).getJMSDestination()));
41 }
42
43 protected QueueMessageContainer getQueueMessageContainer() {
44 return (QueueMessageContainer) container;
45 }
46
47 protected ActiveMQDestination createDestination() {
48 return new ActiveMQQueue(getSubject());
49 }
50
51 protected void recover() throws JMSException {
52 }
53
54 protected MessageAck createMessageAck(MessageIdentity messageIdentity, Destination destination) {
55 MessageAck answer = new MessageAck();
56 answer.setConsumerId("James");
57 answer.setMessageID(messageIdentity.getMessageID());
58 answer.setMessageRead(true);
59 answer.setDestination(this.destination);
60 return answer;
61 }
62
63 protected ActiveMQMessage[] getMessagesToDispatch() throws JMSException {
64 List list = new ArrayList();
65 while (true) {
66 ActiveMQMessage message = getQueueMessageContainer().poll();
67 if (message == null) {
68 break;
69 }
70 list.add(message);
71 }
72 ActiveMQMessage[] answer = new ActiveMQMessage[list.size()];
73 list.toArray(answer);
74 return answer;
75 }
76 }