'Start a new context:
-  Record the initial offsets for rollback
-  Reset the high-water mark'

def __enter__(self):

    self.initial_offsets = dict(self.consumer.offsets)
    self.high_water_mark = dict()
    self.logger.debug('Starting    context    at:    %s', self.initial_offsets)
    return self
