@Test
public void takeShouldGetToken() {

    TestRefillStrategy testRefillStrategy = createTestRefillStrategy();
    TokenBucket tokenBucket = createTokenBucket(10, testRefillStrategy, 0);
    testRefillStrategy.setAmountToRefill(1);
    tokenBucket.take();
    testRefillStrategy.setAmountToRefill(10);
    tokenBucket.take(10);
}