@Test
public void testAllocateRequests()
throws Exception {
StringRequest.StringRequestEstimator estimator = new StringRequest.StringRequestEstimator();
RequestAllocatorConfig<StringRequest> configuration =
RequestAllocatorConfig.builder(estimator).withPrioritizer(new StringRequest.StringRequestComparator()).build();
PreOrderAllocator<StringRequest> allocator =
new PreOrderAllocator<>(configuration);
ResourcePool pool = ResourcePool.builder().maxResource(StringRequest.MEMORY, 100.).build();
List<Requestor<StringRequest>> requests = Lists.<Requestor<StringRequest>>newArrayList(
new StringRequestor("r1", "a-50", "f-50", "k-20"),
new StringRequestor("r2", "j-10", "b-20", "e-20"),
new StringRequestor("r3", "g-20", "c-200", "d-30"));
AllocatedRequestsIterator<StringRequest> result = allocator.allocateRequests(requests.iterator(), pool);
List<StringRequest> resultList = Lists.newArrayList(result);
Assert.assertEquals(resultList.size(), 3);
Assert.assertEquals(resultList.get(0).getString(), "a-50");
Assert.assertEquals(resultList.get(1).getString(), "b-20");
Assert.assertEquals(resultList.get(2).getString(), "d-30");
Assert.assertTrue(estimator.getQueriedRequests().contains("a-50"));
Assert.assertTrue(estimator.getQueriedRequests().contains("b-20"));
Assert.assertTrue(estimator.getQueriedRequests().contains("c-200"));
Assert.assertTrue(estimator.getQueriedRequests().contains("d-30"));
Assert.assertFalse(estimator.getQueriedRequests().contains("e-20"));
Assert.assertFalse(estimator.getQueriedRequests().contains("f-50"));
}