🐛 Fix test race condition

clevyr · May 10, 2022 · 636920d · 636920d
1 parent 8fee630
commit 636920d
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/cmd/cmd_test.go b/cmd/cmd_test.go
@@ -161,8 +161,10 @@ func Test_openAndTemplate(t *testing.T) {
 			}
 
 			var stdoutBuf strings.Builder
+			ch := make(chan struct{})
 			go func() {
 				_, _ = io.Copy(&stdoutBuf, r)
+				ch <- struct{}{}
 			}()
 			defer func(w *os.File) {
 				_ = w.Close()
@@ -196,6 +198,9 @@ func Test_openAndTemplate(t *testing.T) {
 				return
 			}
 
+			_ = w.Close()
+			<-ch
+
 			if (stdoutBuf.Len() != 0) != tt.wantStdout {
 				t.Errorf("openAndTemplate() got stdout len = %v, want stdout %v", stdoutBuf.Len(), tt.wantStdout)
 				return