Example #1
0
 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))
Example #2
0
 def test_can_consume_enoughTokens(self):
     tb = TokenBucket(10, 1)
     self.assertTrue(tb.can_consume(5))
Example #3
0
 def test_can_consume_notEnoughTokens(self):
     tb = TokenBucket(10, 1)
     self.assertFalse(tb.can_consume(50))
Example #4
0
 def test_consume_verifyTokenDecrease(self):
     tb = TokenBucket(10, 1)
     self.assertTrue(tb.consume(10))
     self.assertEquals(0, tb.tokens)
Example #5
0
 def test_consume_notEnoughTokens(self):
     tb = TokenBucket(10, 1)
     self.assertFalse(tb.consume(100))
     self.assertEquals(10, tb.tokens)
Example #6
0
 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))
Example #7
0
 def test_can_consume_notEnoughTokens(self):
     tb = TokenBucket(10, 1)
     self.assertFalse(tb.can_consume(50))
Example #8
0
 def test_can_consume_enoughTokens(self):
     tb = TokenBucket(10, 1)
     self.assertTrue(tb.can_consume(5))
Example #9
0
 def test_consume_notEnoughTokens(self):
     tb = TokenBucket(10, 1)
     self.assertFalse(tb.consume(100))
     self.assertEquals(10, tb.tokens)
Example #10
0
 def test_consume_verifyTokenDecrease(self):
     tb = TokenBucket(10, 1)
     self.assertTrue(tb.consume(10))
     self.assertEquals(0, tb.tokens)