From a3b40eb85a60889a28190cebc6b9c90999b8ce8b Mon Sep 17 00:00:00 2001
From: Benny Joo <sldisek783@gmail.com>
Date: Sun, 4 Dec 2022 19:28:18 +0000
Subject: [PATCH] Improve tests for codemod

---
 .../actual.js                                 | 22 +++++++++++++++----
 .../expected.js                               | 22 +++++++++++++++----
 2 files changed, 36 insertions(+), 8 deletions(-)

diff --git a/packages/mui-codemod/src/v5.0.0/joy-rename-components-to-slots.test/actual.js b/packages/mui-codemod/src/v5.0.0/joy-rename-components-to-slots.test/actual.js
index 0156a966be03bf..25f77a6d872e0f 100644
--- a/packages/mui-codemod/src/v5.0.0/joy-rename-components-to-slots.test/actual.js
+++ b/packages/mui-codemod/src/v5.0.0/joy-rename-components-to-slots.test/actual.js
@@ -1,4 +1,18 @@
-<Autocomplete
-  components={{ Root, Input: CustomInput }}
-  componentsProps={{ root: { className: 'root' }, input: { 'data-testid': 'input' } }}
-/>;
\ No newline at end of file
+// the codemod should transform only Joy UI components;
+import { Alert as JoyAlert } from '@mui/joy';
+import JoyAutocomplete from '@mui/joy/Autocomplete';
+import CustomComponent from 'components/Custom';
+
+<div>
+  <JoyAlert
+    components={{ Root, Input: CustomInput }}
+    componentsProps={{ root: { className: 'root' }, input: { 'data-testid': 'input' } }}
+  />
+  <JoyAutocomplete
+    components={{ Root, Input: CustomInput }}
+    componentsProps={{ root: { className: 'root' }, input: { 'data-testid': 'input' } }}
+  />
+  <CustomComponent
+    componentsProps={{ root: { className: 'root' }, input: { 'data-testid': 'input' } }}
+  />
+</div>;
diff --git a/packages/mui-codemod/src/v5.0.0/joy-rename-components-to-slots.test/expected.js b/packages/mui-codemod/src/v5.0.0/joy-rename-components-to-slots.test/expected.js
index 8e31e325123497..a46c8938a788e2 100644
--- a/packages/mui-codemod/src/v5.0.0/joy-rename-components-to-slots.test/expected.js
+++ b/packages/mui-codemod/src/v5.0.0/joy-rename-components-to-slots.test/expected.js
@@ -1,4 +1,18 @@
-<Autocomplete
-  slots={{ root: Root, input: CustomInput }}
-  slotProps={{ root: { className: 'root' }, input: { 'data-testid': 'input' } }}
-/>;
\ No newline at end of file
+// the codemod should transform only Joy UI slots;
+import { Alert as JoyAlert } from '@mui/joy';
+import JoyAutocomplete from '@mui/joy/Autocomplete';
+import CustomComponent from 'components/Custom';
+
+<div>
+  <JoyAlert
+    slots={{ Root, Input: CustomInput }}
+    slotProps={{ root: { className: 'root' }, input: { 'data-testid': 'input' } }}
+  />
+  <JoyAutocomplete
+    slots={{ Root, Input: CustomInput }}
+    slotProps={{ root: { className: 'root' }, input: { 'data-testid': 'input' } }}
+  />
+  <CustomComponent
+    slotProps={{ root: { className: 'root' }, input: { 'data-testid': 'input' } }}
+  />
+</div>;