Address comment to add tooltip

Author: jcobis

packages/compass-collection/src/components/collection-header-actions/collection-header-actions.spec.tsx

@@ -265,5 +265,34 @@ describe('CollectionHeaderActions [Component]', function () {
 
       expect(onOpenMockDataModal).to.have.been.calledOnce;
     });
+
+    it('should disable button for deeply nested collections', async function () {
+      mockUseAssignment.returns({
+        assignment: {
+          assignmentData: {
+            variant: 'mockDataGeneratorVariant', // Treatment variant
+          },
+        },
+      });
+
+      await renderCollectionHeaderActions(
+        {
+          namespace: 'test.collection',
+          isReadonly: false,
+          hasSchemaAnalysisData: true,
+          analyzedSchemaDepth: 5, // Exceeds MAX_COLLECTION_NESTING_DEPTH (3)
+          schemaAnalysisStatus: 'complete',
+          onOpenMockDataModal: sinon.stub(),
+        },
+        {},
+        atlasConnectionInfo
+      );
+
+      const button = screen.getByTestId(
+        'collection-header-generate-mock-data-button'
+      );
+      expect(button).to.exist;
+      expect(button).to.have.attribute('aria-disabled', 'true');
+    });
   });
 });

packages/compass-collection/src/components/collection-header-actions/collection-header-actions.tsx

@@ -99,8 +99,14 @@ const CollectionHeaderActions: React.FunctionComponent<
     isInMockDataTreatmentVariant &&
     atlasMetadata && // Only show in Atlas
     !isReadonly && // Don't show for readonly collections (views)
-    !sourceName && // sourceName indicates it's a view
-    analyzedSchemaDepth <= MAX_COLLECTION_NESTING_DEPTH; // Filter out overly nested collections
+    !sourceName; // sourceName indicates it's a view
+
+  const exceedsMaxNestingDepth =
+    analyzedSchemaDepth > MAX_COLLECTION_NESTING_DEPTH;
+
+  const isCollectionEmpty =
+    !hasSchemaAnalysisData &&
+    schemaAnalysisStatus !== SCHEMA_ANALYSIS_STATE_ANALYZING;
 
   return (
     <div
@@ -124,16 +130,13 @@ const CollectionHeaderActions: React.FunctionComponent<
       )}
       {shouldShowMockDataButton && (
         <Tooltip
-          enabled={
-            !hasSchemaAnalysisData &&
-            schemaAnalysisStatus !== SCHEMA_ANALYSIS_STATE_ANALYZING
-          }
+          enabled={exceedsMaxNestingDepth || isCollectionEmpty}
           trigger={
             <div>
               <Button
                 data-testid="collection-header-generate-mock-data-button"
                 size={ButtonSize.Small}
-                disabled={!hasSchemaAnalysisData}
+                disabled={!hasSchemaAnalysisData || exceedsMaxNestingDepth}
                 onClick={onOpenMockDataModal}
                 leftGlyph={<Icon glyph="Sparkle" />}
               >
@@ -142,7 +145,10 @@ const CollectionHeaderActions: React.FunctionComponent<
             </div>
           }
         >
-          Please add data to your collection to generate similar mock documents
+          {exceedsMaxNestingDepth &&
+            'At this time we are unable to generate mock data for collections that have deeply nested documents'}
+          {isCollectionEmpty &&
+            'Please add data to your collection to generate similar mock documents'}
         </Tooltip>
       )}
       {atlasMetadata && (

packages/compass-collection/src/components/collection-header/collection-header.spec.tsx

@@ -473,7 +473,7 @@ describe('CollectionHeader [Component]', function () {
       expect(button).to.have.attribute('aria-disabled', 'true');
     });
 
-    it('should not show Mock Data Generator button for collections with excessive nesting depth', async function () {
+    it('should disable Mock Data Generator button for collections with excessive nesting depth', async function () {
       await renderCollectionHeaderWithExperimentation(
         {
           isAtlas: true,
@@ -495,9 +495,11 @@ describe('CollectionHeader [Component]', function () {
         atlasConnectionInfo
       );
 
-      expect(
-        screen.queryByTestId('collection-header-generate-mock-data-button')
-      ).to.not.exist;
+      const button = screen.getByTestId(
+        'collection-header-generate-mock-data-button'
+      );
+      expect(button).to.exist;
+      expect(button).to.have.attribute('aria-disabled', 'true');
     });
 
     it('should not show Mock Data Generator button for readonly collections (views)', async function () {