test(drive-integration): add integration test for Save wire in AddEntryForm

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>

Author: JuliRossi

apps/drive-integration/test/locations/Page/components/modals/EditModal.spec.tsx

@@ -33,8 +33,8 @@ const contentTypes = [
 ];
 
 const existingEntries = [
-  { tempId: 'entry-1', label: 'Blog post' },
-  { tempId: 'entry-2', label: 'Landing page' },
+  { tempId: 'entry-1', label: 'Blog post', contentTypeId: 'article' },
+  { tempId: 'entry-2', label: 'Landing page', contentTypeId: 'page' },
 ];
 
 const baseNewLocation = {
@@ -188,7 +188,9 @@ describe('EditModal', () => {
       expect(screen.getByRole('button', { name: 'Back' })).toBeTruthy();
       expect(screen.getByRole('button', { name: 'Save' })).toBeDisabled();
       expect(screen.queryByRole('button', { name: 'Next' })).toBeNull();
-      expect(screen.queryByText('Should this entry be a reference entry?')).toBeNull();
+      expect(
+        screen.queryByText('Should this new entry be a reference of an existing entry?')
+      ).toBeNull();
     });
 
     it('shows radios and field multiselect together after content type is selected', async () => {
@@ -197,16 +199,20 @@ describe('EditModal', () => {
       fireEvent.change(screen.getByRole('combobox'), { target: { value: 'page' } });
 
       await waitFor(() => {
-        expect(screen.getByText('Should this entry be a reference entry?')).toBeTruthy();
+        expect(
+          screen.getByText('Should this new entry be a reference of an existing entry?')
+        ).toBeTruthy();
         expect(screen.getByLabelText('Yes')).toBeTruthy();
         expect(screen.getByLabelText('No')).toBeTruthy();
         expect(screen.getByText('Select the field(s) the content should map to')).toBeTruthy();
       });
 
-      expect(screen.queryByText('Select the entry this should reference')).toBeNull();
+      expect(
+        screen.queryByText('Which existing entry should this new entry be a reference to?')
+      ).toBeNull();
     });
 
-    it('shows reference entry select when Yes is chosen', async () => {
+    it('shows parent entry select when Yes is chosen', async () => {
       await openAddEntryForm();
 
       fireEvent.change(screen.getByRole('combobox'), { target: { value: 'article' } });
@@ -218,11 +224,51 @@ describe('EditModal', () => {
       fireEvent.click(screen.getByLabelText('Yes'));
 
       await waitFor(() => {
-        expect(screen.getByText('Select the entry this should reference')).toBeTruthy();
+        expect(
+          screen.getByText('Which existing entry should this new entry be a reference to?')
+        ).toBeTruthy();
         expect(screen.getByText('Blog post')).toBeTruthy();
       });
     });
 
+    it('calls onAddEntry with the correct params when Save is clicked after filling the form', async () => {
+      await openAddEntryForm();
+
+      fireEvent.change(screen.getByRole('combobox'), { target: { value: 'page' } });
+
+      await waitFor(() => {
+        expect(screen.getByLabelText('No')).toBeTruthy();
+      });
+
+      fireEvent.click(screen.getByLabelText('No'));
+
+      await waitFor(() => {
+        expect(screen.getByRole('button', { name: 'Toggle Multiselect' })).toBeTruthy();
+      });
+
+      fireEvent.click(screen.getByRole('button', { name: 'Toggle Multiselect' }));
+
+      await waitFor(() => {
+        expect(screen.getByText('Title')).toBeTruthy();
+      });
+
+      fireEvent.click(screen.getByText('Title'));
+
+      await waitFor(() => {
+        expect(screen.getByRole('button', { name: 'Save' })).not.toBeDisabled();
+      });
+
+      fireEvent.click(screen.getByRole('button', { name: 'Save' }));
+
+      expect(onAddEntry).toHaveBeenCalledWith({
+        contentTypeId: 'page',
+        isReference: false,
+        referenceEntryId: null,
+        referenceFieldId: null,
+        fieldIds: ['title'],
+      });
+    });
+
     it('exits the form when Back is clicked', async () => {
       await openAddEntryForm();