test(cli/exptest): fix context in TestScaleTestWorkspaceTraffic_UseHostLogin (#16171)

mafredri · web-flow · commit de874442f878 · 2025-01-17T14:51:24.000+02:00
diff --git a/cli/exptest/exptest_scaletest_test.go b/cli/exptest/exptest_scaletest_test.go
@@ -20,9 +20,6 @@ import (
 // can influence other tests in the same package.
 // nolint:paralleltest
 func TestScaleTestWorkspaceTraffic_UseHostLogin(t *testing.T) {
-	ctx, cancelFunc := context.WithTimeout(context.Background(), testutil.WaitMedium)
-	defer cancelFunc()
-
 	log := slogtest.Make(t, &slogtest.Options{IgnoreErrors: true})
 	client := coderdtest.New(t, &coderdtest.Options{
 		Logger:                   &log,
@@ -41,6 +38,9 @@ func TestScaleTestWorkspaceTraffic_UseHostLogin(t *testing.T) {
 		cwr.Name = "scaletest-workspace"
 	})
 
+	ctx, cancel := context.WithTimeout(context.Background(), testutil.WaitLong)
+	defer cancel()
+
 	// Test without --use-host-login first.g
 	inv, root := clitest.New(t, "exp", "scaletest", "workspace-traffic",
 		"--template", tpl.Name,
