Function requireManyInsertedErr

rivertest/rivertest.go:404–459

(ctx context.Context, t testingT, exec riverdriver.Executor, expectedJobs []ExpectedJob)

}

func requireManyInsertedErr[TDriver riverdriver.Driver[TTx], TTx any](ctx context.Context, t testingT, exec riverdriver.Executor, expectedJobs []ExpectedJob) ([]*rivertype.JobRow, error) {
	t.Helper()

	expectedArgsKinds := sliceutil.Map(expectedJobs, func(j ExpectedJob) string { return j.Args.Kind() })

	var schema string
	if len(expectedJobs) > 0 && expectedJobs[0].Opts != nil {
		schema = expectedJobs[0].Opts.Schema
	}

	// For simplicity (and because I can't think of any reason anyone would need
	// to do otherwise), require that if an explicit schema is being set that
	// it's the same explicit schema for all options. Callers may specify the
	// schema only once in the first expectation's options, or specify the same
	// schema for all expectations' options, but they're not allowed to set a
	// schema after the first expectation's options if it wasn't set in the
	// first, and not allowed to mix and match schemas between options.
	for i, expectedJob := range expectedJobs {
		if opts := expectedJob.Opts; opts != nil {
			if schema == "" && opts.Schema != "" ||
				schema != "" && opts.Schema != "" && schema != opts.Schema {
				return nil, fmt.Errorf(
					"when setting RequireInsertedOpts.Schema with RequireMany schema should be set only at index 0 or the same schema set for all options; "+
						"expectedJobs[0].Opts.Schema = %q, expectedJobs[%d].Opts.Schema = %q",
					schema, i, opts.Schema)
			}
		}
	}

	// Returned ordered by ID.
	jobRows, err := exec.JobGetByKindMany(ctx, &riverdriver.JobGetByKindManyParams{
		Kind:   expectedArgsKinds,
		Schema: schema,
	})
	if err != nil {
		return nil, fmt.Errorf("error querying jobs: %w", err)
	}

	actualArgsKinds := sliceutil.Map(jobRows, func(j *rivertype.JobRow) string { return j.Kind })

	if !slices.Equal(expectedArgsKinds, actualArgsKinds) {
		failuref(t, "Inserted jobs didn't match expectation; expected: %+v, actual: %+v",
			expectedArgsKinds, actualArgsKinds)
		return nil, nil
	}

	for i, jobRow := range jobRows {
		if expectedJobs[i].Opts != nil {
			if !compareJobToInsertOpts(t, jobRow, expectedJobs[i].Opts, i, false) {
				return nil, nil
			}
		}
	}

	return jobRows, nil
}

const rfc3339Micro = "2006-01-02T15:04:05.999999Z07:00"