scripts: sysmon: add thread count metric
BUG=b:255782067
TEST=Ran the unit test.
Change-Id: If961c0beca90df44f0342af5c4b45bc1c43b6066
Reviewed-on: https://chromium-review.googlesource.com/c/chromiumos/chromite/+/4242243
Auto-Submit: Congbin Guo <guocb@chromium.org>
Commit-Queue: Sergey Fetisov <sfetisov@google.com>
Reviewed-by: Sergey Fetisov <sfetisov@google.com>
Tested-by: Congbin Guo <guocb@chromium.org>
diff --git a/scripts/sysmon/proc_metrics_unittest.py b/scripts/sysmon/proc_metrics_unittest.py
index 5b45462..f5bdbbe 100644
--- a/scripts/sysmon/proc_metrics_unittest.py
+++ b/scripts/sysmon/proc_metrics_unittest.py
@@ -16,10 +16,11 @@
from chromite.scripts.sysmon import proc_metrics
-def _mock_process(name, cmdline, parent=None):
+def _mock_process(name, cmdline, parent=None, num_threads=10):
proc = mock.Mock(dir(psutil.Process))
proc.name.return_value = name
proc.cmdline.return_value = cmdline
+ proc.num_threads.return_value = num_threads
proc.cpu_percent.return_value = 2
if parent is not None:
proc.parent.return_value = parent
@@ -35,6 +36,7 @@
"""Return expected calls for a process metric."""
return [
mock.call("proc/count", (name,), None, 1, enforce_ge=mock.ANY),
+ mock.call("proc/thread_count", (name,), None, 10, enforce_ge=mock.ANY),
mock.call("proc/cpu_percent", (name,), None, 2, enforce_ge=mock.ANY),
]