Function goEventuallyStartFakeAgent

scaletest/createworkspaces/run_test.go:537–573

Since the runner creates the workspace on it's own, we have to keep listing workspaces until we find it, then wait for the build to finish, then start the agents. It is the caller's responsibility to call the returned function to stop the agents.

(ctx context.Context, t *testing.T, client *codersdk.Client, agentToken string)

// finish, then start the agents. It is the caller's responsibility to
// call the returned function to stop the agents.
func goEventuallyStartFakeAgent(ctx context.Context, t *testing.T, client *codersdk.Client, agentToken string) chan io.Closer {
	t.Helper()
	ch := make(chan io.Closer, 1) // Don't block.
	go func() {
		defer close(ch)
		var workspace codersdk.Workspace
		if !assert.Eventually(t, func() bool {
			res, err := client.Workspaces(ctx, codersdk.WorkspaceFilter{})
			if err != nil {
				return false
			}
			if len(res.Workspaces) == 1 {
				workspace = res.Workspaces[0]
				return true
			}
			return false
		}, testutil.WaitShort, testutil.IntervalMedium) {
			return
		}

		coderdtest.AwaitWorkspaceBuildJobCompleted(t, client, workspace.LatestBuild.ID)

		agentClient := agentsdk.New(client.URL, agentsdk.WithFixedToken(agentToken))
		agentCloser := agent.New(agent.Options{
			Client: agentClient,
			Logger: slogtest.Make(t, &slogtest.Options{IgnoreErrors: true}).
				Named("agent").Leveled(slog.LevelWarn),
		})
		resources := coderdtest.AwaitWorkspaceAgents(t, client, workspace.ID)
		assert.GreaterOrEqual(t, len(resources), 1, "workspace %s has no resources", workspace.ID.String())
		assert.NotEmpty(t, resources[0].Agents, "workspace %s has no agents", workspace.ID.String())
		agentID := resources[0].Agents[0].ID
		t.Logf("agent %s is running for workspace %s", agentID.String(), workspace.ID.String())
		ch <- agentCloser
	}()
	return ch
}