def test_WhisperReader_broken_file(self): self.create_whisper_hosts() self.addCleanup(self.wipe_whisper_hosts) # Test broken whisper file f = open(self.worker2, 'rb+') f.seek(10) f.write('Bad Data') f.close() reader = WhisperReader(self.worker2, 'hosts.worker2.cpu') with self.assertRaises(Exception): reader.fetch(self.start_ts-5, self.start_ts)
def test_WhisperReader_broken_file(self): self.create_whisper_hosts() self.addCleanup(self.wipe_whisper_hosts) # Test broken whisper file f = open(self.worker2, 'rb+') f.seek(10) f.write(b'Bad Data') f.close() reader = WhisperReader(self.worker2, 'hosts.worker2.cpu') with self.assertRaises(Exception): reader.fetch(self.start_ts - 5, self.start_ts)
def test_WhisperReader_missing_file(self, log_exception): path = 'missing/file.wsp' reader = WhisperReader(path, 'hosts.worker2.cpu') self.assertEqual(reader.fetch(self.start_ts - 5, self.start_ts), None) log_exception.assert_called_with("Failed fetch of whisper file '%s'" % path)
def test_WhisperReader_fetch(self): self.create_whisper_hosts() self.addCleanup(self.wipe_whisper_hosts) reader = WhisperReader(self.worker1, 'hosts.worker1.cpu') (_, values) = reader.fetch(self.start_ts - 5, self.start_ts) self.assertEqual(values, [None, None, None, None, 1.0])
def test_WhisperReader_fetch(self): self.create_whisper_hosts() self.addCleanup(self.wipe_whisper_hosts) reader = WhisperReader(self.worker1, 'hosts.worker1.cpu') (_, values) = reader.fetch(self.start_ts-5, self.start_ts) self.assertEqual(values, [None, None, None, None, 1.0])
def test_WhisperReader_fetch_returns_no_data(self, whisper_fetch): self.create_whisper_hosts() self.addCleanup(self.wipe_whisper_hosts) reader = WhisperReader(self.worker1, 'hosts.worker1.cpu') whisper_fetch.return_value = None self.assertEqual(reader.fetch(self.start_ts - 5, self.start_ts), None)
def test_WhisperReader_fetch_returns_no_data(self, whisper_fetch): self.create_whisper_hosts() self.addCleanup(self.wipe_whisper_hosts) reader = WhisperReader(self.worker1, 'hosts.worker1.cpu') whisper_fetch.return_value = None self.assertEqual(reader.fetch(self.start_ts-5, self.start_ts), None)
def test_WhisperReader_CarbonLinkQuery(self, carbonlink_query): self.create_whisper_hosts() self.addCleanup(self.wipe_whisper_hosts) carbonlink_query.return_value = {} reader = WhisperReader(self.worker1, 'hosts.worker1.cpu') (_, values) = reader.fetch(self.start_ts - 5, self.start_ts) self.assertEqual(values, [None, None, None, None, 1.0])
def test_WhisperReader_CarbonLinkQuery(self, carbonlink_query): self.create_whisper_hosts() self.addCleanup(self.wipe_whisper_hosts) carbonlink_query.return_value = {} reader = WhisperReader(self.worker1, 'hosts.worker1.cpu') (_, values) = reader.fetch(self.start_ts-5, self.start_ts) self.assertEqual(values, [None, None, None, None, 1.0])