Function TestRequireInsertedTx

rivertest/rivertest_test.go:87–387

(t *testing.T)

}

func TestRequireInsertedTx(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)

		job := requireInsertedTx[*riverpgxv5.Driver](ctx, t, bundle.tx, &Job1Args{}, nil)
		require.False(t, bundle.mockT.Failed)
		require.Equal(t, "foo", job.Args.String)
	})

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

		riverClient, bundle := setup(t)

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

		_, err = riverClient.InsertTx(ctx, bundle.tx, Job2Args{Int: 123}, nil)
		require.NoError(t, err)

		job1 := requireInsertedTx[*riverpgxv5.Driver](ctx, t, bundle.tx, &Job1Args{}, nil)
		require.False(t, bundle.mockT.Failed)
		require.Equal(t, "foo", job1.Args.String)

		job2 := requireInsertedTx[*riverpgxv5.Driver](ctx, t, bundle.tx, &Job2Args{}, nil)
		require.False(t, bundle.mockT.Failed)
		require.Equal(t, 123, job2.Args.Int)
	})

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