KAFKA-20783: Completion handling and more tests (3/N) (#22804)