only close for test

terrastruct · Mar 3, 2023 · 56bcfcd · 56bcfcd
1 parent f04d4d9
commit 56bcfcd
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/xmain/xmain.go b/xmain/xmain.go
@@ -94,9 +94,6 @@ type State struct {
 }
 
 func (ms *State) Main(ctx context.Context, sigs <-chan os.Signal, run func(context.Context, *State) error) error {
-	defer ms.Stdout.Close()
-	defer ms.Stderr.Close()
-
 	ctx, cancel := context.WithCancel(ctx)
 	defer cancel()
 

diff --git a/xmain/xmaintest.go b/xmain/xmaintest.go
@@ -122,6 +122,8 @@ func (ts *TestState) Start(tb testing.TB, ctx context.Context) {
 	go func() {
 		var err error
 		defer func() {
+			ts.ms.Stdout.Close()
+			ts.ms.Stderr.Close()
 			pipeWG.Wait()
 			if tempDirCleanup != nil {
 				tempDirCleanup()