packages/react-query/test/ssg.test.ts - main - githubmirror/trpc

packages/react-query/test/ssg.test.ts
80 строк2 KB
Julius Marminge
refactor: move react tests (#6542)
24 фев 2025, 16:59
24 фев 2025, 16:594df1eb8
import type { InfiniteData } from '@tanstack/react-query';
import { QueryClient } from '@tanstack/react-query';
import { createServerSideHelpers } from '@trpc/react-query/server';
import { initTRPC } from '@trpc/server';
import { z } from 'zod';

const t = initTRPC.create();

const appRouter = t.router({
  post: t.router({
    byId: t.procedure
      .input(
        z.object({
          id: z.string(),
        }),
      )
      .query(() => '__result' as const),
    list: t.procedure
      .input(
        z.object({
          cursor: z.string().optional(),
        }),
      )
      .query(() => '__infResult' as const),
  }),
});

test('fetch', async () => {
  const queryClient = new QueryClient();
  const ssg = createServerSideHelpers({
    router: appRouter,
    ctx: {},
    queryClient,
  });

  const post1 = await queryClient.fetchQuery(
    ssg.post.byId.queryOptions({ id: '1' }),
  );

  const post2 = await ssg.post.byId.fetch({ id: '1' });
  expectTypeOf<'__result'>(post2);
  expectTypeOf(post1).toEqualTypeOf(post2);
});

test('fetchInfinite', async () => {
  const ssg = createServerSideHelpers({ router: appRouter, ctx: {} });

  const post = await ssg.post.list.fetchInfinite({});
  expectTypeOf<InfiniteData<'__infResult', string | null>>(post);

  expect(post.pages).toStrictEqual(['__infResult']);
});

test('prefetch and dehydrate', async () => {
  const ssg = createServerSideHelpers({ router: appRouter, ctx: {} });
  await ssg.post.byId.prefetch({ id: '1' });

  const data = JSON.stringify(ssg.dehydrate());
  expect(data).toContain('__result');
});

test('prefetchInfinite and dehydrate', async () => {
  const ssg = createServerSideHelpers({ router: appRouter, ctx: {} });
  await ssg.post.list.prefetchInfinite({});

  const data = JSON.stringify(ssg.dehydrate());
  expect(data).toContain('__infResult');
});

test('using queryOptions', async () => {
  const ssg = createServerSideHelpers({ router: appRouter, ctx: {} });

  const opts = ssg.post.byId.queryOptions({ id: '1' });
  //    ^?
  const post = await ssg.queryClient.fetchQuery(
    ssg.post.byId.queryOptions({ id: '1' }),
  );
  expectTypeOf<'__result'>(post);
  expect(post).toStrictEqual('__result');
});