fixing _get_leader_for_partition unittests