def test_can_consume_tokenRefill(self): tb = TokenBucket(10, 1) self.assertFalse(tb.can_consume(50)) self.clock.set(40) self.assertFalse(tb.can_consume(50))
def test_can_consume_notEnoughTokens(self): tb = TokenBucket(10, 1) self.assertFalse(tb.can_consume(50))
def test_can_consume_enoughTokens(self): tb = TokenBucket(10, 1) self.assertTrue(tb.can_consume(5))
def test_can_consume_tokenRefill(self): tb = TokenBucket(10, 1) self.assertFalse(tb.can_consume(50)) self.clock.set(40) self.assertFalse(tb.can_consume(50))
def test_can_consume_notEnoughTokens(self): tb = TokenBucket(10, 1) self.assertFalse(tb.can_consume(50))
def test_can_consume_enoughTokens(self): tb = TokenBucket(10, 1) self.assertTrue(tb.can_consume(5))