class TestKafkaInfluxDB(unittest.TestCase): def setUp(self): self.config = Config(100) self.encoder = echo_encoder.Encoder() self.writer = DummyWriter() self.writer = Mock() self.writer.write.return_value = True def test_buffering(self): self.reader = DummyReader( ["myhost.load.load.shortterm 0.05 1436357630"], self.config.buffer_size - 1) self.client = KafkaInfluxDB(self.reader, self.encoder, self.writer, self.config) self.client.consume() self.assertFalse(self.writer.write.called) def test_flush(self): self.reader = DummyReader( ["myhost.load.load.shortterm 0.05 1436357630"], self.config.buffer_size) self.client = KafkaInfluxDB(self.reader, self.encoder, self.writer, self.config) self.client.consume() self.assertTrue(self.writer.write.called)
def test_flush(self): self.reader = DummyReader( ["myhost.load.load.shortterm 0.05 1436357630"], self.config.buffer_size) self.client = KafkaInfluxDB(self.reader, self.encoder, self.writer, self.config) self.client.consume() self.assertTrue(self.writer.write.called)
class TestKafkaInfluxDB(unittest.TestCase): def setUp(self): self.config = Config(100) self.encoder = echo_encoder.Encoder() self.writer = DummyWriter() self.writer = Mock() self.writer.write.return_value = True def test_buffering(self): self.reader = DummyReader(["myhost.load.load.shortterm 0.05 1436357630"], self.config.buffer_size - 1) self.client = KafkaInfluxDB(self.reader, self.encoder, self.writer, self.config) self.client.consume() self.assertFalse(self.writer.write.called) def test_flush(self): self.reader = DummyReader(["myhost.load.load.shortterm 0.05 1436357630"], self.config.buffer_size) self.client = KafkaInfluxDB(self.reader, self.encoder, self.writer, self.config) self.client.consume() self.assertTrue(self.writer.write.called)
def test_flush(self): self.reader = DummyReader(["myhost.load.load.shortterm 0.05 1436357630"], self.config.buffer_size) self.client = KafkaInfluxDB(self.reader, self.encoder, self.writer, self.config) self.client.consume() self.assertTrue(self.writer.write.called)