    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.multicast;
    19
    20  import org.codehaus.activemq.ActiveMQConnectionFactory;
    21  import org.codehaus.activemq.JmsTopicSendReceiveWithTwoConnectionsTest;
    22
    23  import javax.jms.Connection;
    24  import javax.jms.JMSException;
    25
    26  /***
    27   * @version $Revision: 1.1 $
    28   */
    29  public class MulticastTwoBrokerTest extends JmsTopicSendReceiveWithTwoConnectionsTest {
    30      private int counter;
    31
    32      protected Connection createConnection() throws JMSException {
    33          MockConnectionFactory factory = (MockConnectionFactory) getConnectionFactory();
    34          if (++counter == 1) {
    35              factory.setClientID("producer");
    36          }
    37          else {
    38              if (counter == 2) {
    39                  factory.setClientID("consumer");
    40              }
    41              else {
    42                  factory.setClientID("id:" + counter);
    43              }
    44          }
    45          Connection answer = factory.createConnection();
    46          factory.unregisterBroker();
    47          return answer;
    48      }
    49
    50      protected ActiveMQConnectionFactory createConnectionFactory() {
    51          return new MockConnectionFactory(getBrokerURL());
    52      }
    53
    54      protected String getBrokerURL() {
    55          return "multicast://224.1.2.3:5123";
    56      }
    57  }