X-Git-Url: https://gerrit.fd.io/r/gitweb?a=blobdiff_plain;f=test%2Ftest_pipe.py;h=6a0b4a0d093dea4ee9e44ccd36711b9b4df18997;hb=8a6e1b190c63cc596aed33f84b5989b64d276423;hp=a89c1e90236176fba37e18b39cbf6b691b354b3b;hpb=17ff3c1fa5687255a118c53223fa2cd49132d929;p=vpp.git diff --git a/test/test_pipe.py b/test/test_pipe.py index a89c1e90236..6a0b4a0d093 100644 --- a/test/test_pipe.py +++ b/test/test_pipe.py @@ -1,14 +1,15 @@ #!/usr/bin/env python from socket import AF_INET, AF_INET6, inet_pton - -from framework import VppTestCase, VppTestRunner -from vpp_interface import VppInterface -from vpp_ip_route import VppIpTable, VppIpRoute, VppRoutePath +import unittest from scapy.packet import Raw from scapy.layers.l2 import Ether from scapy.layers.inet import IP, UDP +from framework import VppTestCase, VppTestRunner +from vpp_interface import VppInterface +from vpp_ip_route import VppIpTable, VppIpRoute, VppRoutePath + class VppPipe(VppInterface): """ @@ -32,14 +33,11 @@ class VppPipe(VppInterface): self.result = self._test.vapi.pipe_create( 0 if self.instance == 0xffffffff else 1, self.instance) - self.set_sw_if_index(self.result.parent_sw_if_index) + self.set_sw_if_index(self.result.sw_if_index) def remove_vpp_config(self): self._test.vapi.pipe_delete( - self.result.parent_sw_if_index) - - def __str__(self): - return self.object_id() + self.result.sw_if_index) def object_id(self): return "pipe-%d" % (self._sw_if_index) @@ -47,20 +45,28 @@ class VppPipe(VppInterface): def query_vpp_config(self): pipes = self._test.vapi.pipe_dump() for p in pipes: - if p.parent_sw_if_index == self.result.parent_sw_if_index: + if p.sw_if_index == self.result.sw_if_index: return True return False def set_unnumbered(self, ip_sw_if_index, is_add=True): - res = self._test.vapi.sw_interface_set_unnumbered( - self.east, ip_sw_if_index, is_add) - res = self._test.vapi.sw_interface_set_unnumbered( - self.west, ip_sw_if_index, is_add) + res = self._test.vapi.sw_interface_set_unnumbered(ip_sw_if_index, + self.east, is_add) + res = self._test.vapi.sw_interface_set_unnumbered(ip_sw_if_index, + self.west, is_add) class TestPipe(VppTestCase): """ Pipes """ + @classmethod + def setUpClass(cls): + super(TestPipe, cls).setUpClass() + + @classmethod + def tearDownClass(cls): + super(TestPipe, cls).tearDownClass() + def setUp(self): super(TestPipe, self).setUp() @@ -78,9 +84,7 @@ class TestPipe(VppTestCase): def test_pipe(self): """ Pipes """ - pipes = [] - pipes.append(VppPipe(self)) - pipes.append(VppPipe(self, 10)) + pipes = [VppPipe(self), VppPipe(self, 10)] for p in pipes: p.add_vpp_config() @@ -102,7 +106,7 @@ class TestPipe(VppTestCase): self.pg1.sw_if_index, enable=1) - # test bi-drectional L2 flow pg0<->pg1 + # test bi-directional L2 flow pg0<->pg1 p = (Ether(src=self.pg0.remote_mac, dst=self.pg1.remote_mac) / IP(src="1.1.1.1",