1 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 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 [] results = new String [20]; 47 48 for (int i = 0; i < results.length; ++i) 49 { 50 results[i] = balancer.next().getId(); 51 } 52 53 String [] expected = new String [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 |