Function TestRequireInserted

rivertest/rivertest_test.go:37–85

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 TestRequireInserted(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)

		job := requireInserted(ctx, t, bundle.driver, &Job1Args{}, bundle.schemaOpts)
		require.False(t, bundle.mockT.Failed)
		require.Equal(t, "foo", job.Args.String)
	})
}

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