Test for the token bucket rate limiter

TokenBucketRateLimiterTest.java

package org.lobster1234.misc;

import junit.framework.TestCase;

import java.util.concurrent.TimeUnit;

public class TokenBucketRateLimiterTest extends TestCase {

    public void testTokenBucketRateLimiter() {
        try {
            TokenBucketRateLimiter limiter = new TokenBucketRateLimiter(10, TimeUnit.SECONDS);
            for (int i = 0; i < 11; i++) {
                limiter.execute(new RequestPayLoad("Test Request"), payLoad -> System.out.println("Success " + payLoad.getPayload()),
                        payLoad -> System.out.println("Failure " + payLoad.getPayload()));
            }
            Thread.sleep(5000); //sleep for 5 seconds
            //try again
            for (int i = 0; i < 11; i++) {
                limiter.execute(new RequestPayLoad("Test Request"), payLoad -> System.out.println("Success " + payLoad.getPayload()),
                        payLoad -> System.out.println("Failure " + payLoad.getPayload()));
            }
        }catch(Exception e){
            e.printStackTrace();
            fail();
        }
    }
}