def test_consume_notEnoughTokens(self):
tb = TokenBucket(10, 1)
self.assertFalse(tb.consume(100))
self.assertEquals(10, tb.tokens)
def test_consume_notEnoughTokens(self):
tb = TokenBucket(10, 1)
self.assertFalse(tb.consume(100))
self.assertEquals(10, tb.tokens)
def test_consume_verifyTokenDecrease(self):
tb = TokenBucket(10, 1)
self.assertTrue(tb.consume(10))
self.assertEquals(0, tb.tokens)
def test_consume_verifyTokenDecrease(self):
tb = TokenBucket(10, 1)
self.assertTrue(tb.consume(10))
self.assertEquals(0, tb.tokens)
