Function exerciseJobInsert

riverdriver/riverdrivertest/job_insert.go:27–946

(ctx context.Context, t *testing.T,
	driverWithSchema func(ctx context.Context, t *testing.T, opts *riverdbtest.TestSchemaOpts) (riverdriver.Driver[TTx], string),
	executorWithTx func(ctx context.Context, t *testing.T) (riverdriver.Executor, riverdriver.Driver[TTx]),
)

)

func exerciseJobInsert[TTx any](ctx context.Context, t *testing.T,
	driverWithSchema func(ctx context.Context, t *testing.T, opts *riverdbtest.TestSchemaOpts) (riverdriver.Driver[TTx], string),
	executorWithTx func(ctx context.Context, t *testing.T) (riverdriver.Executor, riverdriver.Driver[TTx]),
) {
	t.Helper()

	type testBundle struct {
		driver riverdriver.Driver[TTx]
	}

	setup := func(ctx context.Context, t *testing.T) (riverdriver.Executor, *testBundle) {
		t.Helper()

		exec, driver := executorWithTx(ctx, t)

		return exec, &testBundle{
			driver: driver,
		}
	}

	t.Run("JobInsertFastMany", func(t *testing.T) {
		t.Parallel()

		t.Run("AllArgs", func(t *testing.T) {
			t.Parallel()

			exec, bundle := setup(ctx, t)

			var (
				idStart = rand.Int64()
				now     = time.Now().UTC()
			)

			insertParams := make([]*riverdriver.JobInsertFastParams, 10)
			for i := range insertParams {
				insertParams[i] = &riverdriver.JobInsertFastParams{
					ID:           ptrutil.Ptr(idStart + int64(i)),
					CreatedAt:    ptrutil.Ptr(now.Add(time.Duration(i) * 5 * time.Second)),
					EncodedArgs:  []byte(`{"encoded": "args"}`),
					Kind:         "test_kind",
					MaxAttempts:  rivercommon.MaxAttemptsDefault,
					Metadata:     []byte(`{"meta": "data"}`),
					Priority:     rivercommon.PriorityDefault,
					Queue:        rivercommon.QueueDefault,
					ScheduledAt:  ptrutil.Ptr(now.Add(time.Duration(i) * time.Minute)),
					State:        rivertype.JobStateAvailable,
					Tags:         []string{"tag"},
					UniqueKey:    []byte("unique-key-fast-many-" + strconv.Itoa(i)),
					UniqueStates: 0xff,
				}
			}

			resultRows, err := exec.JobInsertFastMany(ctx, &riverdriver.JobInsertFastManyParams{
				Jobs: insertParams,
			})
			require.NoError(t, err)
			require.Len(t, resultRows, len(insertParams))