another attempt to fix window stest

mafredri · mafredri · commit b9895ac5d7e4 · 2024-02-19T15:31:43.000+02:00
diff --git a/agent/agent_test.go b/agent/agent_test.go
@@ -324,7 +324,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 +343,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 +353,15 @@ 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)
+			require.NoError(t, s.Err())
 		})
 	}
 }
