def setup(self):
self._patcher = patch('zabby.hostos.collectors.time')
self.mock_time = self._patcher.start()
self.now = 0
def increment_and_return_now():
self.now += 1
return self.now
self.mock_time.side_effect = increment_and_return_now
self.host_os = Mock()
self.stat = 0
def increment_and_return_stats(device):
self.stat += 1
return DiskDeviceStats(read_sectors=0,
read_operations=self.stat,
read_bytes=0,
write_sectors=0,
write_operations=0,
write_bytes=0)
self.host_os.disk_device_stats.side_effect = increment_and_return_stats
self.host_os.disk_device_names.return_value = set([DEVICE_NAME])
self.shift = 5
self.current_time = self.shift + 1
self.collector = DiskDeviceStatsCollector(self.shift, self.host_os)
def __init__(self):
super(Linux, self).__init__()
self._disk_device_stats_collector = DiskDeviceStatsCollector(900, self)
self._collectors.append(self._disk_device_stats_collector)
self._cpu_times_collector = CpuTimesCollector(900, self)
self._collectors.append(self._cpu_times_collector)
