Function ResumableStepAtCursor

rivertest/resumable.go:35–45

ResumableStepAtCursor configures insert options so that the job resumes at the named step with the provided cursor data. The named step and all subsequent steps will execute, with all steps before it skipped. Used with [Worker.Work], simulates a job that was partway through a [river.ResumableStepCu

(opts *river.InsertOpts, stepName string, cursor TCursor)

//	result, err := testWorker.Work(ctx, t, tx, args,
//	    rivertest.ResumableStepAtCursor(&river.InsertOpts{}, "process_ids", MyCursor{LastID: 42}))
func ResumableStepAtCursor[TCursor any](opts *river.InsertOpts, stepName string, cursor TCursor) *river.InsertOpts {
	cursorBytes, err := json.Marshal(cursor)
	if err != nil {
		panic(fmt.Sprintf("rivertest: marshal resumable cursor: %s", err))
	}

	mergeResumableMetadata(opts, stepName, map[string]json.RawMessage{
		stepName: json.RawMessage(cursorBytes),
	})
	return opts
}

func mergeResumableMetadata(opts *river.InsertOpts, step string, cursors map[string]json.RawMessage) {
	var existing map[string]any