algorithm/Java
1
0
Fork 0
mirror of https://github.com/TheAlgorithms/Java.git synced 2025-12-19 07:00:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add TokenBucket algorithm (#5840)

Browse Source
This commit is contained in:
Hardik Pawar
2024-10-26 14:09:31 +05:30
committed by GitHub
parent 9dbdaf6afb
commit 82dee61701
3 changed files with 153 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

90
src/test/java/com/thealgorithms/datastructures/queues/TokenBucketTest.java Normal file
View File

@@ -0,0 +1,90 @@
package com.thealgorithms.datastructures.queues;
import static org.junit.jupiter.api.Assertions.assertFalse;
import static org.junit.jupiter.api.Assertions.assertTrue;
import org.junit.jupiter.api.Test;
public class TokenBucketTest {
@Test
public void testRateLimiter() throws InterruptedException {
TokenBucket bucket = new TokenBucket(5, 1);
for (int i = 0; i < 5; i++) {
assertTrue(bucket.allowRequest());
}
assertFalse(bucket.allowRequest());
Thread.sleep(1000);
assertTrue(bucket.allowRequest());
}
@Test
public void testRateLimiterWithExceedingRequests() throws InterruptedException {
TokenBucket bucket = new TokenBucket(3, 1);
for (int i = 0; i < 3; i++) {
assertTrue(bucket.allowRequest());
}
assertFalse(bucket.allowRequest());
Thread.sleep(1000);
assertTrue(bucket.allowRequest());
assertFalse(bucket.allowRequest());
}
@Test
public void testRateLimiterMultipleRefills() throws InterruptedException {
TokenBucket bucket = new TokenBucket(2, 1);
assertTrue(bucket.allowRequest());
assertTrue(bucket.allowRequest());
assertFalse(bucket.allowRequest());
Thread.sleep(1000);
assertTrue(bucket.allowRequest());
Thread.sleep(1000);
assertTrue(bucket.allowRequest());
assertFalse(bucket.allowRequest());
}
@Test
public void testRateLimiterEmptyBucket() {
TokenBucket bucket = new TokenBucket(0, 1);
assertFalse(bucket.allowRequest());
}
@Test
public void testRateLimiterWithHighRefillRate() throws InterruptedException {
TokenBucket bucket = new TokenBucket(5, 10);
for (int i = 0; i < 5; i++) {
assertTrue(bucket.allowRequest());
}
assertFalse(bucket.allowRequest());
Thread.sleep(1000);
for (int i = 0; i < 5; i++) {
assertTrue(bucket.allowRequest());
}
}
@Test
public void testRateLimiterWithSlowRequests() throws InterruptedException {
TokenBucket bucket = new TokenBucket(5, 1);
for (int i = 0; i < 5; i++) {
assertTrue(bucket.allowRequest());
}
Thread.sleep(1000);
assertTrue(bucket.allowRequest());
Thread.sleep(2000);
assertTrue(bucket.allowRequest());
assertTrue(bucket.allowRequest());
}
}