def test_local_with_engine(self): config = {'metrics': ['cpu', 'engine-loop']} obj = LocalClient(logging.getLogger(''), 'label', config) obj.engine = EngineEmul() data = obj.get_data() self.assertTrue(all('source' in item.keys() and 'ts' in item.keys() for item in data)) return data
def test_local_with_engine(self): config = {'metrics': ['cpu', 'engine-loop']} obj = LocalClient(logging.getLogger(''), 'label', config) obj.engine = EngineEmul() obj.connect() data = obj.get_data() self.assertTrue(all('source' in item.keys() and 'ts' in item.keys() for item in data)) return data
def test_deprecated_local(self): # strange metrics, anyway full list of available ones will be used config = {'metrics': ['memory', 'disk']} obj = LocalClient(ROOT_LOGGER, 'label', config) obj.engine = EngineEmul() obj.connect() data = obj.engine_resource_stats() self.assertTrue(all(val is not None for val in [ data.cpu, data.disk_usage, data.mem_usage, data.mem_usage, data.rx, data.tx, data.dru, data.dwu, data.engine_loop, data.conn_all]))
def test_deprecated_local(self): # strange metrics, anyway full list of available ones will be used config = {'metrics': ['memory', 'disk']} obj = LocalClient(logging.getLogger(''), 'label', config) obj.engine = EngineEmul() obj.connect() data = obj.engine_resource_stats() self.assertTrue(all(val is not None for val in [ data.cpu, data.disk_usage, data.mem_usage, data.mem_usage, data.rx, data.tx, data.dru, data.dwu, data.engine_loop, data.conn_all]))
def test_psutil_potential_bugs(self): client = LocalClient(logging.getLogger(''), 'label', {'metrics': ['cpu', 'mem', 'disks', 'conn-all']}) client.engine = EngineEmul() client.connect() try: import psutil net_io_counters = psutil.net_io_counters disk_io_counters = psutil.disk_io_counters psutil.net_io_counters = lambda: None psutil.disk_io_counters = lambda: None client.engine_resource_stats() # should throw no exception finally: psutil.net_io_counters = net_io_counters psutil.disk_io_counters = disk_io_counters