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

diff --git a/test/test_srmpls.py b/test/test_srmpls.py
index ded4a71fa40..7e729658b43 100644
--- a/test/test_srmpls.py
+++ b/test/test_srmpls.py
@@ -4,10 +4,9 @@ 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()