refactor: remove PrebuildsOrchestrator context from claim tests

evgeniy-scherbina · evgeniy-scherbina · commit 9f66169e5061 · 2025-04-23T19:26:35.000Z
diff --git a/enterprise/coderd/prebuilds/claim_test.go b/enterprise/coderd/prebuilds/claim_test.go
@@ -16,7 +16,6 @@ import (
 
 	"github.com/coder/coder/v2/coderd/coderdtest"
 	"github.com/coder/coder/v2/coderd/database"
-	"github.com/coder/coder/v2/coderd/database/dbauthz"
 	"github.com/coder/coder/v2/coderd/database/dbtestutil"
 	agplprebuilds "github.com/coder/coder/v2/coderd/prebuilds"
 	"github.com/coder/coder/v2/coderd/rbac"
@@ -155,8 +154,6 @@ func TestClaimPrebuild(t *testing.T) {
 
 			userClient, user := coderdtest.CreateAnotherUser(t, client, owner.OrganizationID, rbac.RoleMember())
 
-			ctx = dbauthz.AsPrebuildsOrchestrator(ctx)
-
 			// Given: the reconciliation state is snapshot.
 			state, err := reconciler.SnapshotState(ctx, spy)
 			require.NoError(t, err)
@@ -429,8 +426,6 @@ func TestClaimPrebuild_CheckDifferentErrors(t *testing.T) {
 
 			userClient, user := coderdtest.CreateAnotherUser(t, client, owner.OrganizationID, rbac.RoleMember())
 
-			ctx = dbauthz.AsPrebuildsOrchestrator(ctx)
-
 			// Given: the reconciliation state is snapshot.
 			state, err := reconciler.SnapshotState(ctx, errorStore)
 			require.NoError(t, err)
