KickJava   Java API By Example, From Geeks To Geeks.

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


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 java.util.Arrays JavaDoc;
24
25 import org.testng.annotations.Test;
26
27 import net.sf.hajdbc.Balancer;
28 import net.sf.hajdbc.MockDatabase;
29
30 /**
31  * @author Paul Ferraro
32  * @since 1.0
33  */

34 @Test
35 public class TestSimpleBalancer extends AbstractTestBalancer
36 {
37     protected Balancer createBalancer()
38     {
39         return new SimpleBalancer();
40     }
41
42     protected void testNext(Balancer balancer)
43     {
44         balancer.add(new MockDatabase("0", 0));
45         
46         String JavaDoc[] results = new String JavaDoc[20];
47         
48         for (int i = 0; i < results.length; ++i)
49         {
50             results[i] = balancer.next().getId();
51         }
52
53         String JavaDoc[] expected = new String JavaDoc[results.length];
54         
55         Arrays.fill(expected, "0");
56
57         assert Arrays.equals(results, expected) : Arrays.asList(results);
58
59         balancer.add(new MockDatabase("2", 2));
60         
61         for (int i = 0; i < results.length; ++i)
62         {
63             results[i] = balancer.next().getId();
64         }
65         
66         Arrays.fill(expected, "2");
67         
68         assert Arrays.equals(results, expected) : Arrays.asList(results);
69
70         balancer.add(new MockDatabase("1", 1));
71         
72         for (int i = 0; i < results.length; ++i)
73         {
74             results[i] = balancer.next().getId();
75         }
76
77         assert Arrays.equals(results, expected) : Arrays.asList(results);
78     }
79 }
80
Popular Tags