remove MustWaitForAnyProvisionerWithClient

cstyan · cstyan · commit 224acfcb95fb · 2025-08-12T21:46:36.000Z
Signed-off-by: Callum Styan &lt;callumstyan@gmail.com&gt;
diff --git a/coderd/autobuild/lifecycle_executor_test.go b/coderd/autobuild/lifecycle_executor_test.go
@@ -694,9 +694,9 @@ func TestExecuteAutostopSuspendedUser(t *testing.T) {
 	t.Parallel()
 
 	var (
-		tickCh  = make(chan time.Time)
-		statsCh = make(chan autobuild.Stats)
-		client  = coderdtest.New(t, &coderdtest.Options{
+		tickCh     = make(chan time.Time)
+		statsCh    = make(chan autobuild.Stats)
+		client, db = coderdtest.NewWithDatabase(t, &coderdtest.Options{
 			AutobuildTicker:          tickCh,
 			IncludeProvisionerDaemon: true,
 			AutobuildStats:           statsCh,
@@ -705,7 +705,7 @@ func TestExecuteAutostopSuspendedUser(t *testing.T) {
 
 	admin := coderdtest.CreateFirstUser(t, client)
 	// Wait for provisioner to be available
-	coderdtest.MustWaitForAnyProvisionerWithClient(t, client)
+	coderdtest.MustWaitForAnyProvisioner(t, db)
 	version := coderdtest.CreateTemplateVersion(t, client, admin.OrganizationID, nil)
 	coderdtest.AwaitTemplateVersionJobCompleted(t, client, version.ID)
 	template := coderdtest.CreateTemplate(t, client, admin.OrganizationID, version.ID)
diff --git a/coderd/coderdtest/coderdtest.go b/coderd/coderdtest/coderdtest.go
@@ -1652,15 +1652,6 @@ func MustWaitForAnyProvisioner(t *testing.T, db database.Store) {
 	}, testutil.WaitShort, testutil.IntervalFast)
 }
 
-func MustWaitForAnyProvisionerWithClient(t *testing.T, client *codersdk.Client) {
-	t.Helper()
-	ctx := ctxWithProvisionerPermissions(testutil.Context(t, testutil.WaitShort))
-	require.Eventually(t, func() bool {
-		daemons, err := client.ProvisionerDaemons(ctx)
-		return err == nil && len(daemons) > 0
-	}, testutil.WaitShort, testutil.IntervalFast)
-}
-
 // MustWaitForProvisionersAvailable waits for provisioners to be available for a specific workspace.
 func MustWaitForProvisionersAvailable(t *testing.T, db database.Store, workspace codersdk.Workspace) uuid.UUID {
 	t.Helper()
