@Test public void testGetProvisionersWithFreeCapacity() throws IOException { ProvisionerStore store = getProvisionerStore(); store.writeProvisioner(provisioner1); Assert.assertTrue(store.getProvisionersWithFreeCapacity().isEmpty()); store.writeProvisioner(provisioner2); Assert.assertEquals(ImmutableSet.of(provisioner2), ImmutableSet.copyOf(store.getProvisionersWithFreeCapacity())); Provisioner updatedProvisioner1 = new Provisioner( provisioner1.getId(), provisioner1.getHost(), provisioner1.getPort(), provisioner1.getCapacityTotal(), ImmutableMap.<String, Integer>of("tenantX", 5), ImmutableMap.<String, Integer>of("tenantX", 5)); store.writeProvisioner(updatedProvisioner1); Assert.assertEquals(ImmutableSet.of(updatedProvisioner1, provisioner2), ImmutableSet.copyOf(store.getProvisionersWithFreeCapacity())); }
private void addWorkers(String tenantId, int numToAdd, ResourceCollection resources) throws CapacityException, IOException { for (Provisioner provisioner : provisionerStore.getProvisionersWithFreeCapacity()) { if (numToAdd <= 0) { break;