class KafkaGraphiteServiceTest(unittest.TestCase): def setUp(self): self.collector = KafkaGraphiteService( command="/opt/kafka/bin/kafka-consumer-offset-checker.sh", zookeeper="localhost", group="group1", topic="events") self.collector.protocol = FakeGraphiteProtocol() self.offsets = { 0: {'lag': 22, 'logSize': 734982587, 'offset': 734982565, 'owner': 'group1_consumer.example.org-1436585515700-8c4fbc41-0', }, 1: {'lag': 23, 'logSize': 824687314, 'offset': 824687291, 'owner': 'group1_consumer.example.org-1436585515700-8c4fbc41-0', }, 2: {'lag': None, 'logSize': 551840689, 'offset': None, 'owner': 'group1_consumer.example.org-1436596330605-29002ae5-0', } } def test_gotOffsets(self): self.collector.gotOffsets(self.offsets, 'prefix') result = self.collector.protocol.output self.assertEqual(22.0, result['prefix.0.lag'][0]) def test_gotUnknownLag(self): self.collector.gotOffsets(self.offsets, 'prefix') result = self.collector.protocol.output self.assertNotIn('prefix.2.lag', result)
def setUp(self): self.collector = KafkaGraphiteService( command="/opt/kafka/bin/kafka-consumer-offset-checker.sh", zookeeper="localhost", group="group1", topic="events") self.collector.protocol = FakeGraphiteProtocol() self.offsets = { 0: {'lag': 22, 'logSize': 734982587, 'offset': 734982565, 'owner': 'group1_consumer.example.org-1436585515700-8c4fbc41-0', }, 1: {'lag': 23, 'logSize': 824687314, 'offset': 824687291, 'owner': 'group1_consumer.example.org-1436585515700-8c4fbc41-0', }, 2: {'lag': None, 'logSize': 551840689, 'offset': None, 'owner': 'group1_consumer.example.org-1436596330605-29002ae5-0', } }