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 java.util.Arrays;
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: @SuppressWarnings("nls")
36: public class TestRandomBalancer extends AbstractTestBalancer {
37: @Override
38: protected Balancer<Void> createBalancer() {
39: return new RandomBalancer<Void>();
40: }
41:
42: @Override
43: protected void next(Balancer<Void> balancer) {
44: int count = 100;
45: int[] results = new int[3];
46:
47: Arrays.fill(results, 0);
48:
49: balancer.add(new MockDatabase("0", 0));
50:
51: for (int i = 0; i < count; ++i) {
52: results[balancer.next().getWeight()] += 1;
53: }
54:
55: assert results[0] == count : results[0];
56: assert results[1] == 0 : results[1];
57: assert results[2] == 0 : results[2];
58:
59: balancer.add(new MockDatabase("1", 1));
60:
61: Arrays.fill(results, 0);
62:
63: for (int i = 0; i < count; ++i) {
64: results[balancer.next().getWeight()] += 1;
65: }
66:
67: assert results[0] == 0 : results[0];
68: assert results[1] == count : results[1];
69: assert results[2] == 0 : results[2];
70:
71: balancer.add(new MockDatabase("2", 2));
72:
73: Arrays.fill(results, 0);
74:
75: for (int i = 0; i < count; ++i) {
76: results[balancer.next().getWeight()] += 1;
77: }
78:
79: assert results[0] == 0 : results[0];
80: assert results[1] < 50 : results[1];
81: assert results[2] > 50 : results[2];
82: }
83: }
|