Example #1
0
class PyMetricsTest(unittest.TestCase):
    def setUp(self):
        metrics_collector = mock_generator.MockMetricsCollector()
        with patch(
                "heron.common.src.python.config.system_config.get_sys_config",
                side_effect=lambda:
            {constants.HERON_METRICS_EXPORT_INTERVAL_SEC: 10}):
            self.py_metrics = PyMetrics(metrics_collector)
        self.py_metrics.process = Mock()
        self.py_metrics.process.cpu_times = Mock(
            return_value=Cputime(system=23, user=29))
        t1 = Thread(id=100, user_time=3, system_time=5)
        t2 = Thread(id=500, user_time=7, system_time=11)
        self.py_metrics.process.threads = Mock(return_value=[t1, t2])
        self.py_metrics.process.num_fds = Mock(return_value=3)
        self.py_metrics.process.memory_info = Mock(
            return_value=Mem(rss=13, vms=19))

    def test_gc(self):
        with patch("gc.get_count", side_effect=lambda: (1, 2, 3)):
            with patch("gc.get_threshold", side_effect=lambda: (4, 5, 6)):
                self.py_metrics.update_gc_stat()
        self.assertEqual(self.py_metrics.g1_count.get_value_and_reset(), 1)
        self.assertEqual(self.py_metrics.g2_count.get_value_and_reset(), 2)
        self.assertEqual(self.py_metrics.g3_count.get_value_and_reset(), 3)
        self.assertEqual(self.py_metrics.g1_threshold.get_value_and_reset(), 4)
        self.assertEqual(self.py_metrics.g2_threshold.get_value_and_reset(), 5)
        self.assertEqual(self.py_metrics.g3_threshold.get_value_and_reset(), 6)

    def test_cpu_times(self):
        self.py_metrics.update_cpu_time()
        self.assertEqual(self.py_metrics.sys_cpu_time.get_value_and_reset(),
                         23)
        self.assertEqual(self.py_metrics.user_cpu_time.get_value_and_reset(),
                         29)

    def test_threads_time(self):
        self.py_metrics.update_threads_time()
        tmap = self.py_metrics.threads.get_value_and_reset()
        self.assertEqual(tmap[100], (3, 5))
        self.assertEqual(tmap[500], (7, 11))

    def test_fd_num(self):
        self.py_metrics.update_fds()
        self.assertEqual(self.py_metrics.fd_nums.get_value_and_reset(), 3)

    def test_mem(self):
        self.py_metrics.update_memory_usage()
        self.assertEqual(self.py_metrics.physical_memory.get_value_and_reset(),
                         13)
        self.assertEqual(self.py_metrics.virtual_memory.get_value_and_reset(),
                         19)
Example #2
0
 def setUp(self):
     metrics_collector = mock_generator.MockMetricsCollector()
     with patch(
             "heron.common.src.python.config.system_config.get_sys_config",
             side_effect=lambda:
         {constants.HERON_METRICS_EXPORT_INTERVAL_SEC: 10}):
         self.py_metrics = PyMetrics(metrics_collector)
     self.py_metrics.process = Mock()
     self.py_metrics.process.cpu_times = Mock(
         return_value=Cputime(system=23, user=29))
     t1 = Thread(id=100, user_time=3, system_time=5)
     t2 = Thread(id=500, user_time=7, system_time=11)
     self.py_metrics.process.threads = Mock(return_value=[t1, t2])
     self.py_metrics.process.num_fds = Mock(return_value=3)
     self.py_metrics.process.memory_info = Mock(
         return_value=Mem(rss=13, vms=19))
Example #3
0
class PyMetricsTest(unittest.TestCase):
  def setUp(self):
    metrics_collector = mock_generator.MockMetricsCollector()
    with patch("heron.common.src.python.config.system_config.get_sys_config",
               side_effect=lambda: {constants.HERON_METRICS_EXPORT_INTERVAL_SEC: 10}):
          self.py_metrics = PyMetrics(metrics_collector)
    self.py_metrics.process = Mock()
    self.py_metrics.process.cpu_times = Mock(return_value=Cputime(system=23, user=29))
    t1 = Thread(id=100, user_time=3, system_time=5)
    t2 = Thread(id=500, user_time=7, system_time=11)
    self.py_metrics.process.threads = Mock(return_value=[t1, t2])
    self.py_metrics.process.num_fds = Mock(return_value=3)
    self.py_metrics.process.memory_info = Mock(return_value=Mem(rss=13, vms=19))
  
  def test_gc(self):
    with patch("gc.get_count", side_effect=lambda:(1, 2, 3)):
      with patch("gc.get_threshold", side_effect=lambda:(4, 5, 6)):
        self.py_metrics.update_gc_stat()
    self.assertEqual(self.py_metrics.g1_count.get_value_and_reset(), 1)
    self.assertEqual(self.py_metrics.g2_count.get_value_and_reset(), 2)
    self.assertEqual(self.py_metrics.g3_count.get_value_and_reset(), 3)
    self.assertEqual(self.py_metrics.g1_threshold.get_value_and_reset(), 4)
    self.assertEqual(self.py_metrics.g2_threshold.get_value_and_reset(), 5)
    self.assertEqual(self.py_metrics.g3_threshold.get_value_and_reset(), 6)
    
  def test_cpu_times(self):
    self.py_metrics.update_cpu_time()
    self.assertEqual(self.py_metrics.sys_cpu_time.get_value_and_reset(), 23)
    self.assertEqual(self.py_metrics.user_cpu_time.get_value_and_reset(), 29)
    
  def test_threads_time(self):
    self.py_metrics.update_threads_time()
    tmap = self.py_metrics.threads.get_value_and_reset()
    self.assertEqual(tmap[100], (3, 5))
    self.assertEqual(tmap[500], (7, 11))
    
  def test_fd_num(self):
    self.py_metrics.update_fds()
    self.assertEqual(self.py_metrics.fd_nums.get_value_and_reset(), 3)
    
  def test_mem(self):
    self.py_metrics.update_memory_usage()
    self.assertEqual(self.py_metrics.physical_memory.get_value_and_reset(), 13)
    self.assertEqual(self.py_metrics.virtual_memory.get_value_and_reset(), 19)
Example #4
0
 def setUp(self):
   metrics_collector = mock_generator.MockMetricsCollector()
   with patch("heron.common.src.python.config.system_config.get_sys_config",
              side_effect=lambda: {constants.HERON_METRICS_EXPORT_INTERVAL_SEC: 10}):
         self.py_metrics = PyMetrics(metrics_collector)
   self.py_metrics.process = Mock()
   self.py_metrics.process.cpu_times = Mock(return_value=Cputime(system=23, user=29))
   t1 = Thread(id=100, user_time=3, system_time=5)
   t2 = Thread(id=500, user_time=7, system_time=11)
   self.py_metrics.process.threads = Mock(return_value=[t1, t2])
   self.py_metrics.process.num_fds = Mock(return_value=3)
   self.py_metrics.process.memory_info = Mock(return_value=Mem(rss=13, vms=19))