small fixes

Author: Lightning00Blade

src/McpResponse.ts

@@ -699,9 +699,11 @@ export class McpResponse implements Response {
       };
       pages?: object[];
       pagination?: object;
-      heapSnapshot?: object[];
-      heapSnapshotStats?: object;
-      heapSnapshotStaticData?: object;
+      heapSnapshot?: {
+        stats?: object;
+        staticData?: object;
+      };
+      heapSnapshotData?: object[];
       extensionServiceWorkers?: object[];
       extensionPages?: object[];
     } = {};
@@ -904,11 +906,13 @@ Call ${handleDialog.name} to handle it before continuing.`);
       const staticData = this.#heapSnapshotOptions.staticData;
       if (stats) {
         response.push(`Statistics: ${JSON.stringify(stats, null, 2)}`);
-        structuredContent.heapSnapshotStats = stats;
+        structuredContent.heapSnapshot = structuredContent.heapSnapshot || {};
+        structuredContent.heapSnapshot.stats = stats;
       }
       if (staticData) {
         response.push(`Static Data: ${JSON.stringify(staticData, null, 2)}`);
-        structuredContent.heapSnapshotStaticData = staticData;
+        structuredContent.heapSnapshot = structuredContent.heapSnapshot || {};
+        structuredContent.heapSnapshot.staticData = staticData;
       }
       const aggregates = this.#heapSnapshotOptions.aggregates;
       if (aggregates) {
@@ -926,7 +930,7 @@ Call ${handleDialog.name} to handle it before continuing.`);
         const formatter = new HeapSnapshotFormatter(paginatedRecord);
 
         response.push(formatter.toString());
-        structuredContent.heapSnapshot = formatter.toJSON();
+        structuredContent.heapSnapshotData = formatter.toJSON();
       }
     }
 

tests/formatters/HeapSnapshotFormatter.test.js.snapshot

@@ -0,0 +1,5 @@
+exports[`HeapSnapshotFormatter > toString > formats data as CSV and sorts by self size 1`] = `
+className,count,selfSize,maxRetainedSize
+"ObjectA",10,100,1000
+"ObjectB",5,50,500
+`;

tests/formatters/HeapSnapshotFormatter.test.ts

@@ -34,15 +34,10 @@ describe('HeapSnapshotFormatter', () => {
   };
 
   describe('toString', () => {
-    it('formats data as CSV and sorts by self size', () => {
+    it('formats data as CSV and sorts by self size', t => {
       const formatter = new HeapSnapshotFormatter(mockAggregates);
       const result = formatter.toString();
-      const expected = [
-        'className,count,selfSize,maxRetainedSize',
-        '"ObjectA",10,100,1000',
-        '"ObjectB",5,50,500',
-      ].join('\n');
-      assert.strictEqual(result, expected);
+      t.assert.snapshot?.(result);
     });
   });
 
@@ -76,12 +71,15 @@ describe('HeapSnapshotFormatter', () => {
         ObjectB: {
           name: 'ObjectB',
           self: 50,
-        } as unknown as DevTools.HeapSnapshotModel.HeapSnapshotModel.AggregatedInfo,
+        },
         ObjectA: {
           name: 'ObjectA',
           self: 100,
-        } as unknown as DevTools.HeapSnapshotModel.HeapSnapshotModel.AggregatedInfo,
-      };
+        },
+      } as unknown as Record<
+        string,
+        DevTools.HeapSnapshotModel.HeapSnapshotModel.AggregatedInfo
+      >;
 
       const result = HeapSnapshotFormatter.sort(unsortedAggregates);
       assert.strictEqual(result.length, 2);