def test_on_tp_commit(self, monitor: PrometheusMonitor, metrics: FaustMetrics) -> None: offsets = {TP("foo", 0): 1001, TP("foo", 1): 2002, TP("bar", 3): 3003} monitor.on_tp_commit(offsets) self.assert_has_sample_value( metrics.topic_partition_offset_commited, "topic_partition_offset_commited", { "topic": "foo", "partition": "0" }, 1001, ) self.assert_has_sample_value( metrics.topic_partition_offset_commited, "topic_partition_offset_commited", { "topic": "foo", "partition": "1" }, 2002, ) self.assert_has_sample_value( metrics.topic_partition_offset_commited, "topic_partition_offset_commited", { "topic": "bar", "partition": "3" }, 3003, )
def _handle_event( self, monitor: PrometheusMonitor, topic_partition: TP, stream: StreamT, event: EventT, offset: int, ) -> None: monitor.track_tp_end_offset(topic_partition, offset + 5) monitor.on_message_in(topic_partition, offset, event.message) monitor.on_stream_event_in(topic_partition, offset, stream, event) monitor.on_tp_commit({topic_partition: offset})