123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144
import path from 'path'import { transformSync } from '@babel/core'
const plugin = path.join(__dirname, '../../src/babel/plugin-debug-label')
const transform = (  code: string,  filename?: string,  customAtomNames?: string[]) =>  transformSync(code, {    babelrc: false,    configFile: false,    filename,    plugins: [[plugin, { customAtomNames }]],  })?.code
it('Should add a debugLabel to an atom', () => {  expect(transform(`const countAtom = atom(0);`)).toMatchInlineSnapshot(`    "const countAtom = atom(0);    countAtom.debugLabel = "countAtom";"  `)})
it('Should handle a atom from a default export', () => {  expect(transform(`const countAtom = jotai.atom(0);`)).toMatchInlineSnapshot(`    "const countAtom = jotai.atom(0);    countAtom.debugLabel = "countAtom";"  `)})
it('Should handle a atom being exported', () => {  expect(transform(`export const countAtom = atom(0);`)).toMatchInlineSnapshot(`    "export const countAtom = atom(0);    countAtom.debugLabel = "countAtom";"  `)})
it('Should handle a default exported atom', () => {  expect(transform(`export default atom(0);`, 'countAtom.ts'))    .toMatchInlineSnapshot(`    "const countAtom = atom(0);    countAtom.debugLabel = "countAtom";    export default countAtom;"  `)})
it('Should handle a default exported atom in a barrel file', () => {  expect(transform(`export default atom(0);`, 'atoms/index.ts'))    .toMatchInlineSnapshot(`    "const atoms = atom(0);    atoms.debugLabel = "atoms";    export default atoms;"  `)})
it('Should handle all types of exports', () => {  expect(    transform(      `      export const countAtom = atom(0);      export default atom(0);    `,      'atoms/index.ts'    )  ).toMatchInlineSnapshot(`    "export const countAtom = atom(0);    countAtom.debugLabel = "countAtom";    const atoms = atom(0);    atoms.debugLabel = "atoms";    export default atoms;"  `)})
it('Should handle all atom types', () => {  expect(    transform(      `      export const countAtom = atom(0);
      const myFamily = atomFamily((param) => atom(param));
      const countAtomWithDefault = atomWithDefault((get) => get(countAtom) * 2);
      const observableAtom = atomWithObservable(() => {});
      const reducerAtom = atomWithReducer(0, () => {});
      const resetAtom = atomWithReset(0);
      const storageAtom = atomWithStorage('count', 1);
      const freezedAtom = freezeAtom(atom({ count: 0 }));
      const loadedAtom = loadable(countAtom);
      const selectedValueAtom = selectAtom(atom({ a: 0, b: 'othervalue' }), (v) => v.a);
      const splittedAtom = splitAtom(atom([]));    `,      'atoms/index.ts'    )  ).toMatchInlineSnapshot(`    "export const countAtom = atom(0);    countAtom.debugLabel = "countAtom";    const myFamily = atomFamily(param => atom(param));    myFamily.debugLabel = "myFamily";    const countAtomWithDefault = atomWithDefault(get => get(countAtom) * 2);    countAtomWithDefault.debugLabel = "countAtomWithDefault";    const observableAtom = atomWithObservable(() => {});    observableAtom.debugLabel = "observableAtom";    const reducerAtom = atomWithReducer(0, () => {});    reducerAtom.debugLabel = "reducerAtom";    const resetAtom = atomWithReset(0);    resetAtom.debugLabel = "resetAtom";    const storageAtom = atomWithStorage('count', 1);    storageAtom.debugLabel = "storageAtom";    const freezedAtom = freezeAtom(atom({      count: 0    }));    freezedAtom.debugLabel = "freezedAtom";    const loadedAtom = loadable(countAtom);    loadedAtom.debugLabel = "loadedAtom";    const selectedValueAtom = selectAtom(atom({      a: 0,      b: 'othervalue'    }), v => v.a);    selectedValueAtom.debugLabel = "selectedValueAtom";    const splittedAtom = splitAtom(atom([]));    splittedAtom.debugLabel = "splittedAtom";"  `)})
it('Handles custom atom names a debugLabel to an atom', () => {  expect(    transform(`const mySpecialThing = myCustomAtom(0);`, undefined, [      'myCustomAtom',    ])  ).toMatchInlineSnapshot(`    "const mySpecialThing = myCustomAtom(0);    mySpecialThing.debugLabel = "mySpecialThing";"  `)})