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',
}
}
