lint fix
tests passing kz_handler - thanks @cyli
correct disconnecting comments
correct comment - thanks @cyli
partition acquired check - thanks @cyli

Checking partition.acquired immediately instead of expecting
process_partitioner to be called again since it will be acquired after
wait_for_acquire returns