Function testLiveQuery

packages/pglite-react/test/hooks.test.tsx:73–424

(queryHook: 'useLiveQuery' | 'useLiveIncrementalQuery')

})

function testLiveQuery(queryHook: 'useLiveQuery' | 'useLiveIncrementalQuery') {
  describe(queryHook, () => {
    let db: PGliteWithLive
    let wrapper: ({
      children,
    }: {
      children: React.ReactNode
    }) => React.ReactElement
    const hookFn =
      queryHook === 'useLiveQuery' ? useLiveQuery : useLiveIncrementalQuery
    const incKey = 'id'
    beforeEach(async () => {
      db = await PGlite.create({
        extensions: {
          live,
        },
      })
      wrapper = ({ children }) => {
        return <PGliteProvider db={db}>{children}</PGliteProvider>
      }

      await db.exec(`
        CREATE TABLE IF NOT EXISTS test (
          id SERIAL PRIMARY KEY,
          name TEXT
        );
      `)
      await db.exec(`TRUNCATE test;`)
    })

    it('can receive initial results', async () => {
      await db.exec(`INSERT INTO test (name) VALUES ('test1'),('test2');`)

      const { result } = renderHook(
        () => hookFn(`SELECT * FROM test`, [], incKey),
        { wrapper },
      )

      await waitFor(() => expect(result.current).not.toBe(undefined))
      expect(result.current).toEqual({
        rows: [
          {
            id: 1,
            name: 'test1',
          },
          {
            id: 2,
            name: 'test2',
          },
        ],
        fields: [
          {
            name: 'id',
            dataTypeID: 23,
          },
          {
            name: 'name',
            dataTypeID: 25,