policer: output interface policer

[vpp.git] / test / vpp_policer.py

diff --git a/test/vpp_policer.py b/test/vpp_policer.py
index 49d1185..0f3b073 100644 (file)
--- a/test/vpp_policer.py
+++ b/test/vpp_policer.py
@@ -1,5 +1,11 @@
 from vpp_object import VppObject
 from vpp_ip import INVALID_INDEX
+from enum import Enum
+
+
+class Dir(Enum):
+    RX = 0
+    TX = 1
 
 
 class PolicerAction():
@@ -57,6 +63,18 @@ class VppPolicer(VppObject):
         self._test.vapi.policer_add_del(is_add=False, name=self.name)
         self._policer_index = INVALID_INDEX
 
+    def bind_vpp_config(self, worker, bind):
+        self._test.vapi.policer_bind(name=self.name, worker_index=worker,
+                                     bind_enable=bind)
+
+    def apply_vpp_config(self, if_index, dir: Dir, apply):
+        if dir == Dir.RX:
+            self._test.vapi.policer_input(
+                name=self.name, sw_if_index=if_index, apply=apply)
+        else:
+            self._test.vapi.policer_output(
+                name=self.name, sw_if_index=if_index, apply=apply)
+
     def query_vpp_config(self):
         dump = self._test.vapi.policer_dump(
             match_name_valid=True, match_name=self.name)
@@ -67,3 +85,24 @@ class VppPolicer(VppObject):
 
     def object_id(self):
         return ("policer-%s" % (self.name))
+
+    def get_stats(self, worker=None):
+        conform = self._test.statistics.get_counter("/net/policer/conform")
+        exceed = self._test.statistics.get_counter("/net/policer/exceed")
+        violate = self._test.statistics.get_counter("/net/policer/violate")
+
+        counters = {"conform": conform, "exceed": exceed, "violate": violate}
+
+        total = {}
+        for name, c in counters.items():
+            total[f'{name}_packets'] = 0
+            total[f'{name}_bytes'] = 0
+            for i in range(len(c)):
+                t = c[i]
+                if worker is not None and i != worker + 1:
+                    continue
+                stat_index = self._policer_index
+                total[f'{name}_packets'] += t[stat_index]['packets']
+                total[f'{name}_bytes'] += t[stat_index]['bytes']
+
+        return total