Function TestResumableOptsIntegration

rivertest/resumable_test.go:80–201

(t *testing.T)

}

func TestResumableOptsIntegration(t *testing.T) {
	t.Parallel()

	ctx := context.Background()

	type testBundle struct {
		driver *riverpgxv5.Driver
		tx     pgx.Tx
	}

	setup := func(t *testing.T) *testBundle {
		t.Helper()

		return &testBundle{
			driver: riverpgxv5.New(nil),
			tx:     riverdbtest.TestTxPgx(ctx, t),
		}
	}

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

		bundle := setup(t)

		var ran []string
		worker := river.WorkFunc(func(ctx context.Context, job *river.Job[resumableTestArgs]) error {
			river.ResumableStep(ctx, "step1", nil, func(ctx context.Context) error {
				ran = append(ran, "step1")
				return nil
			})
			river.ResumableStep(ctx, "step2", nil, func(ctx context.Context) error {
				ran = append(ran, "step2")
				return nil
			})
			river.ResumableStep(ctx, "step3", nil, func(ctx context.Context) error {
				ran = append(ran, "step3")
				return nil
			})
			return nil
		})

		config := &river.Config{ID: "rivertest-resumable"}
		tw := NewWorker(t, bundle.driver, config, worker)

		opts := &river.InsertOpts{}
		ResumableStepAfter(opts, "step1")

		result, err := tw.Work(ctx, t, bundle.tx, resumableTestArgs{}, opts)
		require.NoError(t, err)
		require.Equal(t, river.EventKindJobCompleted, result.EventKind)
		require.Equal(t, []string{"step2", "step3"}, ran)
	})

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

		bundle := setup(t)