123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183
import { proxy, ref, subscribe } from 'valtio'import { subscribeKey } from 'valtio/utils'
const consoleWarn = console.warnbeforeEach(() => {  console.warn = jest.fn((message) => {    if (message === 'Please use proxy object') {      return    }    consoleWarn(message)  })})afterEach(() => {  console.warn = consoleWarn})
describe('subscribe', () => {  it('should call subscription', async () => {    const obj = proxy({ count: 0 })    const handler = jest.fn()
    subscribe(obj, handler)
    obj.count += 1
    await Promise.resolve()    expect(handler).toBeCalledTimes(1)  })
  it('should be able to unsubscribe', async () => {    const obj = proxy({ count: 0 })    const handler = jest.fn()
    const unsubscribe = subscribe(obj, handler)    unsubscribe()
    obj.count += 1
    await Promise.resolve()    expect(handler).toBeCalledTimes(0)  })
  it('should call subscription of object property', async () => {    const obj = proxy({ nested: { count: 0 } })    const handler = jest.fn()
    subscribe(obj.nested, handler)
    obj.nested.count += 1
    await Promise.resolve()    expect(handler).toBeCalledTimes(1)  })
  it('should thow if subscribing to primitive property', async () => {    const obj = proxy({ count: 0 })    const handler = jest.fn()
    expect(() => subscribe(obj.count as any, handler)).toThrow()  })
  it('should not re-run subscription if no change', async () => {    const obj = proxy({ count: 0 })    const handler = jest.fn()
    subscribe(obj, handler)
    obj.count = 0
    await Promise.resolve()    expect(handler).toBeCalledTimes(0)  })
  it('should not cause infinite loop', async () => {    const obj = proxy({ count: 0 })    const handler = () => {      // Reset count if above 5      if (obj.count > 5) {        obj.count = 0      }    }
    subscribe(obj, handler)
    obj.count = 10  })
  it('should batch updates', async () => {    const obj = proxy({ count1: 0, count2: 0 })    const handler = jest.fn()
    subscribe(obj, handler)
    obj.count1 += 1    obj.count2 += 1
    await Promise.resolve()    expect(handler).toBeCalledTimes(1)  })
  it('should not call subscription for objects wrapped in ref', async () => {    const obj = proxy({ nested: ref({ count: 0 }) })    const handler = jest.fn()
    subscribe(obj, handler)
    obj.nested.count += 1
    await Promise.resolve()    expect(handler).toBeCalledTimes(0)  })
  it('should notify ops', async () => {    const obj = proxy<{ count1: number; count2?: number }>({      count1: 0,      count2: 0,    })    const handler = jest.fn()
    subscribe(obj, handler)
    obj.count1 += 1    obj.count2 = 2
    await Promise.resolve()    expect(handler).toBeCalledTimes(1)    expect(handler).lastCalledWith([      ['set', ['count1'], 1, 0],      ['set', ['count2'], 2, 0],    ])
    delete obj.count2
    await Promise.resolve()    expect(handler).toBeCalledTimes(2)    expect(handler).lastCalledWith([['delete', ['count2'], 2]])  })
  it('should notify nested ops', async () => {    const obj = proxy<{ nested: { count?: number } }>({ nested: { count: 0 } })    const handler = jest.fn()
    subscribe(obj, handler)
    obj.nested.count = 1
    await Promise.resolve()    expect(handler).toBeCalledTimes(1)    expect(handler).lastCalledWith([['set', ['nested', 'count'], 1, 0]])
    delete obj.nested.count
    await Promise.resolve()    expect(handler).toBeCalledTimes(2)    expect(handler).lastCalledWith([['delete', ['nested', 'count'], 1]])  })})
describe('subscribeKey', () => {  it('should call subscription', async () => {    const obj = proxy({ count1: 0, count2: 0 })    const handler1 = jest.fn()    const handler2 = jest.fn()
    subscribeKey(obj, 'count1', handler1)    subscribeKey(obj, 'count2', handler2)
    obj.count1 += 10
    await Promise.resolve()    expect(handler1).toBeCalledTimes(1)    expect(handler1).lastCalledWith(10)    expect(handler2).toBeCalledTimes(0)
    obj.count2 += 20
    await Promise.resolve()    expect(handler1).toBeCalledTimes(1)    expect(handler2).toBeCalledTimes(1)    expect(handler2).lastCalledWith(20)  })})