@kaizen/components 1.69.1 → 1.70.0

package/codemods/migrateWellVariantToColor/index.ts

@@ -1,14 +1,16 @@
 import { transformComponentsInDir } from '../utils'
 import { transformWellVariantToColor } from './transformWellVariantToColor'
 
-const migrateWellVariantToColor = (): void => {
+const run = (): void => {
   console.log(' ~(-_- ~) Running Well transformer (~ -_-)~')
   const targetDir = process.argv[2]
   if (!targetDir) {
     process.exit(1)
   }
 
-  transformComponentsInDir(targetDir, transformWellVariantToColor, 'Well')
+  transformComponentsInDir(targetDir, ['Well'], (tagNames) => [
+    transformWellVariantToColor(tagNames),
+  ])
 }
 
-migrateWellVariantToColor()
+run()

package/codemods/migrateWellVariantToColor/transformWellVariantToColor.spec.ts

@@ -1,8 +1,21 @@
 import { parseJsx } from '../__tests__/utils'
-import { printAst, transformSourceForTagName } from '../utils'
+import {
+  getKaioTagNamesMapByComponentName,
+  printAst,
+  transformSource,
+  type TransformSourceArgs,
+} from '../utils'
 import { transformWellVariantToColor } from './transformWellVariantToColor'
 
-describe('transformWellVariantToColor', () => {
+const transformWell = (sourceFile: TransformSourceArgs['sourceFile']): string => {
+  const kaioTagNamesMap = getKaioTagNamesMapByComponentName(sourceFile, ['Well'])
+  return transformSource({
+    sourceFile,
+    transformers: [transformWellVariantToColor(kaioTagNamesMap!)],
+  })
+}
+
+describe('transformWellVariantToColor()', () => {
   it('replaces variant="default" with color="gray"', () => {
     const inputAst = parseJsx(`
       import {Well} from "@kaizen/components"
@@ -12,12 +25,7 @@ describe('transformWellVariantToColor', () => {
       import {Well} from "@kaizen/components"
       export const TestComponent = () => <Well color="gray">Test</Well>
     `)
-    const transformed = transformSourceForTagName({
-      sourceFile: inputAst,
-      astTransformer: transformWellVariantToColor,
-      tagName: 'Well',
-    })
-    expect(transformed).toEqual(printAst(outputAst))
+    expect(transformWell(inputAst)).toEqual(printAst(outputAst))
   })
 
   it('replaces variant="informative" with color="blue"', () => {
@@ -29,12 +37,7 @@ describe('transformWellVariantToColor', () => {
       import {Well} from "@kaizen/components"
       export const TestComponent = () => <Well color="blue">Test</Well>
     `)
-    const transformed = transformSourceForTagName({
-      sourceFile: inputAst,
-      astTransformer: transformWellVariantToColor,
-      tagName: 'Well',
-    })
-    expect(transformed).toBe(printAst(outputAst))
+    expect(transformWell(inputAst)).toBe(printAst(outputAst))
   })
 
   it('adds color="gray" if variant is not specified', () => {
@@ -46,12 +49,7 @@ describe('transformWellVariantToColor', () => {
       import {Well} from "@kaizen/components"
       export const TestComponent = () => <Well color="gray">Test</Well>
     `)
-    const transformed = transformSourceForTagName({
-      sourceFile: inputAst,
-      astTransformer: transformWellVariantToColor,
-      tagName: 'Well',
-    })
-    expect(transformed).toBe(printAst(outputAst))
+    expect(transformWell(inputAst)).toBe(printAst(outputAst))
   })
 
   it('handles multiple attributes and replace only variant', () => {
@@ -63,12 +61,7 @@ describe('transformWellVariantToColor', () => {
       import {Well} from "@kaizen/components"
       export const TestComponent = () => <Well color="blue" id="123">Test</Well>
     `)
-    const transformed = transformSourceForTagName({
-      sourceFile: inputAst,
-      astTransformer: transformWellVariantToColor,
-      tagName: 'Well',
-    })
-    expect(transformed).toBe(printAst(outputAst))
+    expect(transformWell(inputAst)).toBe(printAst(outputAst))
   })
 
   it('handles multiple attributes and add color if variant is not specified', () => {
@@ -80,12 +73,7 @@ describe('transformWellVariantToColor', () => {
       import {Well} from "@kaizen/components"
       export const TestComponent = () => <Well id="123" color="gray">Test</Well>
     `)
-    const transformed = transformSourceForTagName({
-      sourceFile: inputAst,
-      astTransformer: transformWellVariantToColor,
-      tagName: 'Well',
-    })
-    expect(transformed).toBe(printAst(outputAst))
+    expect(transformWell(inputAst)).toBe(printAst(outputAst))
   })
 
   it('handles nested Well components', () => {
@@ -97,12 +85,7 @@ describe('transformWellVariantToColor', () => {
       import {Well} from "@kaizen/components"
       export const TestComponent = () => <div><Well id="123" color="gray">Test</Well></div>
     `)
-    const transformed = transformSourceForTagName({
-      sourceFile: inputAst,
-      astTransformer: transformWellVariantToColor,
-      tagName: 'Well',
-    })
-    expect(transformed).toBe(printAst(outputAst))
+    expect(transformWell(inputAst)).toBe(printAst(outputAst))
   })
 
   it('transforms multiple Wells', () => {
@@ -114,12 +97,7 @@ describe('transformWellVariantToColor', () => {
       import {Well} from "@kaizen/components"
       export const TestComponent = () => <div><Well color="blue">Test</Well><Well color="gray">Test 2</Well></div>
     `)
-    const transformed = transformSourceForTagName({
-      sourceFile: inputAst,
-      astTransformer: transformWellVariantToColor,
-      tagName: 'Well',
-    })
-    expect(transformed).toBe(printAst(outputAst))
+    expect(transformWell(inputAst)).toBe(printAst(outputAst))
   })
 
   it('transforms Well with arbitrary braces', () => {
@@ -131,12 +109,7 @@ describe('transformWellVariantToColor', () => {
       import {Well} from "@kaizen/components"
       export const TestComponent = () => <div><Well color="blue">Test</Well><Well color="orange">Test</Well><Well color="green">Test</Well></div>
     `)
-    const transformed = transformSourceForTagName({
-      sourceFile: inputAst,
-      astTransformer: transformWellVariantToColor,
-      tagName: 'Well',
-    })
-    expect(transformed).toBe(printAst(outputAst))
+    expect(transformWell(inputAst)).toBe(printAst(outputAst))
   })
 
   it("won't add color if color already exists", () => {
@@ -148,12 +121,7 @@ describe('transformWellVariantToColor', () => {
       import {Well} from "@kaizen/components"
       export const TestComponent = () => <div><Well color="blue">Test</Well></div>
     `)
-    const transformed = transformSourceForTagName({
-      sourceFile: inputAst,
-      astTransformer: transformWellVariantToColor,
-      tagName: 'Well',
-    })
-    expect(transformed).toBe(printAst(outputAst))
+    expect(transformWell(inputAst)).toBe(printAst(outputAst))
   })
 
   it("won't modify variants usings variables", () => {
@@ -165,12 +133,7 @@ describe('transformWellVariantToColor', () => {
       import {Well} from "@kaizen/components"
       export const TestComponent = () => <div><Well variant={wellVariable}>Test</Well></div>
     `)
-    const transformed = transformSourceForTagName({
-      sourceFile: inputAst,
-      astTransformer: transformWellVariantToColor,
-      tagName: 'Well',
-    })
-    expect(transformed).toBe(printAst(outputAst))
+    expect(transformWell(inputAst)).toBe(printAst(outputAst))
   })
 
   it('transforms aliased Well components', () => {
@@ -182,11 +145,6 @@ describe('transformWellVariantToColor', () => {
       import {Well as KaizenWell} from "@kaizen/components"
       export const TestComponent = () => <div><KaizenWell color="blue">Test</KaizenWell></div>
     `)
-    const transformed = transformSourceForTagName({
-      sourceFile: inputAst,
-      astTransformer: transformWellVariantToColor,
-      tagName: 'KaizenWell',
-    })
-    expect(transformed).toBe(printAst(outputAst))
+    expect(transformWell(inputAst)).toEqual(printAst(outputAst))
   })
 })

package/codemods/migrateWellVariantToColor/transformWellVariantToColor.ts

@@ -1,13 +1,19 @@
 import ts from 'typescript'
-import { updateJsxElementWithNewProps } from '../utils'
+import { updateJsxElementWithNewProps, type TagImportAttributesMap } from '../utils'
 import { getPropValueText } from '../utils/getPropValueText'
 
 export const transformWellVariantToColor =
-  (context: ts.TransformationContext, tagName: string) =>
-  (rootNode: ts.SourceFile): ts.SourceFile => {
+  (tagsMap: TagImportAttributesMap): ts.TransformerFactory<ts.SourceFile> =>
+  (context) =>
+  (rootNode) => {
     function visit(node: ts.Node): ts.Node {
       if (ts.isJsxOpeningElement(node) || ts.isJsxSelfClosingElement(node)) {
-        if (node.tagName.getText() === tagName) {
+        const tagName = node.tagName.getText()
+        const tagImportAttributes = tagsMap.get(tagName)
+
+        if (!tagImportAttributes) return node
+
+        if (node.tagName.getText() === tagImportAttributes.tagName) {
           let hasVariant = false
           let hasColor = false
           let newAttributes = node.attributes.properties.map((attr) => {

package/codemods/removeInputEditModalMood/index.ts

@@ -8,7 +8,9 @@ const run = (): void => {
     process.exit(1)
   }
 
-  transformComponentsInDir(targetDir, removeInputEditModalMood, 'InputEditModal')
+  transformComponentsInDir(targetDir, ['InputEditModal'], (tagNames) => [
+    removeInputEditModalMood(tagNames),
+  ])
 }
 
 run()

package/codemods/removeInputEditModalMood/removeInputEditModalMood.spec.ts

@@ -1,8 +1,26 @@
 import { parseJsx } from '../__tests__/utils'
-import { printAst, transformSourceForTagName } from '../utils'
+import { printAst, transformSource, type TransformSourceArgs } from '../utils'
 import { removeInputEditModalMood } from './removeInputEditModalMood'
 
-describe('removeInputEditModalMood', () => {
+const transformInputEditModal = (sourceFile: TransformSourceArgs['sourceFile']): string => {
+  const tagsMap = new Map([
+    [
+      'InputEditModal',
+      {
+        importModuleName: '@kaizen/components',
+        tagName: 'InputEditModal',
+        originalName: 'InputEditModal',
+      },
+    ],
+  ])
+
+  return transformSource({
+    sourceFile,
+    transformers: [removeInputEditModalMood(tagsMap)],
+  })
+}
+
+describe('removeInputEditModalMood()', () => {
   it('removes mood', () => {
     const inputAst = parseJsx(`
       export const TestComponent = () => <InputEditModal mood="positive"/>
@@ -10,12 +28,7 @@ describe('removeInputEditModalMood', () => {
     const outputAst = parseJsx(`
       export const TestComponent = () => <InputEditModal />
     `)
-    const transformed = transformSourceForTagName({
-      sourceFile: inputAst,
-      astTransformer: removeInputEditModalMood,
-      tagName: 'InputEditModal',
-    })
-    expect(transformed).toEqual(printAst(outputAst))
+    expect(transformInputEditModal(inputAst)).toEqual(printAst(outputAst))
   })
 
   it('handles multiple attributes and remove only mood', () => {
@@ -25,12 +38,7 @@ describe('removeInputEditModalMood', () => {
     const outputAst = parseJsx(`
       export const TestComponent = () => <InputEditModal id="123"/>
     `)
-    const transformed = transformSourceForTagName({
-      sourceFile: inputAst,
-      astTransformer: removeInputEditModalMood,
-      tagName: 'InputEditModal',
-    })
-    expect(transformed).toBe(printAst(outputAst))
+    expect(transformInputEditModal(inputAst)).toBe(printAst(outputAst))
   })
 
   it('transforms multiple InputEditModals', () => {
@@ -40,11 +48,6 @@ describe('removeInputEditModalMood', () => {
     const outputAst = parseJsx(`
       export const TestComponent = () => <div><InputEditModal /><InputEditModal /></div>
     `)
-    const transformed = transformSourceForTagName({
-      sourceFile: inputAst,
-      astTransformer: removeInputEditModalMood,
-      tagName: 'InputEditModal',
-    })
-    expect(transformed).toBe(printAst(outputAst))
+    expect(transformInputEditModal(inputAst)).toBe(printAst(outputAst))
   })
 })

package/codemods/removePopoverVariant/index.ts

@@ -8,7 +8,7 @@ const run = (): void => {
     process.exit(1)
   }
 
-  transformComponentsInDir(targetDir, removePopoverVariant, 'Popover')
+  transformComponentsInDir(targetDir, ['Popover'], (tagNames) => [removePopoverVariant(tagNames)])
 }
 
 run()

package/codemods/removePopoverVariant/removePopoverVariant.spec.ts

@@ -1,7 +1,25 @@
 import { parseJsx } from '../__tests__/utils'
-import { printAst, transformSourceForTagName } from '../utils'
+import { printAst, transformSource, type TransformSourceArgs } from '../utils'
 import { removePopoverVariant } from './removePopoverVariant'
 
+const transformPopover = (sourceFile: TransformSourceArgs['sourceFile']): string => {
+  const tagsMap = new Map([
+    [
+      'Popover',
+      {
+        importModuleName: '@kaizen/components',
+        tagName: 'Popover',
+        originalName: 'Popover',
+      },
+    ],
+  ])
+
+  return transformSource({
+    sourceFile,
+    transformers: [removePopoverVariant(tagsMap)],
+  })
+}
+
 describe('removePopoverVariant()', () => {
   it('removes variant', () => {
     const inputAst = parseJsx(`
@@ -10,12 +28,7 @@ describe('removePopoverVariant()', () => {
     const outputAst = parseJsx(`
       export const TestComponent = () => <Popover />
     `)
-    const transformed = transformSourceForTagName({
-      sourceFile: inputAst,
-      astTransformer: removePopoverVariant,
-      tagName: 'Popover',
-    })
-    expect(transformed).toEqual(printAst(outputAst))
+    expect(transformPopover(inputAst)).toEqual(printAst(outputAst))
   })
 
   it('removes customIcon', () => {
@@ -25,12 +38,7 @@ describe('removePopoverVariant()', () => {
     const outputAst = parseJsx(`
       export const TestComponent = () => <Popover />
     `)
-    const transformed = transformSourceForTagName({
-      sourceFile: inputAst,
-      astTransformer: removePopoverVariant,
-      tagName: 'Popover',
-    })
-    expect(transformed).toEqual(printAst(outputAst))
+    expect(transformPopover(inputAst)).toEqual(printAst(outputAst))
   })
 
   it('handles multiple attributes and remove only variant', () => {
@@ -40,12 +48,7 @@ describe('removePopoverVariant()', () => {
     const outputAst = parseJsx(`
       export const TestComponent = () => <Popover id="123"/>
     `)
-    const transformed = transformSourceForTagName({
-      sourceFile: inputAst,
-      astTransformer: removePopoverVariant,
-      tagName: 'Popover',
-    })
-    expect(transformed).toBe(printAst(outputAst))
+    expect(transformPopover(inputAst)).toBe(printAst(outputAst))
   })
 
   it('transforms multiple Popovers', () => {
@@ -55,11 +58,6 @@ describe('removePopoverVariant()', () => {
     const outputAst = parseJsx(`
       export const TestComponent = () => <div><Popover /><Popover /></div>
     `)
-    const transformed = transformSourceForTagName({
-      sourceFile: inputAst,
-      astTransformer: removePopoverVariant,
-      tagName: 'Popover',
-    })
-    expect(transformed).toBe(printAst(outputAst))
+    expect(transformPopover(inputAst)).toBe(printAst(outputAst))
   })
 })

package/codemods/upgradeIconV1/index.ts

@@ -1,6 +1,4 @@
-import fs from 'fs'
-import { getKaioTagNamesByRegex, transformSource, traverseDir } from '../utils'
-import { createEncodedSourceFile } from '../utils/createEncodedSourceFile'
+import { transformComponentsInDirByPattern } from '../utils'
 import { upgradeIconV1 } from './upgradeIconV1'
 
 const run = (): void => {
@@ -11,20 +9,7 @@ const run = (): void => {
     process.exit(1)
   }
 
-  const transformFile = (componentFilePath: string, sourceCode: string): void => {
-    const sourceFile = createEncodedSourceFile(componentFilePath, sourceCode)
-    const tagNames = getKaioTagNamesByRegex(sourceFile, 'Icon$')
-    if (tagNames) {
-      const updatedSourceFile = transformSource({
-        sourceFile,
-        transformers: [upgradeIconV1(tagNames)],
-      })
-
-      fs.writeFileSync(componentFilePath, updatedSourceFile, 'utf8')
-    }
-  }
-
-  traverseDir(targetDir, transformFile)
+  transformComponentsInDirByPattern(targetDir, 'Icon$', (tagNames) => [upgradeIconV1(tagNames)])
 }
 
 run()

package/codemods/upgradeIconV1/upgradeIconV1.spec.ts

@@ -1,20 +1,19 @@
 import { parseJsx } from '../__tests__/utils'
 import {
+  getKaioTagNamesMapByPattern,
   printAst,
   transformSource,
-  type ImportModuleNameTagsMap,
   type TransformSourceArgs,
 } from '../utils'
 import { upgradeIconV1 } from './upgradeIconV1'
 
-const transformIcons = (
-  sourceFile: TransformSourceArgs['sourceFile'],
-  tagNames: ImportModuleNameTagsMap,
-): string =>
-  transformSource({
+const transformIcons = (sourceFile: TransformSourceArgs['sourceFile']): string => {
+  const kaioTagNamesMap = getKaioTagNamesMapByPattern(sourceFile, 'Icon$')
+  return transformSource({
     sourceFile,
-    transformers: [upgradeIconV1(tagNames)],
+    transformers: [upgradeIconV1(kaioTagNamesMap!)],
   })
+}
 
 describe('upgradeIconV1()', () => {
   describe('CaMonogramIcon to Brand', () => {
@@ -29,12 +28,7 @@ describe('upgradeIconV1()', () => {
         import { Brand } from "@kaizen/components"
         export const TestComponent = () => <Brand ${transformedBrandProps} />
       `)
-      expect(
-        transformIcons(
-          inputAst,
-          new Map([['@kaizen/components', new Map([['CaMonogramIcon', 'CaMonogramIcon']])]]),
-        ),
-      ).toEqual(printAst(outputAst))
+      expect(transformIcons(inputAst)).toEqual(printAst(outputAst))
     })
 
     it('updates import from CaMonogramIcon using alias to Brand', () => {
@@ -46,12 +40,7 @@ describe('upgradeIconV1()', () => {
         import { Brand } from "@kaizen/components"
         export const TestComponent = () => <Brand ${transformedBrandProps} />
       `)
-      expect(
-        transformIcons(
-          inputAst,
-          new Map([['@kaizen/components', new Map([['LogoAlias', 'CaMonogramIcon']])]]),
-        ),
-      ).toEqual(printAst(outputAst))
+      expect(transformIcons(inputAst)).toEqual(printAst(outputAst))
     })
 
     it('does not add another Brand import if it is already imported', () => {
@@ -63,20 +52,7 @@ describe('upgradeIconV1()', () => {
         import { Brand } from "@kaizen/components"
         export const TestComponent = () => <Brand ${transformedBrandProps} />
       `)
-      expect(
-        transformIcons(
-          inputAst,
-          new Map([
-            [
-              '@kaizen/components',
-              new Map([
-                ['Brand', 'Brand'],
-                ['CaMonogramIcon', 'CaMonogramIcon'],
-              ]),
-            ],
-          ]),
-        ),
-      ).toEqual(printAst(outputAst))
+      expect(transformIcons(inputAst)).toEqual(printAst(outputAst))
     })
 
     it('uses Brand alias for an existing import', () => {
@@ -88,20 +64,7 @@ describe('upgradeIconV1()', () => {
         import { Brand as KzBrand } from "@kaizen/components"
         export const TestComponent = () => <KzBrand ${transformedBrandProps} />
       `)
-      expect(
-        transformIcons(
-          inputAst,
-          new Map([
-            [
-              '@kaizen/components',
-              new Map([
-                ['KzBrand', 'Brand'],
-                ['CaMonogramIcon', 'CaMonogramIcon'],
-              ]),
-            ],
-          ]),
-        ),
-      ).toEqual(printAst(outputAst))
+      expect(transformIcons(inputAst)).toEqual(printAst(outputAst))
     })
   })
 
@@ -117,12 +80,7 @@ describe('upgradeIconV1()', () => {
         import { LoadingSpinner } from "@kaizen/components"
         export const TestComponent = () => <LoadingSpinner ${transformedLoadingSpinnerProps} />
       `)
-      expect(
-        transformIcons(
-          inputAst,
-          new Map([['@kaizen/components', new Map([['SpinnerIcon', 'SpinnerIcon']])]]),
-        ),
-      ).toEqual(printAst(outputAst))
+      expect(transformIcons(inputAst)).toEqual(printAst(outputAst))
     })
 
     it('updates import from SpinnerIcon using alias to LoadingSpinner', () => {
@@ -134,12 +92,7 @@ describe('upgradeIconV1()', () => {
         import { LoadingSpinner } from "@kaizen/components"
         export const TestComponent = () => <LoadingSpinner ${transformedLoadingSpinnerProps} />
       `)
-      expect(
-        transformIcons(
-          inputAst,
-          new Map([['@kaizen/components', new Map([['LogoAlias', 'SpinnerIcon']])]]),
-        ),
-      ).toEqual(printAst(outputAst))
+      expect(transformIcons(inputAst)).toEqual(printAst(outputAst))
     })
 
     it('does not add another LoadingSpinner import if it is already imported', () => {
@@ -151,20 +104,7 @@ describe('upgradeIconV1()', () => {
         import { LoadingSpinner } from "@kaizen/components"
         export const TestComponent = () => <LoadingSpinner ${transformedLoadingSpinnerProps} />
       `)
-      expect(
-        transformIcons(
-          inputAst,
-          new Map([
-            [
-              '@kaizen/components',
-              new Map([
-                ['LoadingSpinner', 'LoadingSpinner'],
-                ['SpinnerIcon', 'SpinnerIcon'],
-              ]),
-            ],
-          ]),
-        ),
-      ).toEqual(printAst(outputAst))
+      expect(transformIcons(inputAst)).toEqual(printAst(outputAst))
     })
 
     it('uses LoadingSpinner alias for an existing import', () => {
@@ -176,20 +116,7 @@ describe('upgradeIconV1()', () => {
         import { LoadingSpinner as KzLoadingSpinner } from "@kaizen/components"
         export const TestComponent = () => <KzLoadingSpinner ${transformedLoadingSpinnerProps} />
       `)
-      expect(
-        transformIcons(
-          inputAst,
-          new Map([
-            [
-              '@kaizen/components',
-              new Map([
-                ['KzLoadingSpinner', 'LoadingSpinner'],
-                ['SpinnerIcon', 'SpinnerIcon'],
-              ]),
-            ],
-          ]),
-        ),
-      ).toEqual(printAst(outputAst))
+      expect(transformIcons(inputAst)).toEqual(printAst(outputAst))
     })
   })
 
@@ -202,12 +129,7 @@ describe('upgradeIconV1()', () => {
       import { Icon } from "@kaizen/components/future"
       export const TestComponent = () => <Icon name="flag" isFilled />
     `)
-    expect(
-      transformIcons(
-        inputAst,
-        new Map([['@kaizen/components', new Map([['FlagOnIcon', 'FlagOnIcon']])]]),
-      ),
-    ).toEqual(printAst(outputAst))
+    expect(transformIcons(inputAst)).toEqual(printAst(outputAst))
   })
 
   it('transforms aliased old Icon', () => {
@@ -219,12 +141,7 @@ describe('upgradeIconV1()', () => {
       import { Icon } from "@kaizen/components/future"
       export const TestComponent = () => <Icon name="menu" />
     `)
-    expect(
-      transformIcons(
-        inputAst,
-        new Map([['@kaizen/components', new Map([['IconAlias', 'HamburgerIcon']])]]),
-      ),
-    ).toEqual(printAst(outputAst))
+    expect(transformIcons(inputAst)).toEqual(printAst(outputAst))
   })
 
   describe('import statements', () => {
@@ -238,12 +155,7 @@ describe('upgradeIconV1()', () => {
         import { Icon } from "@kaizen/components/future"
         export const TestComponent = () => <Card><Icon name="add" /></Card>
       `)
-      expect(
-        transformIcons(
-          inputAst,
-          new Map([['@kaizen/components', new Map([['AddIcon', 'AddIcon']])]]),
-        ),
-      ).toEqual(printAst(outputAst))
+      expect(transformIcons(inputAst)).toEqual(printAst(outputAst))
     })
 
     it('does not update import of components which are not from KAIO', () => {
@@ -269,21 +181,7 @@ describe('upgradeIconV1()', () => {
           </>
         )
       `)
-      expect(
-        transformIcons(
-          inputAst,
-          new Map([
-            [
-              '@kaizen/components',
-              new Map([
-                ['AddIcon', 'AddIcon'],
-                ['IconAlias', 'HamburgerIcon'],
-              ]),
-            ],
-            ['somewhere-else', new Map([['HamHam', 'HamburgerIcon']])],
-          ]),
-        ),
-      ).toEqual(printAst(outputAst))
+      expect(transformIcons(inputAst)).toEqual(printAst(outputAst))
     })
   })
 })