minor fixup, tests

Author: phryneas

src/__tests__/renderToRenderStream.test.tsx

@@ -58,6 +58,37 @@ describe('snapshotDOM', () => {
       expect(input.value).toBe('2')
     }
   })
+
+  test('queries option', async () => {
+    function Component() {
+      return null
+    }
+    const queries = {
+      foo: (_: any) => {
+        return null
+      },
+    }
+    const {takeRender, renderResultPromise} = renderToRenderStream(
+      <Component />,
+      {
+        queries,
+        snapshotDOM: true,
+      },
+    )
+    const utils = await renderResultPromise
+    expect(utils.foo()).toBe(null)
+    const {withinDOM} = await takeRender()
+    expect(withinDOM().foo()).toBe(null)
+    function _typeTest() {
+      // @ts-expect-error should not be present
+      utils.getByText
+      // @ts-expect-error should not be present
+      withinDOM().getByText
+      utils.debug()
+      withinDOM().debug()
+      const _str: string = withinDOM().logTestingPlaygroundURL()
+    }
+  })
 })
 
 // for more tests, see the `createRenderStream` test suite, as `renderToRenderStream` is just a wrapper around that

src/renderStream/__tests__/createRenderStream.test.tsx

@@ -87,6 +87,32 @@ describe('snapshotDOM', () => {
       }
     }
   })
+
+  test('queries option', async () => {
+    function Component() {
+      return null
+    }
+    const queries = {
+      foo: (_: any) => {
+        return null
+      },
+    }
+
+    const {takeRender, render} = createRenderStream({
+      snapshotDOM: true,
+      queries,
+    })
+    render(<Component />)
+
+    const {withinDOM} = await takeRender()
+    expect(withinDOM().foo()).toBe(null)
+    function _typeTest() {
+      // @ts-expect-error should not be present
+      withinDOM().getByText
+      withinDOM().debug()
+      const _str: string = withinDOM().logTestingPlaygroundURL()
+    }
+  })
 })
 
 describe('replaceSnapshot', () => {

src/renderStream/syncQueries.ts

@@ -10,7 +10,7 @@ export type SyncQueries = {
     : K]: OriginalQueries[K]
 }
 
-export const syncQueries = Object.values(
+export const syncQueries = Object.fromEntries(
   Object.entries(queries).filter(
     ([key]) => key.startsWith('get') || key.startsWith('find'),
   ),

src/renderToRenderStream.ts

@@ -14,7 +14,7 @@ import {SyncQueries} from './renderStream/syncQueries.js'
 type RenderOptions<
   Snapshot extends ValidSnapshot = void,
   Q extends Queries = SyncQueries,
-> = BaseOptions<Q> & RenderStreamOptions<Snapshot>
+> = BaseOptions<Q> & RenderStreamOptions<Snapshot, Q>
 
 export interface RenderStreamWithRenderResult<
   Snapshot extends ValidSnapshot = void,