package org.apache.commons.jcs3.utils.discovery;
   
   /*
    * 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.
   */
  
  import java.io.IOException;
  import java.util.ArrayList;
  
  import org.apache.commons.jcs3.utils.net.HostNameUtil;
  import org.apache.commons.jcs3.utils.serialization.StandardSerializer;
  import org.apache.commons.jcs3.utils.timing.SleepUtil;
  
  import junit.framework.TestCase;
  
  /**
   * Unit tests for discovery
   */
  public class UDPDiscoveryUnitTest
      extends TestCase
  {
      /**
       * <p>
       * @throws IOException
       */
      public void testSimpleUDPDiscoveryIPv4()
          throws IOException
      {
          if (HostNameUtil.getMulticastNetworkInterface() == null)
          {
              System.out.println("This machine does not support multicast");
              return;
          }
          simpleUDPDiscovery("228.5.6.7");
      }
  
      /**
       * <p>
       * @throws IOException
       */
      public void testSimpleUDPDiscoveryIPv6()
          throws IOException
      {
          if (HostNameUtil.getMulticastNetworkInterface() == null)
          {
              System.out.println("This machine does not support multicast");
              return;
          }
  
          simpleUDPDiscovery("FF02::5678");
      }
  
      /**
       * <p>
       * @throws IOException
       */
      private void simpleUDPDiscovery(String discoveryAddress)
          throws IOException
      {
          final UDPDiscoveryAttributes attributes = new UDPDiscoveryAttributes();
          attributes.setUdpDiscoveryAddr(discoveryAddress);
          attributes.setUdpDiscoveryPort(6789);
          attributes.setServicePort(1000);
          attributes.setUdpTTL(4); /* datagram TTL */
  
          // create the service
          final UDPDiscoveryService service = new UDPDiscoveryService(attributes, new StandardSerializer());
          service.startup();
          service.addParticipatingCacheName( "testCache1" );
  
          final MockDiscoveryListener discoveryListener = new MockDiscoveryListener();
          service.addDiscoveryListener( discoveryListener );
  
          // create a receiver with the service
          final UDPDiscoveryReceiver receiver = new UDPDiscoveryReceiver( service,
                  null,
                  attributes.getUdpDiscoveryAddr(),
                  attributes.getUdpDiscoveryPort() );
          final Thread t = new Thread( receiver );
          t.start();
  
          // create a sender
          try (final UDPDiscoverySender sender = new UDPDiscoverySender(
                  attributes, service.getSerializer()))
         {
             // create more names than we have no wait facades for
             // the only one that gets added should be testCache1
             final ArrayList<String> cacheNames = new ArrayList<>();
             final int numJunk = 10;
             for ( int i = 0; i < numJunk; i++ )
             {
                 cacheNames.add( "junkCacheName" + i );
             }
             cacheNames.add( "testCache1" );
 
             // send max messages
             final int max = 10;
             int cnt = 0;
             try
             {
                 for ( ; cnt < max; cnt++ )
                 {
                     sender.passiveBroadcast( "localhost", 1111, cacheNames, 1 );
                     SleepUtil.sleepAtLeast( 20 );
                 }
             }
             catch (IOException e)
             {
                 // We may face network configuration issues with multicast - give up then
                 System.out.println("Problem sending multicast packet: " + e.getMessage());
             }
 
             SleepUtil.sleepAtLeast( 200 );
 
             // request broadcasts change things.
             assertTrue( "Receiver count [" + receiver.getCnt() + "] should be the at least the number sent [" + cnt + "].",
                         cnt <= receiver.getCnt() );
         }
     }
 }