Function TestWorkContext

rivertest/rivertest_test.go:1274–1298

(t *testing.T)

}

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

	ctx := context.Background()

	type testBundle struct{}

	setup := func(ctx context.Context, t *testing.T) (context.Context, *testBundle) {
		t.Helper()

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

		return WorkContext(ctx, client), &testBundle{}
	}

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

		ctx, _ := setup(ctx, t)

		client := river.ClientFromContext[pgx.Tx](ctx)
		require.NotNil(t, client)
	})
}