use constant instead of util function for getting button label

Author: ncarbon

packages/compass-collection/src/components/mock-data-generator-modal/constants.ts

@@ -0,0 +1,10 @@
+import { MockDataGeneratorStep } from './types';
+
+export const StepButtonLabelMap = {
+  [MockDataGeneratorStep.AI_DISCLAIMER]: 'Use Natural Language',
+  [MockDataGeneratorStep.SCHEMA_CONFIRMATION]: 'Confirm',
+  [MockDataGeneratorStep.SCHEMA_EDITOR]: 'Next',
+  [MockDataGeneratorStep.DOCUMENT_COUNT]: 'Next',
+  [MockDataGeneratorStep.PREVIEW_DATA]: 'Generate Script',
+  [MockDataGeneratorStep.GENERATE_DATA]: 'Done',
+} as const;

packages/compass-collection/src/components/mock-data-generator-modal/mock-data-generator-modal.spec.tsx

@@ -4,6 +4,7 @@ import { render, screen } from '@mongodb-js/testing-library-compass';
 import Sinon from 'sinon';
 import MockDataGeneratorModal from './mock-data-generator-modal';
 import { MockDataGeneratorStep } from './types';
+import { StepButtonLabelMap } from './constants';
 
 describe('MockDataGeneratorModal', () => {
   const sandbox = Sinon.createSandbox();
@@ -87,20 +88,9 @@ describe('MockDataGeneratorModal', () => {
   it('renders the next step button with the correct label on each step', () => {
     renderModal();
 
-    expect(screen.getByTestId('next-step-button')).to.have.text(
-      'Use Natural Language'
-    );
-    screen.getByTestId('next-step-button').click();
-    expect(screen.getByTestId('next-step-button')).to.have.text('Confirm');
-    screen.getByTestId('next-step-button').click();
-    expect(screen.getByTestId('next-step-button')).to.have.text('Next');
-    screen.getByTestId('next-step-button').click();
-    expect(screen.getByTestId('next-step-button')).to.have.text('Next');
-    screen.getByTestId('next-step-button').click();
-    expect(screen.getByTestId('next-step-button')).to.have.text(
-      'Generate Script'
-    );
-    screen.getByTestId('next-step-button').click();
-    expect(screen.getByTestId('next-step-button')).to.have.text('Done');
+    Object.values(StepButtonLabelMap).forEach((label) => {
+      expect(screen.getByTestId('next-step-button')).to.have.text(label);
+      screen.getByTestId('next-step-button').click();
+    });
   });
 });

packages/compass-collection/src/components/mock-data-generator-modal/mock-data-generator-modal.tsx

@@ -9,7 +9,7 @@ import {
   ButtonVariant,
 } from '@mongodb-js/compass-components';
 import { MockDataGeneratorStep } from './types';
-import { getNextStepButtonLabel } from './utils';
+import { StepButtonLabelMap } from './constants';
 
 const footerStyles = css`
   flex-direction: row;
@@ -84,7 +84,7 @@ const MockDataGeneratorModal = ({
             onClick={onNext}
             data-testid="next-step-button"
           >
-            {getNextStepButtonLabel(currentStep)}
+            {StepButtonLabelMap[currentStep]}
           </Button>
         </div>
       </ModalFooter>