コード例 #1
0
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)
コード例 #2
0
 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)
コード例 #3
0
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)
コード例 #4
0
 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)