1ea13f3482
Added the ability to get the avg cpu perc since last run instead of blocking and getting the current cpu perc. Change-Id: I46682759ba5495838745a7e3dca2fc0d405df7a3
68 lines
2.4 KiB
Python
68 lines
2.4 KiB
Python
import os
|
|
import psutil
|
|
import unittest
|
|
|
|
from tests.common import load_check
|
|
|
|
|
|
class TestSimpleProcess(unittest.TestCase):
|
|
def setUp(self):
|
|
p = psutil.Process(os.getpid())
|
|
config = {'init_config': {}, 'instances': [{'name': 'test',
|
|
'search_string': [p.name()],
|
|
'detailed': False}]}
|
|
self.check = load_check('process', config)
|
|
|
|
def testPidCount(self):
|
|
self.check.run()
|
|
metrics = self.check.get_metrics()
|
|
self.assertTrue(len(metrics) == 1, metrics)
|
|
self.assertTrue(metrics[0].name == 'process.pid_count')
|
|
|
|
|
|
class TestDetailedProcess(unittest.TestCase):
|
|
def setUp(self):
|
|
p = psutil.Process(os.getpid())
|
|
config = {'init_config': {}, 'instances': [{'name': 'test',
|
|
'search_string': [p.name()],
|
|
'detailed': True}]}
|
|
self.check = load_check('process', config)
|
|
|
|
def testPidCount(self):
|
|
self.check.run()
|
|
metrics = self.check.get_metrics()
|
|
self.assertTrue(len(metrics) > 1, metrics)
|
|
|
|
def run_check(self):
|
|
self.check.run()
|
|
metrics = self.check.get_metrics()
|
|
|
|
measurement_names = []
|
|
for metric in metrics:
|
|
measurement_names.append(metric.name)
|
|
|
|
measurement_names.sort()
|
|
return measurement_names
|
|
|
|
def testMeasurements(self):
|
|
measurement_names = self.run_check()
|
|
|
|
# first run will not have cpu_perc in it
|
|
expected_names = ['process.involuntary_ctx_switches',
|
|
'process.io.read_count',
|
|
'process.io.read_kbytes',
|
|
'process.io.write_count',
|
|
'process.io.write_kbytes',
|
|
'process.mem.real_mbytes',
|
|
'process.mem.rss_mbytes',
|
|
'process.open_file_descriptors',
|
|
'process.pid_count',
|
|
'process.thread_count',
|
|
'process.voluntary_ctx_switches']
|
|
self.assertEquals(measurement_names, expected_names)
|
|
|
|
# run again to get cpu_perc
|
|
expected_names.insert(0, 'process.cpu_perc')
|
|
measurement_names = self.run_check()
|
|
self.assertEquals(measurement_names, expected_names)
|