TestBucketAssigner.java

package org.hivedb.meta;

import org.hivedb.BucketAssigner;
import org.hivedb.util.database.HiveDbDialect;
import org.junit.Test;import static org.junit.Assert.assertNotNull;import static org.junit.Assert.assertEquals;

import java.util.ArrayList;
import java.util.Collection;

public class TestBucketAssigner {

	@Test
	public void testAssignment() {
		Assigner assigner = new BucketAssigner(24);
		Collection<Node> nodes = createNodes(6);
		Node assigned = assigner.chooseNode(nodes, 6);
		assertNotNull(assigned);
		assertEquals(new Integer(0), assigned.getId());
		assigned = assigner.chooseNode(nodes, 1);
		assertNotNull(assigned);
		assertEquals(new Integer(1), assigned.getId());
		assigned = assigner.chooseNode(nodes, 8);
		assertNotNull(assigned);
		assertEquals(new Integer(2), assigned.getId());
	}
	
	private Collection<Node> createNodes(int nodeCount) {
		Collection<Node> nodes = new ArrayList<Node>();
		for(int i=0; i<nodeCount; i++)
			nodes.add(new Node(i,"Node"+i, "NodeUri"+i, "", HiveDbDialect.H2));
		return nodes;
	}
}