Function TestJobListWithJobs

internal/dblist/db_list_test.go:75–350

(t *testing.T)

}

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

	ctx := context.Background()

	type testBundle struct {
		baselineTime time.Time // baseline time frozen at now when setup is called
		driver       riverdriver.Driver[pgx.Tx]
		exec         riverdriver.Executor
		jobs         []*rivertype.JobRow
	}

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

		var (
			driver = riverpgxv5.New(nil)
			tx     = riverdbtest.TestTxPgx(ctx, t)
			exec   = driver.UnwrapExecutor(tx)
		)

		job1 := testfactory.Job(ctx, t, exec, &testfactory.JobOpts{Queue: ptrutil.Ptr("priority"), Priority: ptrutil.Ptr(1)})
		job2 := testfactory.Job(ctx, t, exec, &testfactory.JobOpts{EncodedArgs: []byte(`{"job_num": 2}`), Priority: ptrutil.Ptr(2)})
		job3 := testfactory.Job(ctx, t, exec, &testfactory.JobOpts{Metadata: []byte(`{"some_key": "some_value"}`), Priority: ptrutil.Ptr(3)})
		job4 := testfactory.Job(ctx, t, exec, &testfactory.JobOpts{State: ptrutil.Ptr(rivertype.JobStateRunning), Priority: ptrutil.Ptr(1)})
		job5 := testfactory.Job(ctx, t, exec, &testfactory.JobOpts{Kind: ptrutil.Ptr("alternate_kind"), Priority: ptrutil.Ptr(2)})

		return &testBundle{
			baselineTime: time.Now(),
			driver:       driver,
			exec:         exec,
			jobs:         []*rivertype.JobRow{job1, job2, job3, job4, job5},
		}
	}

	type testListFunc func(jobs []*rivertype.JobRow, err error)

	execTest := func(ctx context.Context, t *testing.T, bundle *testBundle, params *JobListParams, testFunc testListFunc) {
		t.Helper()

		listParams, err := JobMakeDriverParams(ctx, params, bundle.driver.SQLFragmentColumnIn)
		require.NoError(t, err)

		t.Logf("testing JobList in Executor")
		{
			jobs, err := bundle.exec.JobList(ctx, listParams)
			require.NoError(t, err)
			testFunc(jobs, err)
		}

		t.Logf("testing JobListTx")
		{
			// use a sub-transaction in case it's rolled back or errors:
			execTx, err := bundle.exec.Begin(ctx)
			require.NoError(t, err)
			defer execTx.Rollback(ctx)

			jobs, err := execTx.JobList(ctx, listParams)