Added trap test cases

IGordynskyi · Aug 11, 2016 · 1446459 · 1446459
1 parent 6c3d18d
commit 1446459
Showing 1 changed file with 44 additions and 0 deletions.
diff --git a/test/saithrift/tests/saiacl.py b/test/saithrift/tests/saiacl.py
@@ -206,6 +206,8 @@ def runTestFunc(self, action = 0, setup_acltbl=True, **kwargs):
             self.verifyRouting(PKT, EXP_PKT, PORT_IN, PORT_OUT)
         elif action == SAI_PACKET_ACTION_COPY:
             self.verifyRouting(PKT, EXP_PKT, PORT_IN, PORT_OUT)
+        elif action == SAI_PACKET_ACTION_TRAP:
+            self.verifyNoRouting(PKT, EXP_PKT, PORT_IN, PORT_OUT)
 
     def runTest(self):
         print
@@ -350,6 +352,48 @@ def runTestFunc(self):
                                                    mac_dst = router_mac)
         print "Destination MAC address copied to CPU"
 
+###############################################################################################################
+################################# testing TRAP action (SAI_ACL_STAGE_INGRESS) #################################
+###############################################################################################################
+
+@group('acl')
+class testAclSrcIpTrap(sai_base_test.ThriftInterfaceDataPlane, aclTestBase):
+    def runTestFunc(self):
+        super(testAclSrcIpTrap, self).runTestFunc(stage = SAI_ACL_STAGE_INGRESS,
+                                                  action = SAI_PACKET_ACTION_TRAP,
+                                                  in_ports = [port_list[PORT_OUT], port_list[PORT_IN]],
+                                                  ip_src = IP_SRC,
+                                                  ip_src_mask = IP_MASK)
+        print "Source IPv4 address droped in port 1 and forwarded to CPU"
+
+@group('acl')
+class testAclDstIpTrap(sai_base_test.ThriftInterfaceDataPlane, aclTestBase):
+    def runTestFunc(self):
+        super(testAclDstIpTrap, self).runTestFunc(stage = SAI_ACL_STAGE_INGRESS,
+                                                  action = SAI_PACKET_ACTION_TRAP,
+                                                  in_ports = [port_list[PORT_OUT], port_list[PORT_IN]],
+                                                  ip_dst = IP_DST,
+                                                  ip_dst_mask = IP_MASK)
+        print "Destination IPv4 address droped in port 1 and forwarded to CPU"
+
+@group('acl')
+class testAclSrcMacTrap(sai_base_test.ThriftInterfaceDataPlane, aclTestBase):
+    def runTestFunc(self):
+        super(testAclSrcMacTrap, self).runTestFunc(stage = SAI_ACL_STAGE_INGRESS,
+                                                   action = SAI_PACKET_ACTION_TRAP,
+                                                   in_ports = [port_list[PORT_OUT], port_list[PORT_IN]],
+                                                   mac_src = rewrite_mac1)
+        print "Source MAC address droped in port 1 and forwarded to CPU"
+
+@group('acl')
+class testAclDstMacTrap(sai_base_test.ThriftInterfaceDataPlane, aclTestBase):
+    def runTestFunc(self):
+        super(testAclDstMacTrap, self).runTestFunc(stage = SAI_ACL_STAGE_INGRESS,
+                                                   action = SAI_PACKET_ACTION_TRAP,
+                                                   in_ports = [port_list[PORT_OUT], port_list[PORT_IN]],
+                                                   mac_dst = router_mac)
+        print "Destination MAC address droped in port 1 and forwarded to CPU"
+
 ###############################################################################################################
 ############################### testing REDIRECT action (SAI_ACL_STAGE_INGRESS) ###############################
 ###############################################################################################################