From b887f4f2e496ba15c8d6c53a987171191313c427 Mon Sep 17 00:00:00 2001
From: Mathias Fredriksson <mafredri@gmail.com>
Date: Tue, 27 Feb 2024 19:40:18 +0200
Subject: [PATCH] test(agent/agentscripts): fix test flake in `TestEnv`

---
 agent/agentscripts/agentscripts_test.go | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/agent/agentscripts/agentscripts_test.go b/agent/agentscripts/agentscripts_test.go
index b9c8ae9f04c19..2be7e76c54f6a 100644
--- a/agent/agentscripts/agentscripts_test.go
+++ b/agent/agentscripts/agentscripts_test.go
@@ -69,12 +69,18 @@ func TestEnv(t *testing.T) {
 
 	ctx := testutil.Context(t, testutil.WaitLong)
 
-	testutil.Go(t, func() {
+	done := testutil.Go(t, func() {
 		err := runner.Execute(ctx, func(script codersdk.WorkspaceAgentScript) bool {
 			return true
 		})
 		assert.NoError(t, err)
 	})
+	defer func() {
+		select {
+		case <-ctx.Done():
+		case <-done:
+		}
+	}()
 
 	var log []agentsdk.Log
 	for {