    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.transport;
    19
    20  import junit.framework.TestCase;
    21  import org.codehaus.activecluster.ClusterException;
    22
    23  import javax.jms.JMSException;
    24  import java.util.HashMap;
    25  import java.util.Map;
    26
    27  /***
    28   * @version $Revision: 1.1 $
    29   */
    30  public abstract class DiscoveryAgentTestSupport extends TestCase implements DiscoveryListener {
    31      protected boolean waitForever = false;
    32
    33      protected DiscoveryAgent agent;
    34      protected Map localDetails = new HashMap();
    35      protected String type = "_activemq_testcase.";
    36
    37      public void testDiscovery() throws Exception {
    38          localDetails.put("cheese", "Cheddar");
    39          localDetails.put("location", "London");
    40
    41          agent.setDiscoveryListener(this);
    42          agent.start();
    43          agent.registerService(createName(), localDetails);
    44
    45          if (waitForever) {
    46              synchronized (this) {
    47                  this.wait();
    48              }
    49          }
    50
    51          agent.stop();
    52      }
    53
    54      protected String createName() {
    55          return getName() + "@" + System.currentTimeMillis();
    56      }
    57
    58      protected void setUp() throws Exception {
    59          super.setUp();
    60          agent = createDiscoveryAgent();
    61          if (System.getProperty("waitForever") != null) {
    62              waitForever = true;
    63          }
    64      }
    65
    66      protected abstract DiscoveryAgent createDiscoveryAgent() throws JMSException, ClusterException;
    67
    68      public void addService(DiscoveryEvent event) {
    69          System.out.println("Added service: " + event + " with details: " + event.getServiceDetails() + " name: " + event.getServiceName());
    70      }
    71
    72      public void removeService(DiscoveryEvent event) {
    73          System.out.println("Removed service: " + event);
    74      }
    75  }