Sri Lanka declared at 602-5 in its first innings and left New Zealand reeling at 22-2 by stumps