another attempt to fix window stest

mafredri · mafredri · commit ffae4ad3a56f · 2024-02-19T15:45:00.000+02:00
diff --git a/agent/agent_test.go b/agent/agent_test.go
@@ -5,6 +5,7 @@ import (
 	"bytes"
 	"context"
 	"encoding/json"
+	"errors"
 	"fmt"
 	"io"
 	"math/rand"
@@ -324,7 +325,12 @@ func TestAgent_Session_EnvironmentVariables(t *testing.T) {
 	err = session.Start(command)
 	require.NoError(t, err)
 
+	// Context is fine here since we're not doing a parallel subtest.
 	ctx := testutil.Context(t, testutil.WaitLong)
+	go func() {
+		<-ctx.Done()
+		_ = session.Close()
+	}()
 
 	s := bufio.NewScanner(stdout)
 	out := make(chan string)
@@ -338,8 +344,6 @@ func TestAgent_Session_EnvironmentVariables(t *testing.T) {
 		}
 	})
 
-	// Until we have gotten the first result, the shell may spit out some data.
-	first := true
 	//nolint:paralleltest // These tests need to run sequentially.
 	for k, partialV := range map[string]string{
 		"CODER":               "true",  // From the agent.
@@ -350,21 +354,17 @@ func TestAgent_Session_EnvironmentVariables(t *testing.T) {
 	} {
 		t.Run(k, func(t *testing.T) {
 			echoEnv(t, stdin, k)
-			if first {
-				for {
-					s := testutil.RequireRecvCtx(ctx, t, out)
-					t.Logf("%s=%s", k, s)
-					s = strings.TrimSpace(s)
-					if strings.Contains(s, partialV) {
-						first = false
-						return
-					}
+			// Windows is unreliable, so keep scanning until we find a match.
+			for s.Scan() {
+				got := strings.TrimSpace(s.Text())
+				t.Logf("%s=%s", k, got)
+				if strings.Contains(got, partialV) {
+					break
 				}
 			}
-			s := testutil.RequireRecvCtx(ctx, t, out)
-			t.Logf("%s=%s", k, s)
-			s = strings.TrimSpace(s)
-			require.Contains(t, s, partialV)
+			if err := s.Err(); !errors.Is(err, io.EOF) {
+				require.NoError(t, err)
+			}
 		})
 	}
 }
