From: Rafael Zalamena <rzalamena@gmail.com>
Date: Thu, 22 Jun 2017 18:55:42 +0000 (-0300)
Subject: topogen: implement memory leak test call
X-Git-Tag: frr-7.1-dev~151^2~307
X-Git-Url: https://git.puffer.fish/?a=commitdiff_plain;h=38c3993248ac22e4a85758038cb1f3c5d3f725d4;p=matthieu%2Ffrr.git

topogen: implement memory leak test call

This is a shortcut for the memory leak test, it should be used in the
new tests in order to avoid boilerplate code.
---

diff --git a/tests/topotests/lib/topogen.py b/tests/topotests/lib/topogen.py
index 08bff0fa16..e87b6bd0a7 100644
--- a/tests/topotests/lib/topogen.py
+++ b/tests/topotests/lib/topogen.py
@@ -411,6 +411,22 @@ class TopoRouter(TopoGear):
 
         return res
 
+    def report_memory_leaks(self, testname):
+        """
+        Runs the router memory leak check test. Has the following parameter:
+        testname: the test file name for identification
+
+        NOTE: to run this you must have the environment variable
+        TOPOTESTS_CHECK_MEMLEAK set to the appropriated path.
+        """
+        memleak_file = os.environ.get('TOPOTESTS_CHECK_MEMLEAK')
+        if memleak_file is None:
+            print "SKIPPED check on Memory leaks: Disabled (TOPOTESTS_CHECK_MEMLEAK undefined)"
+            return
+
+        self.tgen.net[self.name].stopRouter()
+        self.tgen.net[self.name].report_memory_leaks(memleak_file, testname)
+
 class TopoSwitch(TopoGear):
     """
     Switch abstraction. Has the following properties: