Function TestRequireManyInsertedTx

rivertest/rivertest_test.go:866–1272

(t *testing.T)

}

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

	ctx := context.Background()

	type testBundle struct {
		mockT *testutil.MockT
		tx    pgx.Tx
	}

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

		riverClient, err := river.NewClient(riverpgxv5.New(nil), &river.Config{})
		require.NoError(t, err)

		return riverClient, &testBundle{
			mockT: testutil.NewMockT(t),
			tx:    riverdbtest.TestTxPgx(ctx, t),
		}
	}

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

		riverClient, bundle := setup(t)

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

		jobs := requireManyInsertedTx[*riverpgxv5.Driver](ctx, bundle.mockT, bundle.tx, []ExpectedJob{
			{Args: &Job1Args{}},
		})
		require.False(t, bundle.mockT.Failed)
		require.Equal(t, "job1", jobs[0].Kind)
	})

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

		riverClient, bundle := setup(t)

		// Start a second transaction with different visibility.
		otherTx := riverdbtest.TestTxPgx(ctx, t)

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

		// Visible in the original transaction.
		jobs := requireManyInsertedTx[*riverpgxv5.Driver](ctx, bundle.mockT, bundle.tx, []ExpectedJob{
			{Args: &Job1Args{}},
		})
		require.False(t, bundle.mockT.Failed)
		require.Equal(t, "job1", jobs[0].Kind)

		// Not visible in the second transaction.
		_ = requireManyInsertedTx[*riverpgxv5.Driver](ctx, bundle.mockT, otherTx, []ExpectedJob{
			{Args: &Job1Args{}},