def test_consume_overflowHostAndReset(self): rl = RateLimiter() # Consume first packet consumed = rl.consume(self.packet, self.address) self.assertTrue(consumed) # Drop second packet consumed = rl.consume(self.packet, self.address) self.assertFalse(consumed) # Wait a second for the limiter to reset self.clock.set(1) # And consume a packet again consumed = rl.consume(self.packet, self.address) self.assertTrue(consumed)
def test_consume_overflowGlobalAndReset(self): address1 = self.address address2 = ("127.0.0.1", 9999) address3 = ("127.0.0.1", 8888) address4 = ("127.0.0.1", 7777) rl = RateLimiter() # Consume first, second, and third packets consumed = rl.consume(self.packet, address1) self.assertTrue(consumed) consumed = rl.consume(self.packet, address2) self.assertTrue(consumed) consumed = rl.consume(self.packet, address3) self.assertTrue(consumed) # Drop the fourth consumed = rl.consume(self.packet, address4) self.assertFalse(consumed) # Wait a second for the limiter to reset self.clock.set(1) # And consume a packet again consumed = rl.consume(self.packet, address3) self.assertTrue(consumed)
def test_consume_enoughBW(self): rl = RateLimiter() consumed = rl.consume(self.packet, self.address) self.assertTrue(consumed)