ci: improve 'tfail in goroutine' ruleguard rule

ethanndickson · ethanndickson · commit 7eba8dd760b4 · 2025-09-03T03:25:49.000Z
diff --git a/coderd/files/cache_test.go b/coderd/files/cache_test.go
@@ -114,7 +114,7 @@ func TestConcurrentFetch(t *testing.T) {
 			hold.Done()
 			hold.Wait()
 			_, err := cache.Acquire(ctx, dbM, fileID)
-			require.NoError(t, err)
+			assert.NoError(t, err)
 		}()
 	}
 
diff --git a/coderd/files_test.go b/coderd/files_test.go
@@ -9,6 +9,7 @@ import (
 	"testing"
 
 	"github.com/google/uuid"
+	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
 
 	"github.com/coder/coder/v2/archive"
@@ -88,7 +89,7 @@ func TestPostFiles(t *testing.T) {
 				data := make([]byte, 1024)
 				_, err := client.Upload(ctx, codersdk.ContentTypeTar, bytes.NewReader(data))
 				end.Done()
-				require.NoError(t, err)
+				assert.NoError(t, err)
 			}()
 		}
 		wg.Done()
diff --git a/enterprise/tailnet/pgcoord_test.go b/enterprise/tailnet/pgcoord_test.go
@@ -409,8 +409,8 @@ func TestPGCoordinatorSingle_SendsHeartbeats(t *testing.T) {
 		if len(heartbeats) < 2 {
 			return false
 		}
-		require.Greater(t, heartbeats[0].Sub(start), time.Duration(0))
-		require.Greater(t, heartbeats[1].Sub(start), time.Duration(0))
+		assert.Greater(t, heartbeats[0].Sub(start), time.Duration(0))
+		assert.Greater(t, heartbeats[1].Sub(start), time.Duration(0))
 		return assert.Greater(t, heartbeats[1].Sub(heartbeats[0]), tailnet.HeartbeatPeriod*3/4)
 	}, testutil.WaitMedium, testutil.IntervalMedium)
 }
diff --git a/scaletest/createworkspaces/run_test.go b/scaletest/createworkspaces/run_test.go
@@ -257,7 +257,7 @@ func Test_Runner(t *testing.T) {
 			err := runner.Run(runnerCtx, "1", logs)
 			logsStr := logs.String()
 			t.Log("Runner logs:\n\n" + logsStr)
-			require.ErrorIs(t, err, context.Canceled)
+			assert.ErrorIs(t, err, context.Canceled)
 			close(done)
 		}()
 
diff --git a/scripts/rules.go b/scripts/rules.go
@@ -182,32 +182,28 @@ func doNotCallTFailNowInsideGoroutine(m dsl.Matcher) {
 	m.Match(`
 	go func($*_){
 		$*_
-		$require.$_($*_)
+		require.$_($*_)
 		$*_
 	}($*_)`).
-		At(m["require"]).
-		Where(m["require"].Text == "require").
 		Report("Do not call functions that may call t.FailNow in a goroutine, as this can cause data races (see testing.go:834)")
 
 	// require.Eventually runs the function in a goroutine.
 	m.Match(`
 	require.Eventually(t, func() bool {
 		$*_
-		$require.$_($*_)
+		require.$_($*_)
 		$*_
 	}, $*_)`).
-		At(m["require"]).
-		Where(m["require"].Text == "require").
 		Report("Do not call functions that may call t.FailNow in a goroutine, as this can cause data races (see testing.go:834)")
 
 	m.Match(`
 	go func($*_){
 		$*_
-		$t.$fail($*_)
+		t.$fail($*_)
 		$*_
 	}($*_)`).
 		At(m["fail"]).
-		Where(m["t"].Type.Implements("testing.TB") && m["fail"].Text.Matches("^(FailNow|Fatal|Fatalf)$")).
+		Where(m["fail"].Text.Matches("^(FailNow|Fatal|Fatalf)$")).
 		Report("Do not call functions that may call t.FailNow in a goroutine, as this can cause data races (see testing.go:834)")
 }
 

Original file line number	Diff line number	Diff line change
`@@ -114,7 +114,7 @@ func TestConcurrentFetch(t *testing.T) {`
`114`	`114`	`hold.Done()`
`115`	`115`	`hold.Wait()`
`116`	`116`	`_, err := cache.Acquire(ctx, dbM, fileID)`
`117`		`- require.NoError(t, err)`
	`117`	`+ assert.NoError(t, err)`
`118`	`118`	`}()`
`119`	`119`	`}`
`120`	`120`
Original file line number	Diff line number	Diff line change
`@@ -409,8 +409,8 @@ func TestPGCoordinatorSingle_SendsHeartbeats(t *testing.T) {`
`409`	`409`	`if len(heartbeats) < 2 {`
`410`	`410`	`return false`
`411`	`411`	`}`
`412`		`- require.Greater(t, heartbeats[0].Sub(start), time.Duration(0))`
`413`		`- require.Greater(t, heartbeats[1].Sub(start), time.Duration(0))`
	`412`	`+ assert.Greater(t, heartbeats[0].Sub(start), time.Duration(0))`
	`413`	`+ assert.Greater(t, heartbeats[1].Sub(start), time.Duration(0))`
`414`	`414`	`return assert.Greater(t, heartbeats[1].Sub(heartbeats[0]), tailnet.HeartbeatPeriod*3/4)`
`415`	`415`	`}, testutil.WaitMedium, testutil.IntervalMedium)`
`416`	`416`	`}`