fix tests

mui · May 22, 2021 · cdfcd7c · cdfcd7c
1 parent f5daf0e
commit cdfcd7c
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/packages/material-ui-codemod/src/v4.0.0/optimal-imports.js b/packages/material-ui-codemod/src/v4.0.0/optimal-imports.js
@@ -58,7 +58,9 @@ export default function transformer(fileInfo, api, options) {
           return;
         case 'ImportDefaultSpecifier': {
           const moduleName = match[2];
-          if (!whitelist.has(moduleName)) return;
+          if (!whitelist.has(moduleName) && moduleName !== 'withStyles') {
+            return;
+          }
           addSpecifier(
             targetImportPath,
             j.importSpecifier(j.identifier(moduleName), j.identifier(localName)),

diff --git a/packages/material-ui-codemod/src/v4.0.0/top-level-imports.js b/packages/material-ui-codemod/src/v4.0.0/top-level-imports.js
@@ -50,7 +50,12 @@ export default function transformer(fileInfo, api, options) {
           break;
         }
         case 'ImportSpecifier':
-          if (whitelist[specifier.imported.name] == null) return;
+          if (
+            whitelist[specifier.imported.name] == null &&
+            specifier.imported.name !== 'withStyles'
+          ) {
+            return;
+          }
           resultSpecifiers.push(specifier);
           path.get('specifiers', index).prune();
           break;