def test_partition_offset_counters(self): res = mock.Mock() res.offset = 400 op = OwnedPartition(None) op.set_offset(res.offset) self.assertEqual(op.last_offset_consumed, res.offset) self.assertEqual(op.next_offset, res.offset + 1)