Convert enzyme imports to RTL

src/utils/ast-transformations/individual-transformations/convert-enzyme-imports.ts

@@ -0,0 +1,56 @@
+/**
+ * Find and convert all Enzyme import declarations to RTL imports
+ * Example:
+ * import { shallow } from 'enzyme'; --> import { render, screen } from '@testing-library/react';
+ */
+
+import type { Collection, JSCodeshift, ImportDeclaration } from 'jscodeshift';
+import { astLogger } from '../utils/ast-logger';
+
+/**
+ * Transforms the provided AST by converting all Enzyme imports to RTL imports.
+ * @param j - JSCodeshift library
+ * @param root - The root AST node
+ * @returns {void} - The function does not return a value but mutates the AST directly.
+ */
+export const convertImports = (j: JSCodeshift, root: Collection): void => {
+    // Find all import declarations matching `import { shallow } from 'enzyme';`
+    astLogger.verbose('Query for enzyme import declarations');
+    const enzymeImportDeclaration = root.find(j.ImportDeclaration, {
+        source: {
+            value: 'enzyme',
+        },
+    });
+
+    let newImportDeclaration: ImportDeclaration;
+
+    // Replace shallow and mount imports with import screen and render from rtl
+    astLogger.verbose(
+        'Convert shallow and mount imports with import screen and render from rtl',
+    );
+    if (enzymeImportDeclaration.length > 0) {
+        // create new import declaration for `import { render, screen } from '@testing-library/react';`
+        newImportDeclaration = j.importDeclaration(
+            [
+                j.importSpecifier(j.identifier('render')),
+                j.importSpecifier(j.identifier('screen')),
+            ],
+            j.literal('@testing-library/react'),
+        );
+
+        // remove enzyme imports
+        enzymeImportDeclaration.remove();
+    } else {
+        astLogger.verbose('No enzyme imports found');
+    }
+
+    // Get the top file node
+    const fileTopNode = root.find(j.Program);
+
+    // Add the new import
+    astLogger.verbose('Add RTL import');
+    fileTopNode.replaceWith((path) => {
+        path.node.body.unshift(newImportDeclaration);
+        return path.node;
+    });
+};

src/utils/ast-transformations/main-ast-transform.ts

@@ -2,6 +2,7 @@
 import jscodeshift from 'jscodeshift';
 import { convertFind } from './individual-transformations/convert-find';
 import { convertHostNodes } from './individual-transformations/remove-enzyme-hostNodes-method';
+import { convertImports } from './individual-transformations/convert-enzyme-imports';
 import { convertMountShallowMethods } from './individual-transformations/convert-mount-shallow-methods';
 import { convertMountShallowVars } from './individual-transformations/convert-mount-shallow-vars';
 import { convertSimulate } from './individual-transformations/convert-simulate';
@@ -28,7 +29,8 @@
     j.withParser('tsx');
 
     // Convert enzyme imports
-    // convertImports(j, root);
+    astLogger.verbose('Convert imports');
+    convertImports(j, root);
 
     /**
      * Convert mount and shallow to render
@@ -43,7 +45,7 @@
      * Step 2: Collect variable names that reference shallow and mount calls
      */
     astLogger.verbose('Convert mount/shallow vars');
     const renderFunctionVarNames = convertMountShallowVars(
         j,
         root,
         abstractedFunctionName,
@@ -54,6 +56,7 @@
     convertFind(j, root);
 
     // Convert text()
+    astLogger.verbose('Convert text()');
     convertText(j, root);
 
     // Convert simulate()

test/convert-enzyme-imports.jest.ts

@@ -0,0 +1,30 @@
+import { convertImports } from '../src/utils/ast-transformations/individual-transformations/convert-enzyme-imports';
+import jscodeshift from 'jscodeshift';
+
+describe('convertText', () => {
+    let j: jscodeshift.JSCodeshift;
+
+    beforeEach(() => {
+        j = jscodeshift.withParser('tsx');
+    });
+
+    it('Should convert Enzyme imports to RTL imports', () => {
+        const source = `
+            import { mount } from 'enzyme';
+        `;
+
+        // Transform the source code
+        const root = j(source);
+        convertImports(j, root);
+
+        // Generate the transformed source code
+        const transformedSource = root.toSource();
+
+        const expectedSource = `
+            import { render, screen } from "@testing-library/react";
+        `;
+
+        // Check if the transformed source matches the expected source
+        expect(transformedSource).toBe(expectedSource);
+    });
+});

test/data/ast-conversion-test-file-verify.jsx

@@ -1,6 +1,6 @@
 import userEvent from "@testing-library/user-event";
+import { render, screen } from "@testing-library/react";
 import React from 'react';
-import { mount } from 'enzyme';
 import Component from '../components/..';
 
 

test/data/ast-conversion-test-file-verify.tsx

@@ -1,6 +1,6 @@
 import userEvent from "@testing-library/user-event";
+import { render, screen } from "@testing-library/react";
 import React from 'react';
-import { mount } from 'enzyme';
 import Component from '../components/..';