@Test
public void test() throws Exception {
try(Closer closer = Closer.create()) {
Map<String, String> configMap = Maps.newHashMap();
TestingServer zkTestingServer = closer.register(new TestingServer(-1));
configMap.put(ThrottlingGuiceServletConfig.ZK_STRING_KEY, zkTestingServer.getConnectString());
configMap.put(ThrottlingGuiceServletConfig.HA_CLUSTER_NAME, TestFailover.class.getSimpleName() + "_cluster");
Config config = ConfigFactory.parseMap(configMap);
ThrottlingGuiceServletConfig server2001 = createServerAtPort(config, 2001);
PermitAllocation allocation = sendRequestToServer(server2001, 10);
Assert.assertTrue(allocation.getPermits() >= 1);
ThrottlingGuiceServletConfig server2002 = createServerAtPort(config, 2002);
allocation = sendRequestToServer(server2001, 10);
Assert.assertTrue(allocation.getPermits() >= 1);
try {
sendRequestToServer(server2002, 10);
Assert.fail();
} catch (RestLiServiceException exc) {
Assert.assertTrue(exc.hasErrorDetails());
Assert.assertTrue(exc.getErrorDetails().containsKey(LimiterServerResource.LOCATION_301));
Assert.assertEquals(new URI(exc.getErrorDetails().get(LimiterServerResource.LOCATION_301).toString()).getPort(), 2001);
}
server2001.close();
allocation = sendRequestToServer(server2002, 10);
Assert.assertTrue(allocation.getPermits() >= 1);
}
}
