@Test
public void tryTakeShouldGetToken() {

    TestRefillStrategy testRefillStrategy = createTestRefillStrategy();
    TokenBucket tokenBucket = createTokenBucket(10, testRefillStrategy, 0);
    testRefillStrategy.setAmountToRefill(1);
    boolean gotToken = tokenBucket.tryTake();
    assertTrue(gotToken);
    testRefillStrategy.setAmountToRefill(10);
    gotToken = tokenBucket.tryTake(10);
    assertTrue(gotToken);
}