X-Git-Url: https://gerrit.fd.io/r/gitweb?a=blobdiff_plain;f=test%2Fvpp_policer.py;h=0f3b073d6e349614b98e85c1b5206f4c5be0f9b3;hb=e5a3ae017;hp=2c47eed75fd770c8f60063c6770bf60eecef71ce;hpb=0ec1c6dc68570b8b824ff55205b4b133bffbf9e4;p=vpp.git

diff --git a/test/vpp_policer.py b/test/vpp_policer.py
index 2c47eed75fd..0f3b073d6e3 100644
--- 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():
@@ -61,9 +67,13 @@ class VppPolicer(VppObject):
         self._test.vapi.policer_bind(name=self.name, worker_index=worker,
                                      bind_enable=bind)
 
-    def apply_vpp_config(self, if_index, apply):
-        self._test.vapi.policer_input(name=self.name, sw_if_index=if_index,
-                                      apply=apply)
+    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(