@SuppressWarnings({"unchecked"})
@Before
public void setUp() {
nodeFactoryMock = createMock(NodeFactory.class);
dataSourceFactoryMock = createMock(DataSourceFactory.class);
shardsFactory = new ShardsFactoryImpl<TestNode>();
shardsFactory.setDataSourceFactory(dataSourceFactoryMock);
shardsFactory.setNodeFactory(nodeFactoryMock);
}
private void twoAddressesTest(
String shard1Id, String shard2Id, int node1Weight, int node2Weight) {
DataSource testDataSource =
new DataSource() {
public RedisConnection getConnection() {
return null;
}
};
TestNode node1 = new TestNode();
TestNode node2 = new TestNode();
expect(dataSourceFactoryMock.create("addr1", 1)).andReturn(testDataSource);
expect(nodeFactoryMock.create(testDataSource)).andReturn(node1);
expect(dataSourceFactoryMock.create("addr2", 2)).andReturn(testDataSource);
expect(nodeFactoryMock.create(testDataSource)).andReturn(node2);
replay(nodeFactoryMock, dataSourceFactoryMock);
List<Shard<TestNode>> shards = new ArrayList<Shard<TestNode>>(shardsFactory.create());
verify(nodeFactoryMock, dataSourceFactoryMock);
Shard<TestNode> shard = shards.get(0);
assertEquals(node1, shard.getNode());
assertEquals(shard1Id, shard.getShardId());
assertEquals(node1Weight, shard.getWeight());
shard = shards.get(1);
assertEquals(node2, shard.getNode());
assertEquals(shard2Id, shard.getShardId());
assertEquals(node2Weight, shard.getWeight());
}
@Test(expected = IllegalStateException.class)
public void addressAndWrongWeightTest() {
shardsFactory.setAddresses("addr1:1, addr2:2");
shardsFactory.setWeights("2");
twoAddressesTest("addr1:1", "addr2:2", 2, 5);
}
@Test(expected = IllegalStateException.class)
public void addressAndWrongIdsTest() {
shardsFactory.setAddresses("addr1:1, addr2:2");
shardsFactory.setIds("shard1");
twoAddressesTest("shard1", "shard2", 1, 1);
}
@Test
public void addressAndWeightTest() {
shardsFactory.setAddresses("addr1:1, addr2:2");
shardsFactory.setWeights("2, 5");
twoAddressesTest("addr1:1", "addr2:2", 2, 5);
}
@Test
public void addressAndIdsTest() {
shardsFactory.setAddresses("addr1:1, addr2:2");
shardsFactory.setIds("shard1, shard2");
twoAddressesTest("shard1", "shard2", 1, 1);
}
@Test
public void onlyAddressTest() {
shardsFactory.setAddresses("addr1:1, addr2:2");
twoAddressesTest("addr1:1", "addr2:2", 1, 1);
}
