diff --git a/exit_test.go b/exit_test.go index d0123d971..e295eb8e8 100644 --- a/exit_test.go +++ b/exit_test.go @@ -24,10 +24,13 @@ import ( func ExampleFlushAndExit() { // Set up klog so that we can test it below. + var fs flag.FlagSet klog.InitFlags(&fs) fs.Set("skip_headers", "true") defer flag.Set("skip_headers", "false") + fs.Set("logtostderr", "false") + defer fs.Set("logtostderr", "true") klog.SetOutput(os.Stdout) defer klog.SetOutput(nil) klog.OsExit = func(exitCode int) {