fix: create contexts per sub-test to fix flake (#15314)

ethanndickson · web-flow · commit 5f60a8d9e35b · 2024-11-01T23:22:36.000+11:00
Flake seen here: https://github.com/coder/coder/actions/runs/11624011543/job/32371950701 https://coder.com/blog/go-testing-contexts-and-t-parallel
diff --git a/coderd/database/querier_test.go b/coderd/database/querier_test.go
@@ -620,7 +620,6 @@ func TestGetAuthorizedWorkspacesAndAgentsByOwnerID(t *testing.T) {
 		t.SkipNow()
 	}
 
-	ctx := testutil.Context(t, testutil.WaitLong)
 	sqlDB := testSQLDB(t)
 	err := migrations.Up(sqlDB)
 	require.NoError(t, err)
@@ -695,6 +694,7 @@ func TestGetAuthorizedWorkspacesAndAgentsByOwnerID(t *testing.T) {
 	}
 	t.Run("sqlQuerier", func(t *testing.T) {
 		t.Parallel()
+		ctx := testutil.Context(t, testutil.WaitMedium)
 
 		userSubject, _, err := httpmw.UserRBACSubject(ctx, db, user.ID, rbac.ExpandableScope(rbac.ScopeAll))
 		require.NoError(t, err)
@@ -717,6 +717,7 @@ func TestGetAuthorizedWorkspacesAndAgentsByOwnerID(t *testing.T) {
 
 	t.Run("dbauthz", func(t *testing.T) {
 		t.Parallel()
+		ctx := testutil.Context(t, testutil.WaitMedium)
 
 		authzdb := dbauthz.New(db, authorizer, slogtest.Make(t, &slogtest.Options{}), coderdtest.AccessControlStorePointer())
 
