01: /*
02: * HA-JDBC: High-Availability JDBC
03: * Copyright (c) 2004-2007 Paul Ferraro
04: *
05: * This library is free software; you can redistribute it and/or modify it
06: * under the terms of the GNU Lesser General Public License as published by the
07: * Free Software Foundation; either version 2.1 of the License, or (at your
08: * option) any later version.
09: *
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: */
31: @Test
32: @SuppressWarnings("nls")
33: public class TestRoundRobinBalancer extends AbstractTestBalancer {
34: @Override
35: protected Balancer<Void> createBalancer() {
36: return new RoundRobinBalancer<Void>();
37: }
38:
39: @Override
40: protected void next(Balancer<Void> balancer) {
41: int count = 100;
42:
43: balancer.add(new MockDatabase("0", 0));
44:
45: for (int i = 0; i < count; ++i) {
46: int weight = balancer.next().getWeight();
47:
48: assert weight == 0 : weight;
49: }
50:
51: balancer.add(new MockDatabase("1", 1));
52:
53: for (int i = 0; i < count; ++i) {
54: int weight = balancer.next().getWeight();
55:
56: assert weight == 1 : weight;
57: }
58:
59: balancer.add(new MockDatabase("2", 2));
60:
61: int[] expected = new int[] { 1, 2, 2 };
62:
63: for (int i = 0; i < count; ++i) {
64: int weight = balancer.next().getWeight();
65:
66: assert expected[i % 3] == weight : weight;
67: }
68: }
69: }
|