[DevTools] Send heap snapshot to the frontend immediatly when it is ready

LayoutTests/inspector-protocol/heap-profiler/resources/heap-snapshot-common.js

 // This script is supposed to be evaluated in dummy inspector front-end which is loaded from
 // ../../../http/tests/inspector-protocol/resources/protocol-test.html and the relative paths
 // below are relative to that location.
 
 if (!window.WebInspector)
     window.WebInspector = {};
 InspectorTest.importScript("../../../../../Source/devtools/front_end/HeapSnapshot.js");
 InspectorTest.importScript("../../../../../Source/devtools/front_end/JSHeapSnapshot.js");
 InspectorTest.importScript("../../../../../Source/devtools/front_end/UIString.js");
 InspectorTest.importScript("../../../../../Source/devtools/front_end/utilities.js");
 
 InspectorTest.fail = function(message)
 {
     InspectorTest.log("FAIL: " + message);
     InspectorTest.completeTest();
 }
 
 InspectorTest.assert = function(result, message)
 {
     if (!result)
         InspectorTest.fail(message);
 }
 
 InspectorTest.takeHeapSnapshot = function(callback)
 {
-    InspectorTest.eventHandler["HeapProfiler.addProfileHeader"] = function(messageObject)
+    var chunks = [];
+    InspectorTest.eventHandler["HeapProfiler.addHeapSnapshotChunk"] = function(messageObject)
     {
-        var profileId = messageObject["params"]["header"]["uid"];
-        InspectorTest.sendCommand("HeapProfiler.getHeapSnapshot", { "uid": profileId }, didGetHeapSnapshot);
-
-        var chunks = [];
-        InspectorTest.eventHandler["HeapProfiler.addHeapSnapshotChunk"] = function(messageObject)
-        {
-            chunks.push(messageObject["params"]["chunk"]);
-        }
-
-        function didGetHeapSnapshot(messageObject)
-        {
-            var serializedSnapshot = chunks.join("");
-            var parsed = JSON.parse(serializedSnapshot);
-            var snapshot = new WebInspector.JSHeapSnapshot(parsed, new WebInspector.HeapSnapshotProgress());
-            callback(snapshot);
-            InspectorTest.log("SUCCESS: didGetHeapSnapshot");
-            InspectorTest.sendCommand("HeapProfiler.removeProfile", { "uid": profileId }, didRemoveSnapshot);
-        }
-
-        function didRemoveSnapshot(messageObject)
-        {
-            InspectorTest.completeTest();
-        }
+        chunks.push(messageObject["params"]["chunk"]);
     }
 
-    InspectorTest.sendCommand("HeapProfiler.takeHeapSnapshot", {});
+    function didTakeHeapSnapshot(messageObject)
+    {
+        var serializedSnapshot = chunks.join("");
+        var parsed = JSON.parse(serializedSnapshot);
+        var snapshot = new WebInspector.JSHeapSnapshot(parsed, new WebInspector.HeapSnapshotProgress());
+        callback(snapshot);
+        InspectorTest.log("SUCCESS: didGetHeapSnapshot");
+        InspectorTest.completeTest();
+    }
+    InspectorTest.sendCommand("HeapProfiler.takeHeapSnapshot", {}, didTakeHeapSnapshot);
 }