summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--src/vnet/policer/test/test_policer.py123
1 files changed, 123 insertions, 0 deletions
diff --git a/src/vnet/policer/test/test_policer.py b/src/vnet/policer/test/test_policer.py
new file mode 100644
index 00000000000..09b6e52da01
--- /dev/null
+++ b/src/vnet/policer/test/test_policer.py
@@ -0,0 +1,123 @@
+#!/usr/bin/env python3
+# Copyright (c) 2021 Graphiant, Inc.
+
+import unittest
+
+from framework import VppTestCase, VppTestRunner
+from vpp_policer import VppPolicer, PolicerAction
+
+# Default for the tests is 10s of "Green" packets at 8Mbps, ie. 10M bytes.
+# The policer helper CLI "sends" 500 byte packets, so default is 20000.
+
+TEST_RATE = 8000 # kbps
+TEST_BURST = 10000 # ms
+
+CIR_OK = 8500 # CIR in kbps, above test rate
+CIR_LOW = 7000 # CIR in kbps, below test rate
+EIR_OK = 9000 # EIR in kbps, above test rate
+EIR_LOW = 7500 # EIR in kbps, below test rate
+
+NUM_PKTS = 20000
+
+CBURST = 100000 # Committed burst in bytes
+EBURST = 200000 # Excess burst in bytes
+
+
+class TestPolicer(VppTestCase):
+ """ Policer Test Case """
+
+ def setUp(self):
+ super(TestPolicer, self).setUp()
+
+ def tearDown(self):
+ super(TestPolicer, self).tearDown()
+
+ def run_policer_test(self, type, cir, cb, eir, eb, rate=8000, burst=10000,
+ colour=0):
+ """
+ Configure a Policer and push traffic through it.
+ """
+ types = {
+ '1R2C': 0,
+ '1R3C': 1,
+ '2R3C': 3,
+ }
+
+ pol_type = types.get(type)
+ policer = VppPolicer(self, "pol1", cir, eir, cb, eb, rate_type=0,
+ type=pol_type, color_aware=colour)
+ policer.add_vpp_config()
+
+ error = self.vapi.cli(
+ f"test policing index {policer.policer_index} rate {rate} "
+ f"burst {burst} colour {colour}")
+
+ stats = policer.get_stats()
+ policer.remove_vpp_config()
+
+ return stats
+
+ def test_policer_1r2c(self):
+ """ Single rate, 2 colour policer """
+ stats = self.run_policer_test("1R2C", CIR_OK, CBURST, 0, 0)
+ self.assertEqual(stats['conform_packets'], NUM_PKTS)
+
+ stats = self.run_policer_test("1R2C", CIR_LOW, CBURST, 0, 0)
+ self.assertLess(stats['conform_packets'], NUM_PKTS)
+ self.assertEqual(stats['exceed_packets'], 0)
+ self.assertGreater(stats['violate_packets'], 0)
+
+ stats = self.run_policer_test("1R2C", CIR_LOW, CBURST, 0, 0, colour=2)
+ self.assertEqual(stats['violate_packets'], NUM_PKTS)
+
+ def test_policer_1r3c(self):
+ """ Single rate, 3 colour policer """
+ stats = self.run_policer_test("1R3C", CIR_OK, CBURST, 0, 0)
+ self.assertEqual(stats['conform_packets'], NUM_PKTS)
+
+ stats = self.run_policer_test("1R3C", CIR_LOW, CBURST, 0, EBURST)
+ self.assertLess(stats['conform_packets'], NUM_PKTS)
+ self.assertGreater(stats['exceed_packets'], 0)
+ self.assertGreater(stats['violate_packets'], 0)
+
+ stats = self.run_policer_test("1R3C", CIR_LOW, CBURST, 0, EBURST,
+ colour=1)
+ self.assertEqual(stats['conform_packets'], 0)
+ self.assertGreater(stats['exceed_packets'], 0)
+ self.assertGreater(stats['violate_packets'], 0)
+
+ stats = self.run_policer_test("1R3C", CIR_LOW, CBURST, 0, EBURST,
+ colour=2)
+ self.assertEqual(stats['violate_packets'], NUM_PKTS)
+
+ def test_policer_2r3c(self):
+ """ Dual rate, 3 colour policer """
+ stats = self.run_policer_test("2R3C", CIR_OK, CBURST, EIR_OK, EBURST)
+ self.assertEqual(stats['conform_packets'], NUM_PKTS)
+
+ stats = self.run_policer_test("2R3C", CIR_LOW, CBURST, EIR_OK, EBURST)
+ self.assertLess(stats['conform_packets'], NUM_PKTS)
+ self.assertGreater(stats['exceed_packets'], 0)
+ self.assertEqual(stats['violate_packets'], 0)
+
+ stats = self.run_policer_test("2R3C", CIR_LOW, CBURST, EIR_LOW, EBURST)
+ self.assertLess(stats['conform_packets'], NUM_PKTS)
+ self.assertGreater(stats['exceed_packets'], 0)
+ self.assertGreater(stats['violate_packets'], 0)
+
+ stats = self.run_policer_test("2R3C", CIR_LOW, CBURST, EIR_OK, EBURST,
+ colour=1)
+ self.assertEqual(stats['exceed_packets'], NUM_PKTS)
+
+ stats = self.run_policer_test("2R3C", CIR_LOW, CBURST, EIR_LOW, EBURST,
+ colour=1)
+ self.assertEqual(stats['conform_packets'], 0)
+ self.assertGreater(stats['exceed_packets'], 0)
+ self.assertGreater(stats['violate_packets'], 0)
+
+ stats = self.run_policer_test("2R3C", CIR_LOW, CBURST, EIR_OK, EBURST,
+ colour=2)
+ self.assertEqual(stats['violate_packets'], NUM_PKTS)
+
+if __name__ == '__main__':
+ unittest.main(testRunner=VppTestRunner)