@Test
public void testSetRLCounter() {
providerImpl.setRLCounter(rl_key, 1);
assertEquals("rl_key counter should be 1", 1, providerImpl.getActiveRL(rl_key).getCount());
providerImpl.setRLCounter(rl_key, 5);
assertEquals("rl_key counter should be 5", 5, providerImpl.getActiveRL(rl_key).getCount());
}
@Test
public void testSetActive() {
providerImpl.addActiveRL(rl_key, createModel());
providerImpl.addActiveRL(rl_key + "01", createModel());
providerImpl.setRLActive(rl_key, false);
assertEquals(
"rl_key counter should be false", false, providerImpl.getActiveRL(rl_key).isActive());
assertEquals(
"rl_key counter should be true", true, providerImpl.getActiveRL(rl_key + "01").isActive());
}
@Test
public void testSetTimestamp() {
providerImpl.addActiveRL(rl_key, createModel());
providerImpl.addActiveRL(rl_key + "01", createModel());
long newCreatedTime = System.currentTimeMillis();
providerImpl.setRLTimestamp(rl_key, new Date(newCreatedTime));
assertEquals(
"rl_key counter should be " + newCreatedTime,
newCreatedTime,
providerImpl.getActiveRL(rl_key).getTimestamp().getTime());
assertEquals(
"rl_key counter should be " + createdTime,
createdTime,
providerImpl.getActiveRL(rl_key + "01").getTimestamp().getTime());
}
@Test
public void testSetEffectDuration() {
providerImpl.addActiveRL(rl_key, createModel());
providerImpl.addActiveRL(rl_key + "01", createModel());
long effectDuration = createdTime + 3600L * 1000;
providerImpl.setRLEffectDuration(rl_key, 5000L);
assertEquals(
"rl_key counter should be 5000 as long",
5000L,
providerImpl.getActiveRL(rl_key).getEffectDuration().longValue());
assertEquals(
"rl_key counter should be " + effectDuration + " as Long",
effectDuration,
providerImpl.getActiveRL(rl_key + "01").getEffectDuration().longValue());
}
@Test
public void testSetEffect() {
providerImpl.addActiveRL(rl_key, createModel());
providerImpl.addActiveRL(rl_key + "01", createModel());
providerImpl.setRLEffect(rl_key, RateLimiterStatus.SERVE_GIF);
assertEquals(
"rl_key counter should be SERVE_GIF",
RateLimiterStatus.SERVE_GIF,
providerImpl.getActiveRL(rl_key).getEffect());
assertEquals(
"rl_key counter should be BLOCK",
RateLimiterStatus.BLOCK,
providerImpl.getActiveRL(rl_key + "01").getEffect());
}
@Test
public void testSetRolloverPeriod() {
providerImpl.addActiveRL(rl_key, createModel());
providerImpl.addActiveRL(rl_key + "01", createModel());
providerImpl.setRLRolloverPeriod(rl_key, 1000L);
assertEquals(
"rl_key counter should be 1000 as Long",
1000L,
providerImpl.getActiveRL(rl_key).getRolloverPeriod().longValue());
assertEquals(
"rl_key counter should be 60 as Long",
60L,
providerImpl.getActiveRL(rl_key + "01").getRolloverPeriod().longValue());
}
@Test
public void testIncrementRLCounter() {
providerImpl.addActiveRL(rl_key, createModel());
providerImpl.addActiveRL(rl_key + "01", createModel());
assertTrue("No element found in activeRL map", providerImpl.cointainKeyInActiveRL(rl_key));
providerImpl.incrementRLCounter(rl_key);
assertEquals("rl_key counter should be 2", 2, providerImpl.getActiveRL(rl_key).getCount());
providerImpl.incrementRLCounter(rl_key);
assertEquals("rl_key counter should be 3", 3, providerImpl.getActiveRL(rl_key).getCount());
assertEquals(
"rl_key counter should be 1", 1, providerImpl.getActiveRL(rl_key + "01").getCount());
}
@Test
public void testIncrementRLCounterNonExistent() {
providerImpl.addActiveRL(rl_key, createModel());
providerImpl.incrementRLCounter(rl_key + "02");
assertEquals(
"rl_key counter should remain in 1", 1, providerImpl.getActiveRL(rl_key).getCount());
}
@Test
public void testAddActiveRL() {
providerImpl.addActiveRL(rl_key, rateLimiterPolicyModel_1);
RateLimiterPolicyModel activeRL = providerImpl.getActiveRL(rl_key);
assertNotNull("rateLimiterPolicyModel_1 element must exist for key" + rl_key, activeRL);
assertEquals(rateLimiterPolicyModel_1.getCount(), activeRL.getCount());
assertEquals(rateLimiterPolicyModel_1.getEffect(), activeRL.getEffect());
assertEquals(rateLimiterPolicyModel_1.getEffectDuration(), activeRL.getEffectDuration());
assertEquals(rateLimiterPolicyModel_1.getIp(), activeRL.getIp());
assertEquals(rateLimiterPolicyModel_1.getRolloverPeriod(), activeRL.getRolloverPeriod());
assertEquals(new Date(createdTime).getTime(), activeRL.getTimestamp().getTime());
}
