diff --git a/packages/compass-collection/src/components/mock-data-generator-modal/mock-data-generator-modal.spec.tsx b/packages/compass-collection/src/components/mock-data-generator-modal/mock-data-generator-modal.spec.tsx index e758e4dbac6..919bf23eb7b 100644 --- a/packages/compass-collection/src/components/mock-data-generator-modal/mock-data-generator-modal.spec.tsx +++ b/packages/compass-collection/src/components/mock-data-generator-modal/mock-data-generator-modal.spec.tsx @@ -36,13 +36,13 @@ describe('MockDataGeneratorModal', () => { async function renderModal({ isOpen = true, currentStep = MockDataGeneratorStep.SCHEMA_CONFIRMATION, - enableGenAISampleDocumentPassing = false, + enableGenAISampleDocumentPassingOnAtlasProject = false, mockServices = createMockServices(), schemaAnalysis = defaultSchemaAnalysisState, connectionInfo, }: { isOpen?: boolean; - enableGenAISampleDocumentPassing?: boolean; + enableGenAISampleDocumentPassingOnAtlasProject?: boolean; currentStep?: MockDataGeneratorStep; mockServices?: any; connectionInfo?: ConnectionInfo; @@ -75,7 +75,7 @@ describe('MockDataGeneratorModal', () => { connectionInfo, { preferences: { - enableGenAISampleDocumentPassing, + enableGenAISampleDocumentPassingOnAtlasProject, }, } ); @@ -229,7 +229,9 @@ describe('MockDataGeneratorModal', () => { }); it('uses the appropriate copy when Generative AI sample document passing is enabled', async () => { - await renderModal({ enableGenAISampleDocumentPassing: true }); + await renderModal({ + enableGenAISampleDocumentPassingOnAtlasProject: true, + }); expect(screen.getByText('Sample Documents Collected')).to.exist; expect( screen.getByText( diff --git a/packages/compass-collection/src/components/mock-data-generator-modal/raw-schema-confirmation-screen.tsx b/packages/compass-collection/src/components/mock-data-generator-modal/raw-schema-confirmation-screen.tsx index e55182e192c..1717e28ac7b 100644 --- a/packages/compass-collection/src/components/mock-data-generator-modal/raw-schema-confirmation-screen.tsx +++ b/packages/compass-collection/src/components/mock-data-generator-modal/raw-schema-confirmation-screen.tsx @@ -50,7 +50,7 @@ const RawSchemaConfirmationScreen = ({ fakerSchemaGenerationStatus, }: RawSchemaConfirmationScreenProps) => { const enableSampleDocumentPassing = usePreference( - 'enableGenAISampleDocumentPassing' + 'enableGenAISampleDocumentPassingOnAtlasProject' ); const subtitleText = enableSampleDocumentPassing