Function TestRequireManyInserted

rivertest/rivertest_test.go:750–864

The tests for this function are quite minimal because it uses the same implementation as the `*Tx` variant, so most of the test happens below.

(t *testing.T)

// The tests for this function are quite minimal because it uses the same
// implementation as the `*Tx` variant, so most of the test happens below.
func TestRequireManyInserted(t *testing.T) {
	t.Parallel()

	ctx := context.Background()

	type testBundle struct {
		dbPool     *pgxpool.Pool
		driver     *riverpgxv5.Driver
		mockT      *testutil.MockT
		schema     string
		schemaOpts *RequireInsertedOpts
	}

	setup := func(t *testing.T) (*river.Client[pgx.Tx], *testBundle) {
		t.Helper()

		var (
			dbPool = riversharedtest.DBPool(ctx, t)
			driver = riverpgxv5.New(dbPool)
			schema = riverdbtest.TestSchema(ctx, t, driver, nil)
		)

		riverClient, err := river.NewClient(driver, &river.Config{
			Schema: schema,
		})
		require.NoError(t, err)

		return riverClient, &testBundle{
			dbPool:     dbPool,
			driver:     driver,
			mockT:      testutil.NewMockT(t),
			schema:     schema,
			schemaOpts: &RequireInsertedOpts{Schema: schema},
		}
	}

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

		riverClient, bundle := setup(t)

		_, err := riverClient.Insert(ctx, Job1Args{String: "foo"}, nil)
		require.NoError(t, err)

		jobs := requireManyInserted(ctx, bundle.mockT, bundle.driver, []ExpectedJob{
			{Args: &Job1Args{}, Opts: bundle.schemaOpts},
		})
		require.False(t, bundle.mockT.Failed)
		require.Equal(t, "job1", jobs[0].Kind)
	})

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

		riverClient, bundle := setup(t)

		_, err := riverClient.InsertMany(ctx, []river.InsertManyParams{
			{Args: Job1Args{String: "foo"}},