[jboss-cvs] JBossAS SVN: r83532 - projects/cluster/ha-client/trunk/src/test/java/org/jboss/test/ha/client/loadbalance.

[jboss-cvs-commits at lists.jboss.org]

Author: bstansberry at jboss.com
Date: 2009-01-27 21:38:24 -0500 (Tue, 27 Jan 2009)
New Revision: 83532

Added:
   projects/cluster/ha-client/trunk/src/test/java/org/jboss/test/ha/client/loadbalance/RandomRobinUnitTestCase.java
Log:
Add a test of RandomRobin

Added: projects/cluster/ha-client/trunk/src/test/java/org/jboss/test/ha/client/loadbalance/RandomRobinUnitTestCase.java
===================================================================
--- projects/cluster/ha-client/trunk/src/test/java/org/jboss/test/ha/client/loadbalance/RandomRobinUnitTestCase.java	                        (rev 0)
+++ projects/cluster/ha-client/trunk/src/test/java/org/jboss/test/ha/client/loadbalance/RandomRobinUnitTestCase.java	2009-01-28 02:38:24 UTC (rev 83532)
@@ -0,0 +1,126 @@
+/*
+ * JBoss, Home of Professional Open Source.
+ * Copyright 2008, Red Hat Middleware LLC, and individual contributors
+ * as indicated by the @author tags. See the copyright.txt file in the
+ * distribution for a full listing of individual contributors.
+ *
+ * This is free software; you can redistribute it and/or modify it
+ * under the terms of the GNU Lesser General Public License as
+ * published by the Free Software Foundation; either version 2.1 of
+ * the License, or (at your option) any later version.
+ *
+ * This software 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 software; if not, write to the Free
+ * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
+ * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
+ */
+
+package org.jboss.test.ha.client.loadbalance;
+
+
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.HashSet;
+import java.util.List;
+import java.util.Set;
+
+import org.jboss.ha.client.loadbalance.LoadBalancePolicy;
+import org.jboss.ha.client.loadbalance.RandomRobin;
+import org.jboss.ha.framework.interfaces.ClusteringTargetsRepository;
+import org.jboss.ha.framework.interfaces.FamilyClusterInfo;
+
+/**
+ * Unit tests for {@link RandomRobin}.
+ * 
+ * @author Brian Stansberry
+ *
+ */
+public class RandomRobinUnitTestCase extends AbstractLoadBalancePolicyTestCase<RandomRobin>
+{
+
+   /**
+    * Create a new RoundRobinUnitTestCase.
+    * 
+    */
+   public RandomRobinUnitTestCase()
+   {
+      
+   }
+
+   /**
+    * Create a new RoundRobinUnitTestCase.
+    * 
+    * @param name
+    */
+   public RandomRobinUnitTestCase(String name)
+   {
+      super(name);
+   }
+
+   protected RandomRobin getLoadBalancePolicy()
+   {
+      return new RandomRobin();
+   }
+
+   public void testSelectionSpread()
+   {
+      testCount++;
+      FamilyClusterInfo fci = ClusteringTargetsRepository.initTarget(FAMILY_BASE + testCount, Arrays.asList(TARGETS));
+      
+      Set<Object> selected = new HashSet<Object>();
+      LoadBalancePolicy lbp = getLoadBalancePolicy();
+      
+      for (int i = 0; i < 10000; i++)
+      {
+         selected.add(lbp.chooseTarget(fci));
+         if (selected.size() == TARGETS.length)
+            break;
+      }
+      
+      assertEquals(TARGETS.length, selected.size());
+   }
+
+   public void testFailover()
+   {
+      testCount++;
+      List<String> targets = new ArrayList<String>(Arrays.asList(TARGETS));
+      FamilyClusterInfo fci = ClusteringTargetsRepository.initTarget(FAMILY_BASE + testCount, targets);
+      LoadBalancePolicy lbp = getLoadBalancePolicy();
+      
+      String target = (String) lbp.chooseTarget(fci);
+      assertTrue(targets.contains(target));
+      
+      while (targets.size() > 1)
+      {
+         fci.removeDeadTarget(target);
+         targets.remove(target);
+         assertEquals(targets, fci.getTargets());
+         
+         target = (String) lbp.chooseTarget(fci);
+         assertTrue(targets.contains(target));
+      }
+   }
+
+   public void testTopologyChange()
+   {
+      testCount++;
+      List<String> targets = Arrays.asList(TARGETS);
+      FamilyClusterInfo fci = ClusteringTargetsRepository.initTarget(FAMILY_BASE + testCount, targets);
+      LoadBalancePolicy lbp = getLoadBalancePolicy();
+      
+      String target = (String) lbp.chooseTarget(fci);
+      assertTrue(targets.contains(target));
+      
+      targets = Arrays.asList(new String[]{"W", "X", "Y", "Z"});
+      fci.updateClusterInfo(targets, 1);
+      
+      target = (String) lbp.chooseTarget(fci);
+      assertTrue(targets.contains(target));
+   }
+
+}


Property changes on: projects/cluster/ha-client/trunk/src/test/java/org/jboss/test/ha/client/loadbalance/RandomRobinUnitTestCase.java
___________________________________________________________________
Name: svn:keywords
   +