def test_disk_info_removable(self): path = '/does/not/exist' system_metrics.get_disk_info(mountpoints=[path]) self.assertIs(None, system_metrics.disk_free.get({'path': path})) self.assertIs(None, system_metrics.disk_total.get({'path': path}))
def test_disk_counters_no_disks(self, mock_counters): mock_counters.side_effect = RuntimeError("couldn't find any physical disk") # Should swallow the exception. system_metrics.get_disk_info() self.assertTrue(mock_counters.called) self.assertIs(None, system_metrics.disk_read.get({'disk': 'sda'})) self.assertIs(None, system_metrics.disk_write.get({'disk': 'sda'}))
def test_disk_counters(self, mock_counters): iostat = collections.namedtuple('iostat', ['read_bytes', 'write_bytes']) mock_counters.return_value = {'sda': iostat(123, 456)} system_metrics.get_disk_info() self.assertTrue(mock_counters.called) self.assertEqual(123, system_metrics.disk_read.get({'disk': 'sda'})) self.assertEqual(456, system_metrics.disk_write.get({'disk': 'sda'}))
def single_iteration(): try: system_metrics.get_cpu_info() system_metrics.get_disk_info() system_metrics.get_mem_info() system_metrics.get_net_info() system_metrics.get_proc_info() puppet_metrics.get_puppet_summary() finally: ts_mon.flush() return True
def test_disk_info(self): system_metrics.get_disk_info() # A disk mountpoint that should always be present. path = 'C:\\' if sys.platform == 'win32' else '/' free = system_metrics.disk_free.get({'path': path}) total = system_metrics.disk_total.get({'path': path}) self.assertIsNotNone(free) self.assertIsNotNone(total) self.assertLessEqual(free, total) if os.name == 'posix': # pragma: no cover inodes_free = system_metrics.inodes_free.get({'path': '/'}) inodes_total = system_metrics.inodes_total.get({'path': '/'}) self.assertIsNotNone(inodes_free) self.assertIsNotNone(inodes_total) self.assertLessEqual(inodes_free, inodes_total)
def test_disk_counters_other_exception(self, mock_counters): mock_counters.side_effect = RuntimeError('different message') with self.assertRaises(RuntimeError): system_metrics.get_disk_info()