From bcc61110d5b76580a2da0d72d07de8efd7525292 Mon Sep 17 00:00:00 2001 From: Christopher Baines Date: Tue, 22 Dec 2015 17:59:32 +0000 Subject: Import python-prometheus-client_0.0.13.orig.tar.gz --- tests/test_process_collector.py | 62 +++++++++++++++++++++++++++++++++++++++++ 1 file changed, 62 insertions(+) create mode 100644 tests/test_process_collector.py (limited to 'tests/test_process_collector.py') diff --git a/tests/test_process_collector.py b/tests/test_process_collector.py new file mode 100644 index 0000000..6455d43 --- /dev/null +++ b/tests/test_process_collector.py @@ -0,0 +1,62 @@ +from __future__ import unicode_literals +import os +import unittest + + +from prometheus_client import CollectorRegistry, ProcessCollector + +class TestProcessCollector(unittest.TestCase): + def setUp(self): + self.registry = CollectorRegistry() + self.test_proc = os.path.join(os.path.dirname(os.path.realpath(__file__)), 'proc') + + def test_working(self): + collector = ProcessCollector(proc=self.test_proc, pid=lambda: 26231, registry=self.registry) + collector._ticks = 100 + + self.assertEqual(17.21, self.registry.get_sample_value('process_cpu_seconds_total')) + self.assertEqual(56274944.0, self.registry.get_sample_value('process_virtual_memory_bytes')) + self.assertEqual(8114176, self.registry.get_sample_value('process_resident_memory_bytes')) + self.assertEqual(1418184099.75, self.registry.get_sample_value('process_start_time_seconds')) + self.assertEqual(2048.0, self.registry.get_sample_value('process_max_fds')) + self.assertEqual(5.0, self.registry.get_sample_value('process_open_fds')) + self.assertEqual(None, self.registry.get_sample_value('process_fake_namespace')) + + def test_namespace(self): + collector = ProcessCollector(proc=self.test_proc, pid=lambda: 26231, registry=self.registry, namespace='n') + collector._ticks = 100 + + self.assertEqual(17.21, self.registry.get_sample_value('n_process_cpu_seconds_total')) + self.assertEqual(56274944.0, self.registry.get_sample_value('n_process_virtual_memory_bytes')) + self.assertEqual(8114176, self.registry.get_sample_value('n_process_resident_memory_bytes')) + self.assertEqual(1418184099.75, self.registry.get_sample_value('n_process_start_time_seconds')) + self.assertEqual(2048.0, self.registry.get_sample_value('n_process_max_fds')) + self.assertEqual(5.0, self.registry.get_sample_value('n_process_open_fds')) + self.assertEqual(None, self.registry.get_sample_value('process_cpu_seconds_total')) + + def test_working_584(self): + collector = ProcessCollector(proc=self.test_proc, pid=lambda: "584\n", registry=self.registry) + collector._ticks = 100 + + self.assertEqual(0.0, self.registry.get_sample_value('process_cpu_seconds_total')) + self.assertEqual(10395648.0, self.registry.get_sample_value('process_virtual_memory_bytes')) + self.assertEqual(634880, self.registry.get_sample_value('process_resident_memory_bytes')) + self.assertEqual(1418291667.75, self.registry.get_sample_value('process_start_time_seconds')) + self.assertEqual(None, self.registry.get_sample_value('process_max_fds')) + self.assertEqual(None, self.registry.get_sample_value('process_open_fds')) + + def test_working_fake_pid(self): + collector = ProcessCollector(proc=self.test_proc, pid=lambda: 123, registry=self.registry) + collector._ticks = 100 + + self.assertEqual(None, self.registry.get_sample_value('process_cpu_seconds_total')) + self.assertEqual(None, self.registry.get_sample_value('process_virtual_memory_bytes')) + self.assertEqual(None, self.registry.get_sample_value('process_resident_memory_bytes')) + self.assertEqual(None, self.registry.get_sample_value('process_start_time_seconds')) + self.assertEqual(None, self.registry.get_sample_value('process_max_fds')) + self.assertEqual(None, self.registry.get_sample_value('process_open_fds')) + self.assertEqual(None, self.registry.get_sample_value('process_fake_namespace')) + + +if __name__ == '__main__': + unittest.main() -- cgit v1.2.3