/*
    * Copyright (c) 1998-2002 The Jgroup Team.
    *
    * This program is free software; you can redistribute it and/or modify
    * it under the terms of the GNU Lesser General Public License version 2 as
    * published by the Free Software Foundation.
    *
    * This program is distributed in the hope that it will be useful,
    * but WITHOUT ANY WARRANTY; without even the implied warranty of
   * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   * GNU Lesser General Public License for more details.
   *
   * You should have received a copy of the GNU Lesser General Public License
   * along with this program; if not, write to the Free Software
   * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
   *
   */
  
  package jgroup.arm;
  
  import java.net.InetAddress;
  
  import jgroup.core.ConfigManager;
  import jgroup.core.MemberId;
  import jgroup.relacs.config.AppConfig;
  import jgroup.relacs.config.DistributedSystemConfig;
  import jgroup.relacs.config.Host;
  import jgroup.relacs.config.HostSet;
  import jgroup.relacs.types.EndPointImpl;
  import jgroup.relacs.types.LocalId;
  import jgroup.relacs.types.MemberIdImpl;
  import jgroup.relacs.types.ViewImpl;
  import jgroup.util.JUnit;
  import junit.framework.Test;
  import junit.framework.TestCase;
  import junit.framework.TestSuite;
  
  /**
   * This class is a JUnit-test for ReplicaCount.
   * 
   * @author Henning Hommeland
   */
  public class TestReplicaCount extends TestCase
  {
  
    /* Fields */
    private AppConfig[] apps;
  
    /* Test-methods */
    public void testAssignRemoveReplicas() throws Exception
    {
      System.out.println("FIRST TEST: testAssignRemoveReplicas()");
      GroupTable gt = new GroupTable(5000, 2);
      DistributedSystemConfig dsc = ConfigManager.getDistributedSystem();
      ReplicaCount rc = new ReplicaCount(gt, dsc.getDomainSet());
  
      /* Make sure that the ReplicaCount object was created */
      assertTrue(!rc.equals(null));
  
      /* Assign replicas */
      HostSet hs1 = rc.assignReplicas(apps[1]);
      HostSet hs2 = rc.assignReplicas(apps[2]);
      System.out.println("hs1: " + hs1);
      System.out.println("app1: " + apps[1]);
      System.out.println("hs2: " + hs2);
      System.out.println("app2: " + apps[2]);
  
      /* Remove replicas */
      HostSet hs3 = rc.removeReplicas(apps[1]);
      assertTrue(hs1.equals(hs3));
      HostSet hs4 = rc.removeReplicas(apps[2]);
      assertTrue(hs2.equals(hs4));
  
      /* Check that the set of hosts is the same */
      hs1.addHosts(hs2);
      hs3.addHosts(hs4);
      assertTrue(hs1.equals(hs3));
    }
  
    public void testReassignReplica() throws Exception
    {
      System.out.println("SECOND TEST: testReassignReplica():");
      GroupTable gt = new GroupTable(5000, 2);
      DistributedSystemConfig dsc = ConfigManager.getDistributedSystem();
      ReplicaCount rc = new ReplicaCount(gt, dsc.getDomainSet());
  
      /* Make sure that the ReplicaCount object was created */
      assertTrue(!rc.equals(null));
  
      /* Assign replicas */
      HostSet hs1 = rc.assignReplicas(apps[1]);
  
      /* Test using removeReplicas() and reassignReplicas() */
      Host[] hostArray = hs1.toArray();
      Host host1 = rc.reassignReplica(apps[1], hostArray[0]);
      Host host2 = rc.reassignReplica(apps[1], hostArray[1]);
      HostSet hs2 = rc.removeReplicas(apps[1]);
      hs1.addHost(host1);
      hs1.addHost(host2);
 
     /* HostSets should now be equal */;
     assertTrue(hs2.equals(hs1));
   }
 
   public void testCollocateReplicas() throws Exception
   {
     System.out.println("FOURTH TEST: testCollocateReplicas()");
     GroupTable gt = new GroupTable(5000, 2);
     DistributedSystemConfig dsc = ConfigManager.getDistributedSystem();
     ReplicaCount rc = new ReplicaCount(gt, dsc.getDomainSet());
 
     /* Make sure that the ReplicaCount object was created */
     assertTrue(!rc.equals(null));
 
     /* Assign and collocate apps */
     HostSet hs1 = rc.assignReplicas(apps[1]);
     HostSet hs2 = rc.collocateReplicas(apps[2], apps[1]);
 
     /* HostSets should be equal */
     assertTrue(hs1.equals(hs2));
     HostSet hs3 = rc.removeReplicas(apps[1]);
     assertTrue(hs1.equals(hs3));
 
     /* Also - removeReplicas() should remove apps[2] as well */
     // GroupTable groupTable = (GroupTable) rc.getState();
     // assertTrue(!groupTable.contains(apps[2]));
   }
 
   public void testViewChange() throws Exception
   {
     System.out.println("FIFTH TEST: testViewChange()");
     GroupTable gt = new GroupTable(5000, 2);
     DistributedSystemConfig dsc = ConfigManager.getDistributedSystem();
     ReplicaCount rc = new ReplicaCount(gt, dsc.getDomainSet());
 
     /* Make sure that the ReplicaCount object was created */
     assertTrue(!rc.equals(null));
 
     /* State before setting new view */
     // System.out.println("Getting state before running new view:");
     // System.out.println("***************************************");
     // System.out.println(rc.getState());
     // System.out.println("***************************************");
     /* Generate a new view */
     MemberId[] memTab = new MemberId[2];
     memTab[0] = generateMemberId("lx0070.ux.his.no");
     memTab[1] = generateMemberId("lx0071.ux.his.no");
     ViewImpl newView = new ViewImpl(apps[1].getGroupId(), 1, 0, memTab);
 
     /* Bootstrapping no longer supported!? */
     // GroupTable groupTable = (GroupTable) rc.getState();
     // groupTable.add(apps[1], hosts);
     /* Ugly hack!!!! */
     // ReplicaManagerImpl rm = new ReplicaManagerImpl(true);
     /* Send out new view */
     rc.viewChangeEvent(newView);
     // System.out.println("Getting state after running new view:");
     // System.out.println("**************************************");
     // System.out.println(rc.getState());
     // System.out.println("**************************************");
 
     /*
      * Make sure that ReplicaCount is in correct state. (In lack of equals()
      * implementation in GroupTable).
      */
     // groupTable = (GroupTable) rc.getState();
     // assertTrue(groupTable.contains(apps[1]));
     // assertTrue(groupTable.size() == 1);
     // MemberId[] memTab2 = new MemberId[2];
     // memTab2 = groupTable.getMembers(apps[1].getGroupId());
     // assertTrue(memTab[0].equals(memTab2[1]));
     // assertTrue(memTab[1].equals(memTab2[0]));
   }
 
   /**
    * Generate a MemberId from parameter host. FIXME: Will generate equal
    * MemberId for same host.
    * 
    * @param host
    *          Hostname for which to generate MemberId.
    * @return MembedId for the given host.
    * @throws Exception
    */
   private MemberId generateMemberId(String host) throws Exception
   {
     InetAddress ia = InetAddress.getByName(host);
     EndPointImpl endPoint = new EndPointImpl(ia, 54300);
     LocalId localId = new LocalId(1, endPoint);
     MemberIdImpl memberId = new MemberIdImpl(endPoint, 1, localId);
     return memberId;
   }
 
   public static Test suite() throws Exception
   {
     TestSuite suite = new TestSuite();
     suite.addTest(new TestReplicaCount("testAssignRemoveReplicas"));
     // suite.addTest(new TestReplicaCount("testReassignReplica"));
     // suite.addTest(new TestReplicaCount("testGetPutState"));
     // suite.addTest(new TestReplicaCount("testCollocateReplicas"));
     // suite.addTest(new TestReplicaCount("testViewChange"));
     return suite;
   }
 
   public static void main(String[] args) throws Exception
   {
     junit.textui.TestRunner.run(suite());
   }
 
   /*
    * @see TestCase#setUp()
    */
   protected void setUp() throws Exception
   {
     super.setUp();
     // Enable JUnit tests for Jgroup
     JUnit.JUNIT_TEST = true;
 
     String siteconfig = System.getProperty("jgroup.system.config",
         "file:target/classes/siteconfig/partition-config.xml");
     ConfigManager.init(siteconfig);
     apps = AppConfig.getApplications();
     for (int i = 0; i < apps.length; i++) {
       try {
         apps[i].getInitialRedundancy();
       } catch (IllegalStateException e) {
         continue;
       }
     }
   }
 
   /*
    * @see TestCase#tearDown()
    */
   protected void tearDown() throws Exception
   {
     super.tearDown();
   }
 
   /**
    * Constructor for TestReplicaCount.
    * 
    * @param name
    */
   public TestReplicaCount(String name) throws Exception
   {
     super(name);
   }
 }