display cpu+mem usage

molotov/dummy.py

@@ -18,4 +18,11 @@ async def scenario_one(session):
     async with session.get("http://example.com") as resp:
         if random.randint(1, 100) == 5:
             raise AssertionError("Failed")
+        from base64 import b64encode
+        from os import urandom
+
+        for i in range(1000):
+            random_bytes = urandom(64)
+            token = b64encode(random_bytes).decode("utf-8")
+
         assert resp.status == 200

molotov/run.py

@@ -96,7 +96,7 @@ def _parser():
         "--console-update",
         help="Delay between each console update",
         type=float,
-        default=0.2,
+        default=0.5,
     )
 
     parser.add_argument(

molotov/ui/app.py

@@ -30,7 +30,7 @@
 class MolotovApp:
     def __init__(
         self,
-        refresh_interval=0.3,
+        refresh_interval=0.5,
         max_lines=25,
         simple_console=False,
         single_process=True,

molotov/ui/controllers.py

@@ -5,6 +5,7 @@
 from datetime import datetime
 
 import humanize
+import psutil
 from prompt_toolkit import HTML
 from prompt_toolkit.formatted_text import to_formatted_text
 from prompt_toolkit.key_binding import KeyBindings
@@ -133,19 +134,35 @@ def __init__(self, max_lines=25):
         super().__init__(max_lines)
         self._status = {}
         self._started = datetime.now()
+        self._p = psutil.Process()
 
     def update(self, results):
         self._status.update(results)
 
+    def get_resource_usage(self):
+        rss = self._p.memory_info().rss
+        cpu_percent = self._p.cpu_percent()
+
+        for child in self._p.children(recursive=True):
+            rss += child.memory_info().rss
+            cpu_percent += child.cpu_percent()
+
+        if cpu_percent > 0:
+            cpu_percent = cpu_percent / psutil.cpu_count()
+        return {"rss": rss, "cpu_percent": cpu_percent}
+
     def formatted(self):
         delta = datetime.now() - self._started
+        resources = self.get_resource_usage()
+
         return to_formatted_text(
             HTML(
                 f'<style fg="green" bg="#cecece">SUCCESS: {self._status.get("OK", 0)} </style>'
                 f'<style fg="red" bg="#cecece"> FAILED: {self._status.get("FAILED", 0)} </style>'
                 f' WORKERS: {self._status.get("WORKER", 0)}'
                 f' PROCESSES: {self._status.get("PROCESS", 0)} '
                 f'<style fg="blue" bg="#cecece"> ELAPSED: {humanize.precisedelta(delta)}</style>'
+                f' CPU% {resources["cpu_percent"]} MEM {humanize.naturalsize(resources["rss"])}'
             )
         )
 

requirements.txt

@@ -3,3 +3,4 @@ aiodogstatsd==0.16.0
 multiprocess==0.70.13
 prompt_toolkit==3.0.31
 humanize==4.4.0
+psutil==5.9.3

setup.py

@@ -12,6 +12,7 @@
     "multiprocess",
     "humanize",
     "prompt_toolkit",
+    "psutil"
 ]
 
 description = ""