Improve tests for walltime handler.