Sri Lanka two wickets away from winning New Zealand Test