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_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_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)
def test_consume_enoughBW(self):
rl = RateLimiter()
consumed = rl.consume(self.packet, self.address)
self.assertTrue(consumed)
