123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117
import { useState } from 'react'import { fireEvent, render, waitFor } from '@testing-library/react'import { expect, it, vi } from 'vitest'import { proxy, useSnapshot } from 'valtio'
it('regression: useSnapshot renders should not fail consistency check with extra render (nested useSnapshot)', async () => {  const obj = proxy({ childCount: 0, parentCount: 0 })
  const childRenderFn = vi.fn()  const Child = () => {    const snap = useSnapshot(obj)    childRenderFn(snap.childCount)    return (      <>        <div>childCount: {snap.childCount}</div>        <button onClick={() => ++obj.childCount}>childButton</button>      </>    )  }
  const parentRenderFn = vi.fn()  const Parent = () => {    const snap = useSnapshot(obj)    parentRenderFn(snap.parentCount)    return (      <>        <div>parentCount: {snap.parentCount}</div>        <button onClick={() => ++obj.parentCount}>parentButton</button>        <Child />      </>    )  }
  const { getByText } = render(<Parent />)
  await waitFor(() => {    getByText('childCount: 0')    getByText('parentCount: 0')  })
  expect(childRenderFn).toBeCalledTimes(1)  expect(childRenderFn).lastCalledWith(0)  expect(parentRenderFn).toBeCalledTimes(1)  expect(parentRenderFn).lastCalledWith(0)
  obj.parentCount += 1
  await waitFor(() => {    getByText('childCount: 0')    getByText('parentCount: 1')  })
  expect(childRenderFn).toBeCalledTimes(2)  expect(childRenderFn).lastCalledWith(0)  expect(parentRenderFn).toBeCalledTimes(2)  expect(parentRenderFn).lastCalledWith(1)})
it('regression: useSnapshot renders should not fail consistency check with extra render', async () => {  const obj = proxy({ childCount: 0, anotherValue: 0 })
  const childRenderFn = vi.fn()  const Child = () => {    const snap = useSnapshot(obj)    childRenderFn(snap.childCount)    return (      <>        <div>childCount: {snap.childCount}</div>        <button onClick={() => ++obj.childCount}>childButton</button>      </>    )  }
  const parentRenderFn = vi.fn()  const Parent = () => {    const [parentCount, setParentCount] = useState(0)
    parentRenderFn(parentCount)
    return (      <>        <div>parentCount: {parentCount}</div>        <button onClick={() => setParentCount((v) => v + 1)}>          parentButton        </button>        <Child />      </>    )  }
  const { getByText } = render(<Parent />)
  await waitFor(() => {    getByText('childCount: 0')    getByText('parentCount: 0')  })
  expect(childRenderFn).toBeCalledTimes(1)  expect(childRenderFn).lastCalledWith(0)  expect(parentRenderFn).toBeCalledTimes(1)  expect(parentRenderFn).lastCalledWith(0)
  obj.anotherValue += 1
  fireEvent.click(getByText('parentButton'))
  await waitFor(() => {    getByText('childCount: 0')    getByText('parentCount: 1')  })
  expect(childRenderFn).toBeCalledTimes(2)  expect(childRenderFn).lastCalledWith(0)  expect(parentRenderFn).toBeCalledTimes(2)  expect(parentRenderFn).lastCalledWith(1)})