Tests: Refactor tearDown show command logging, add lifecycle markers.

[vpp.git] / test / test_lb.py

diff --git a/test/test_lb.py b/test/test_lb.py
index 01af0e5..93b389a 100644 (file)
--- a/test/test_lb.py
+++ b/test/test_lb.py
@@ -63,10 +63,15 @@ class TestLB(VppTestCase):
             super(TestLB, cls).tearDownClass()
             raise
 
+    @classmethod
+    def tearDownClass(cls):
+        super(TestLB, cls).tearDownClass()
+
     def tearDown(self):
         super(TestLB, self).tearDown()
-        if not self.vpp_dead:
-            self.logger.info(self.vapi.cli("show lb vip verbose"))
+
+    def show_commands_at_teardown(self):
+        self.logger.info(self.vapi.cli("show lb vip verbose"))
 
     def getIPv4Flow(self, id):
         return (IP(dst="90.0.%u.%u" % (id / 255, id % 255),