NZ close in on Test win as Stokes makes retirement call