Function NewWorker

rivertest/worker.go:66–82

NewWorker creates a new test Worker for testing the provided [river.Worker]. The worker uses the provided driver and River config to populate default values on test jobs and to configure the execution environment. A pool-less driver is recommended for most usage, as individual job inserts and execu

(tb testing.TB, driver riverdriver.Driver[TTx], config *river.Config, worker river.Worker[T])

// The worker is configured to disable unique enforcement by default, as this
// would otherwise prevent conflicting jobs from being tested in parallel.
func NewWorker[T river.JobArgs, TTx any](tb testing.TB, driver riverdriver.Driver[TTx], config *river.Config, worker river.Worker[T]) *Worker[T, TTx] {
	tb.Helper()

	config = config.WithDefaults()
	config.Test.DisableUniqueEnforcement = true

	client, err := river.NewClient(driver, config)
	if err != nil {
		tb.Fatalf("failed to create client: %s", err)
	}

	return &Worker[T, TTx]{
		client: client,
		config: config,
		worker: worker,
	}
}

func (w *Worker[T, TTx]) insertJob(ctx context.Context, tb testing.TB, tx TTx, args T, opts *river.InsertOpts) (*rivertype.JobRow, error) {
	tb.Helper()