test_node: get_mem_rss fixups

MarcoFalke · furszy · commit 0edfce565c07 · 2021-06-28T11:44:47.000-03:00
diff --git a/test/functional/test_framework/test_node.py b/test/functional/test_framework/test_node.py
@@ -84,22 +84,19 @@ def __init__(self, i, dirname, extra_args, rpchost, timewait, binary, stderr, mo
     def get_mem_rss(self):
         """Get the memory usage (RSS) per `ps`.
 
-        If process is stopped or `ps` is unavailable, return None.
+        Returns None if `ps` is unavailable.
         """
-        if not (self.running and self.process):
-            self.log.warning("Couldn't get memory usage; process isn't running.")
-            return None
+        assert self.running
 
         try:
             return int(subprocess.check_output(
-                "ps h -o rss {}".format(self.process.pid),
-                shell=True, stderr=subprocess.DEVNULL).strip())
+                ["ps", "h", "-o", "rss", "{}".format(self.process.pid)],
+                stderr=subprocess.DEVNULL).split()[-1])
 
-        # Catching `Exception` broadly to avoid failing on platforms where ps
-        # isn't installed or doesn't work as expected, e.g. OpenBSD.
+        # Avoid failing on platforms where ps isn't installed.
         #
         # We could later use something like `psutils` to work across platforms.
-        except Exception:
+        except (FileNotFoundError, subprocess.SubprocessError):
             self.log.exception("Unable to get memory usage")
             return None
 
@@ -262,7 +259,7 @@ def assert_memory_usage_stable(self, perc_increase_allowed=0.03):
             self.log.warning("Unable to detect memory usage (RSS) - skipping memory check.")
             return
 
-        perc_increase_memory_usage = 1 - (float(before_memory_usage) / after_memory_usage)
+        perc_increase_memory_usage = (after_memory_usage / before_memory_usage) - 1
 
         if perc_increase_memory_usage > perc_increase_allowed:
             raise AssertionError(
