1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859
import { expectType } from 'ts-expect'import { it } from 'vitest'import { atom } from 'jotai/vanilla'import type {  Atom,  ExtractAtomArgs,  ExtractAtomResult,  ExtractAtomValue,  PrimitiveAtom,  WritableAtom,} from 'jotai/vanilla'
it('atom() should return the correct types', () => {  function Component() {    // primitive atom    const primitiveAtom = atom(0)    expectType<PrimitiveAtom<number>>(primitiveAtom)
    // read-only derived atom    const readonlyDerivedAtom = atom((get) => get(primitiveAtom) * 2)    expectType<Atom<number>>(readonlyDerivedAtom)
    // read-write derived atom    const readWriteDerivedAtom = atom(      (get) => get(primitiveAtom),      (get, set, value: number) => {        set(primitiveAtom, get(primitiveAtom) + value)      }    )    expectType<WritableAtom<number, [number], void>>(readWriteDerivedAtom)
    // write-only derived atom    const writeonlyDerivedAtom = atom(null, (get, set) => {      set(primitiveAtom, get(primitiveAtom) - 1)    })    expectType<WritableAtom<null, [], void>>(writeonlyDerivedAtom)  }  Component})
it('type utils should work', () => {  function Component() {    const readWriteAtom = atom(      (_get) => 1 as number,      async (_get, _set, _value: string) => {}    )
    const value: ExtractAtomValue<typeof readWriteAtom> = 1    expectType<number>(value)
    const args: ExtractAtomArgs<typeof readWriteAtom> = ['']    expectType<[string]>(args)
    const result: ExtractAtomResult<typeof readWriteAtom> = Promise.resolve()    expectType<Promise<void>>(result)  }  Component})