package org.jgroups.blocks;

import org.jgroups.tests.ChannelTestBase;

public class RpcDispatcherAnycastMultipleCallsTest extends ChannelTestBase
{
   private RpcDispatcherAnycastServerObject[] targets = null;

   public void tearDown() throws Exception {
       if(targets != null) {
           for(int i=0; i < targets.length; i++) {
               if(targets[i] != null) targets[i].shutdown();
               targets[i]=null;
           }
           targets=null;
       }
       super.tearDown();
   }

   public void test2InstancesAnycastIncludeSelf() throws Exception
   {
      performTest(true, 2, false);
   }

   public void test3InstancesAnycastIncludeSelf() throws Exception
   {
      performTest(true, 3, false);
   }

   public void test2InstancesMcastIncludeSelf() throws Exception
   {
      performTest(false, 2, false);
   }

   public void test3InstancesMcastIncludeSelf() throws Exception
   {
      performTest(false, 3, false);
   }

   public void test2InstancesAnycastExcludeSelf() throws Exception
   {
      performTest(true, 2, true);
   }

   public void test3InstancesAnycastExcludeSelf() throws Exception
   {
      performTest(true, 3, true);
   }

   public void test2InstancesMcastExcludeSelf() throws Exception
   {
      performTest(false, 2, true);
   }

   public void test3InstancesMcastExcludeSelf() throws Exception
   {
      performTest(false, 3, true);
   }
   

   /**
    * Simple test that creates n instances of {@link org.jgroups.blocks.RpcDispatcherAnycastServerObject}, each one creates a Channel
    * and registers an RpcDispatcher.
    *
    * This test then calls {@link org.jgroups.blocks.RpcDispatcherAnycastServerObject#callRemote(boolean, boolean)} once on each of the n instances
    * and then tests that the method calls have in fact been executed on each of the n instances.
    *
    * @param useAnycast if true, anycast is used for remote calls.
    * @param n number of instances
    * @param excludeSelf whether or not  to exclude self in rpc calls
    */
   private void performTest(boolean useAnycast, int n, boolean excludeSelf) throws Exception
   {
      // create n instances
      targets = new RpcDispatcherAnycastServerObject[n];
      for (int i=0; i<n; i++){         
         targets[i] = new RpcDispatcherAnycastServerObject(createChannel("A"));   
      }

      // test that the target method has been invoked 0 times on each instance.
      for (int i=0; i<n; i++) assertEquals(0, targets[i].i);

      // if we don't exclude self, the state of all instances should be identical.
      int value = 0;

      // if we are excluding self, we need an array of expected values.
      int[] expectedValues = null;

      if (excludeSelf)
      {
         expectedValues = new int[n];
         for (int i=0; i<n; i++) expectedValues[i] = 0;
      }

      for (int instances = 0; instances < n; instances++)
      {
         targets[instances].callRemote(useAnycast, excludeSelf);

         // the assertions and how we measure test success is different depending on whether we exclude self or not.

         if (excludeSelf)
         {
            for (int i=0; i<n; i++)
            {
               if (i != instances) expectedValues[i]++;
            }
            for (int i=0; i<n; i++) assertEquals("Failure when invoking call on instance " + instances + ".  Did not reach instance " + i + ".", expectedValues[i], targets[i].i);
         }
         else
         {
            value++;
            for (int i=0; i<n; i++) assertEquals("Failure when invoking call on instance " + instances + ".  Did not reach instance " + i + ".", value, targets[i].i);
         }
      }

   }
}
