Function setupCLITaskTestWithSnapshot

cli/task_test.go:367–400

setupCLITaskTestWithSnapshot creates a task in the specified status with a log snapshot. Note: We do not use IncludeProvisionerDaemon because these tests use dbfake to directly set up database state and don't need actual provisioning. This also avoids potential interference from the provisioner daem

(ctx context.Context, t *testing.T, status codersdk.TaskStatus, messages []agentapisdk.Message)

// set up database state and don't need actual provisioning. This also avoids potential
// interference from the provisioner daemon polling for jobs.
func setupCLITaskTestWithSnapshot(ctx context.Context, t *testing.T, status codersdk.TaskStatus, messages []agentapisdk.Message) (*codersdk.Client, codersdk.Task) {
	t.Helper()

	ownerClient, db := coderdtest.NewWithDatabase(t, nil)
	owner := coderdtest.CreateFirstUser(t, ownerClient)
	userClient, user := coderdtest.CreateAnotherUser(t, ownerClient, owner.OrganizationID)

	ownerUser, err := ownerClient.User(ctx, owner.UserID.String())
	require.NoError(t, err)
	ownerSubject := coderdtest.AuthzUserSubject(ownerUser)

	task := createTaskInStatus(t, db, owner.OrganizationID, user.ID, status)

	// Create snapshot envelope with agentapi format.
	envelope := coderd.TaskLogSnapshotEnvelope{
		Format: "agentapi",
		Data: agentapisdk.GetMessagesResponse{
			Messages: messages,
		},
	}
	snapshotJSON, err := json.Marshal(envelope)
	require.NoError(t, err)

	// Insert snapshot into database.
	snapshotTime := time.Now()
	err = db.UpsertTaskSnapshot(dbauthz.As(ctx, ownerSubject), database.UpsertTaskSnapshotParams{
		TaskID:               task.ID,
		LogSnapshot:          json.RawMessage(snapshotJSON),
		LogSnapshotCreatedAt: snapshotTime,
	})
	require.NoError(t, err)

	return userClient, task
}

// setupCLITaskTestWithoutSnapshot creates a task in the specified status without a log snapshot.
// Note: We do not use IncludeProvisionerDaemon because these tests use dbfake to directly