KickJava   Java API By Example, From Geeks To Geeks.

Java > Open Source Codes > net > sf > hajdbc > balancer > TestRoundRobinBalancer


1 /*
2  * HA-JDBC: High-Availability JDBC
3  * Copyright (c) 2004-2006 Paul Ferraro
4  *
5  * This library is free software; you can redistribute it and/or modify it
6  * under the terms of the GNU Lesser General Public License as published by the
7  * Free Software Foundation; either version 2.1 of the License, or (at your
8  * option) any later version.
9  *
10  * This library is distributed in the hope that it will be useful, but WITHOUT
11  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
12  * FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License
13  * for more details.
14  *
15  * You should have received a copy of the GNU Lesser General Public License
16  * along with this library; if not, write to the Free Software Foundation,
17  * Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
18  *
19  * Contact: ferraro@users.sourceforge.net
20  */
21 package net.sf.hajdbc.balancer;
22
23 import org.testng.annotations.Test;
24
25 import net.sf.hajdbc.Balancer;
26 import net.sf.hajdbc.MockDatabase;
27
28 /**
29  * @author Paul Ferraro
30  * @version $Revision: 1022 $
31  * @since 1.0
32  */
33 @Test
34 public class TestRoundRobinBalancer extends AbstractTestBalancer
35 {
36     protected Balancer createBalancer()
37     {
38         return new RoundRobinBalancer();
39     }
40
41     protected void testNext(Balancer balancer)
42     {
43         int count = 100;
44         
45         balancer.add(new MockDatabase("0", 0));
46         
47         for (int i = 0; i < count; ++i)
48         {
49             int weight = balancer.next().getWeight();
50             
51             assert weight == 0 : weight;
52         }
53         
54         balancer.add(new MockDatabase("1", 1));
55         
56         for (int i = 0; i < count; ++i)
57         {
58             int weight = balancer.next().getWeight();
59             
60             assert weight == 1 : weight;
61         }
62         
63         balancer.add(new MockDatabase("2", 2));
64         
65         int[] expected = new int[] { 1, 2, 2 };
66         
67         for (int i = 0; i < count; ++i)
68         {
69             int weight = balancer.next().getWeight();
70             
71             assert expected[i % 3] == weight : weight;
72         }
73     }
74 }
75
Popular Tags