/*** Copyright (c) Meta Platforms, Inc. and affiliates.** This source code is licensed under the MIT license found in the* LICENSE file in the root directory of this source tree.** @flow*/import {parse} from '@babel/parser';
import {generateEncodedHookMap, generateHookMap} from '../generateHookMap';
function expectHookMapToEqual(actual, expected) {
expect(actual.names).toEqual(expected.names);
const formattedMappings = [];
actual.mappings.forEach(lines => {
lines.forEach(segment => {
const name = actual.names[segment[2]];
if (name == null) {
throw new Error(`Expected to find name at position ${segment[2]}`);
}formattedMappings.push(`${name} from ${segment[0]}:${segment[1]}`);
});});expect(formattedMappings).toEqual(expected.mappings);
}describe('generateHookMap', () => {
it('should parse names for built-in hooks', () => {
const code = `
import {useState, useContext, useMemo, useReducer} from 'react';export function Component() {const a = useMemo(() => A);const [b, setB] = useState(0);// prettier-ignoreconst c = useContext(A), d = useContext(B); // eslint-disable-line one-varconst [e, dispatch] = useReducer(reducer, initialState);const f = useRef(null)return a + b + c + d + e + f.current;}`;
const parsed = parse(code, {
sourceType: 'module',
plugins: ['jsx', 'flow'],
});const hookMap = generateHookMap(parsed);
expectHookMapToEqual(hookMap, {
names: ['<no-hook>', 'a', 'b', 'c', 'd', 'e', 'f'],
mappings: ['<no-hook> from 1:0',
'a from 5:12',
'<no-hook> from 5:28',
'b from 6:20',
'<no-hook> from 6:31',
'c from 9:12',
'<no-hook> from 9:25',
'd from 9:31',
'<no-hook> from 9:44',
'e from 11:24',
'<no-hook> from 11:57',
'f from 12:12',
'<no-hook> from 12:24',
],});const encodedHookMap = generateEncodedHookMap(parsed);
expect(encodedHookMap).toMatchInlineSnapshot(`
{"mappings": "CAAD;KYCA,AgBDA;MREA,AWFA;SnBGA,AaHA,AMIA,AaJA;WpBKA,AiCLA;Y7CMA,AYNA","names": ["<no-hook>","a","b","c","d","e","f",],}`);
});it('should parse names for custom hooks', () => {
const code = `
import useTheme from 'useTheme';import useValue from 'useValue';export function Component() {const theme = useTheme();const [val, setVal] = useValue();return theme;}`;
const parsed = parse(code, {
sourceType: 'module',
plugins: ['jsx', 'flow'],
});const hookMap = generateHookMap(parsed);
expectHookMapToEqual(hookMap, {
names: ['<no-hook>', 'theme', 'val'],
mappings: ['<no-hook> from 1:0',
'theme from 6:16',
'<no-hook> from 6:26',
'val from 7:24',
'<no-hook> from 7:34',
],});const encodedHookMap = generateEncodedHookMap(parsed);
expect(encodedHookMap).toMatchInlineSnapshot(`
{"mappings": "CAAD;MgBCA,AUDA;OFEA,AUFA","names": ["<no-hook>","theme","val",],}`);
});it('should parse names for nested hook calls', () => {
const code = `
import {useMemo, useState} from 'react';export function Component() {const InnerComponent = useMemo(() => () => {const [state, setState] = useState(0);return state;});return null;}`;
const parsed = parse(code, {
sourceType: 'module',
plugins: ['jsx', 'flow'],
});const hookMap = generateHookMap(parsed);
expectHookMapToEqual(hookMap, {
names: ['<no-hook>', 'InnerComponent', 'state'],
mappings: ['<no-hook> from 1:0',
'InnerComponent from 5:25',
'state from 6:30',
'InnerComponent from 6:41',
'<no-hook> from 9:4',
],});const encodedHookMap = generateEncodedHookMap(parsed);
expect(encodedHookMap).toMatchInlineSnapshot(`
{"mappings": "CAAD;KyBCA;MKCA,AWDA;SrCDA","names": ["<no-hook>","InnerComponent","state",],}`);
});it('should skip names for non-nameable hooks', () => {
const code = `
import useTheme from 'useTheme';import useValue from 'useValue';export function Component() {const [val, setVal] = useState(0);useEffect(() => {// ...});useLayoutEffect(() => {// ...});return val;}`;
const parsed = parse(code, {
sourceType: 'module',
plugins: ['jsx', 'flow'],
});const hookMap = generateHookMap(parsed);
expectHookMapToEqual(hookMap, {
names: ['<no-hook>', 'val'],
mappings: ['<no-hook> from 1:0', 'val from 6:24', '<no-hook> from 6:35'],
});const encodedHookMap = generateEncodedHookMap(parsed);
expect(encodedHookMap).toMatchInlineSnapshot(`
{"mappings": "CAAD;MwBCA,AWDA","names": ["<no-hook>","val",],}`);
});});