import { expectType } from 'ts-expect'import { atom, useAtom } from 'jotai'import type { Atom, PrimitiveAtom, WritableAtom } from 'jotai'import type { SetAtom } from '../src/core/atom'
it('atom() should return the correct types', () => { function Component() { const primitiveAtom = atom(0) expectType<PrimitiveAtom<number>>(primitiveAtom)
const readonlyDerivedAtom = atom((get) => get(primitiveAtom) * 2) expectType<Atom<number>>(readonlyDerivedAtom)
const readWriteDerivedAtom = atom( (get) => get(primitiveAtom), (get, set, value: number) => { set(primitiveAtom, get(primitiveAtom) + value) } ) expectType<WritableAtom<number, number>>(readWriteDerivedAtom)
const writeonlyDerivedAtom = atom(null, (get, set) => { set(primitiveAtom, get(primitiveAtom) - 1) }) expectType<WritableAtom<null, number>>(writeonlyDerivedAtom) } Component})
it('useAtom should return the correct types', () => { function Component() { const primitiveAtom = atom(0) expectType<[number, SetAtom<number, void>]>(useAtom(primitiveAtom))
const readonlyDerivedAtom = atom((get) => get(primitiveAtom) * 2) expectType<[number, SetAtom<number, void>]>(useAtom(readonlyDerivedAtom))
const readWriteDerivedAtom = atom( (get) => get(primitiveAtom), (get, set, value: number) => { set(primitiveAtom, get(primitiveAtom) + value) } ) expectType<[number, SetAtom<number, void>]>(useAtom(readWriteDerivedAtom))
const writeonlyDerivedAtom = atom(null, (get, set) => { set(primitiveAtom, get(primitiveAtom) - 1) }) expectType<[null, SetAtom<number, void>]>(useAtom(writeonlyDerivedAtom)) } Component})