diff --git a/packages/mui-codemod/src/v5.0.0/top-level-imports.js b/packages/mui-codemod/src/v5.0.0/top-level-imports.js index 45fe3370cc87d5..347af368ad86cc 100644 --- a/packages/mui-codemod/src/v5.0.0/top-level-imports.js +++ b/packages/mui-codemod/src/v5.0.0/top-level-imports.js @@ -69,10 +69,7 @@ export default function transformer(fileInfo, api, options) { break; } case 'ImportSpecifier': - if ( - !whitelist.has(specifier.imported.name) == null && - specifier.imported.name !== 'withStyles' - ) { + if (!whitelist.has(specifier.imported.name) && specifier.imported.name !== 'withStyles') { return; } resultSpecifiers.push(specifier); diff --git a/packages/mui-codemod/src/v5.0.0/top-level-imports.test/actual.js b/packages/mui-codemod/src/v5.0.0/top-level-imports.test/actual.js index ddba1adfb56d53..aac4949c3bc108 100644 --- a/packages/mui-codemod/src/v5.0.0/top-level-imports.test/actual.js +++ b/packages/mui-codemod/src/v5.0.0/top-level-imports.test/actual.js @@ -1,7 +1,7 @@ import * as React from 'react'; import { withStyles } from '@mui/styles'; -import { ThemeProvider } from '@mui/material'; +import { ThemeProvider } from '@mui/material/styles'; import { createTheme } from '@mui/material/styles'; import MenuItem from '@mui/material/MenuItem'; import Tab from '@mui/material/Tab'; @@ -57,3 +57,6 @@ import TableFooter from '@mui/material/TableFooter'; import Zoom from '@mui/material/Zoom'; import ClickAwayListener from '@mui/material/ClickAwayListener'; import ListSubheader from '@mui/material/ListSubheader'; +import { grey, blue } from '@mui/material/colors'; +import { grey as grey2 } from '@mui/material/colors'; +import { blue as blue2 } from '@mui/material/colors'; diff --git a/packages/mui-codemod/src/v5.0.0/top-level-imports.test/expected.js b/packages/mui-codemod/src/v5.0.0/top-level-imports.test/expected.js index 9bc5702332c054..44888ab7ec0d85 100644 --- a/packages/mui-codemod/src/v5.0.0/top-level-imports.test/expected.js +++ b/packages/mui-codemod/src/v5.0.0/top-level-imports.test/expected.js @@ -1,9 +1,13 @@ import * as React from 'react'; import { withStyles } from '@mui/styles'; +import { ThemeProvider } from '@mui/material/styles'; +import { createTheme } from '@mui/material/styles'; +import { grey, blue } from '@mui/material/colors'; +import { grey as grey2 } from '@mui/material/colors'; +import { blue as blue2 } from '@mui/material/colors'; + import { - ThemeProvider, - createTheme, MenuItem, Tab, Tabs as MuiTabs,