Update test code.

googollee · Jul 24, 2024 · 9808f16 · 9808f16
1 parent 19abcf9
commit 9808f16
Showing 1 changed file with 10 additions and 7 deletions.
diff --git a/log/log_test.go b/log/log_test.go
@@ -20,18 +20,21 @@ func captureStderr(t *testing.T, f func(t *testing.T)) string {
 	if err != nil {
 		t.Fatal("create pipe error:", err)
 	}
+	defer func() {
+		r.Close()
+	}()
 
 	orig := os.Stderr
 	os.Stderr = w
 
-	defer func() {
-		os.Stderr = orig
-		w.Close()
-	}()
-
-	f(t)
+	func() {
+		defer func() {
+			os.Stderr = orig
+			w.Close()
+		}()
 
-	w.Close()
+		f(t)
+	}()
 
 	out, err := io.ReadAll(r)
 	if err != nil {