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.codehaus.activemq.service;
19  
20  import junit.framework.TestCase;
21  import org.codehaus.activemq.broker.Broker;
22  import org.codehaus.activemq.broker.impl.DefaultBroker;
23  import org.codehaus.activemq.filter.Filter;
24  import org.codehaus.activemq.filter.FilterFactory;
25  import org.codehaus.activemq.filter.FilterFactoryImpl;
26  import org.codehaus.activemq.message.ActiveMQDestination;
27  import org.codehaus.activemq.message.ActiveMQMessage;
28  import org.codehaus.activemq.message.ActiveMQQueue;
29  import org.codehaus.activemq.message.ActiveMQTopic;
30  import org.codehaus.activemq.message.ConsumerInfo;
31  import org.codehaus.activemq.util.IdGenerator;
32  
33  import javax.jms.JMSException;
34  
35  /***
36   * Useful base class for Broker related test cases
37   *
38   * @version $Revision: 1.4 $
39   */
40  public abstract class BrokerTestSupport extends TestCase {
41  
42      private static final String BROKER_NAME = "TEST_BROKER";
43  
44      protected Broker broker;
45      protected FilterFactory filterFactory = new FilterFactoryImpl();
46      protected IdGenerator idGenerator = new IdGenerator();
47  
48      protected boolean isTopic() {
49          return true;
50      }
51  
52      protected ActiveMQMessage createMessage(String subject) {
53          ActiveMQMessage message = new ActiveMQMessage();
54          message.setJMSMessageID(idGenerator.generateId());
55          message.setJMSDestination(createDestination(subject));
56          return message;
57      }
58  
59      protected Filter createFilter(String subject, String selector) throws JMSException {
60          return filterFactory.createFilter(createDestination(subject), selector);
61      }
62  
63      protected ActiveMQDestination createDestination(String subject) {
64          if (isTopic()) {
65              return new ActiveMQTopic(subject);
66          }
67          else {
68              return new ActiveMQQueue(subject);
69          }
70      }
71  
72      protected Broker createStore() {
73          return new DefaultBroker(BROKER_NAME);
74      }
75  
76      protected ConsumerInfo createConsumer(String subject) {
77          return createConsumer(subject, null);
78      }
79  
80      protected ConsumerInfo createConsumer(String subject, String selector) {
81          ConsumerInfo info = new ConsumerInfo();
82          info.setConsumerId(idGenerator.generateId());
83          info.setDestination(createDestination(subject));
84          info.setSelector(selector);
85          return info;
86      }
87  
88      protected void setUp() throws Exception {
89          broker = new DefaultBroker(BROKER_NAME);
90          broker.start();
91      }
92  
93      protected void tearDown() throws Exception {
94          broker.stop();
95      }
96  }