def test_track_tp_end_offsets( self, monitor: PrometheusMonitor, metrics: FaustMetrics ) -> None: monitor.track_tp_end_offset(TP("foo", 0), 4004) self.assert_has_sample_value( metrics.topic_partition_end_offset, "test_topic_partition_end_offset", {"topic": "foo", "partition": "0"}, 4004, )
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})