X-Git-Url: https://gerrit.fd.io/r/gitweb?a=blobdiff_plain;f=test%2Ftest_srmpls.py;h=b9abeaeffecee1105aa9a5287403ea6e490529b5;hb=cbe25aab3be72154f2c706c39eeba6a77f34450f;hp=ded4a71fa4093162217496012e4bc8b45ccb099a;hpb=7c922dc404c2c0a2d67d53ca05db1c1ae1598f44;p=vpp.git

diff --git a/test/test_srmpls.py b/test/test_srmpls.py
index ded4a71fa40..b9abeaeffec 100644
--- a/test/test_srmpls.py
+++ b/test/test_srmpls.py
@@ -1,13 +1,12 @@
-#!/usr/bin/env python
+#!/usr/bin/env python3
 
 import unittest
 import socket
 
 from framework import VppTestCase, VppTestRunner
+from vpp_ip import DpoProto
 from vpp_ip_route import VppIpRoute, VppRoutePath, VppMplsRoute, \
-    VppMplsIpBind, VppIpMRoute, VppMRoutePath, \
-    MRouteItfFlags, MRouteEntryFlags, DpoProto, VppIpTable, VppMplsTable, \
-    VppMplsLabel, MplsLspMode
+    VppIpTable, VppMplsTable, VppMplsLabel
 from vpp_mpls_tunnel_interface import VppMPLSTunnelInterface
 
 from scapy.packet import Raw
@@ -50,6 +49,14 @@ def verify_mpls_stack(tst, rx, mpls_labels):
 class TestSRMPLS(VppTestCase):
     """ SR-MPLS Test Case """
 
+    @classmethod
+    def setUpClass(cls):
+        super(TestSRMPLS, cls).setUpClass()
+
+    @classmethod
+    def tearDownClass(cls):
+        super(TestSRMPLS, cls).tearDownClass()
+
     def setUp(self):
         super(TestSRMPLS, self).setUp()
 
@@ -77,7 +84,6 @@ class TestSRMPLS(VppTestCase):
         for i in self.pg_interfaces:
             i.unconfig_ip4()
             i.unconfig_ip6()
-            i.ip6_disable()
             i.disable_mpls()
             i.admin_down()
         super(TestSRMPLS, self).tearDown()