Make sure snap_page combined iframe serialized dom

telemetry/telemetry/internal/snap_page_util.py

 # Copyright (c) 2017 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 import os
+import json
+import sys
 
 from telemetry.core import util
 from telemetry.internal.browser import browser_finder
 
 
 def SnapPage(finder_options, url, interactive, snapshot_file):
   """ Save the HTML snapshot of the page whose address is |url| to
   |snapshot_file|.
   """
   possible_browser = browser_finder.FindBrowser(finder_options)
   browser = possible_browser.Create(finder_options)
   try:
     tab = browser.tabs[0]
     tab.Navigate(url)
-    tab.WaitForDocumentReadyStateToBeComplete()
     if interactive:
       raw_input(
           'Activating interactive mode. Press enter after you finish '
           "interacting with the page to snapshot the page's DOM content.")
-    with open(
-        os.path.join(util.GetTelemetryThirdPartyDir(), 'snap-it',
-                     'HTMLSerializer.js')) as f:
+
+    sys.stdout.write(
+        'Snapshotting content of %s. This could take a while...\n' % url)
+    tab.WaitForDocumentReadyStateToBeComplete()
+    tab.action_runner.WaitForNetworkQuiescence()
+
+    with open(os.path.join(util.GetTelemetryThirdPartyDir(), 'snap-it',
+                           'HTMLSerializer.js')) as f:
       snapit_script = f.read()
-    tab.ExecuteJavaScript(snapit_script)
-    tab.ExecuteJavaScript(
-        '''
-        var serializedDomArray;
-        var htmlSerializer = new HTMLSerializer();
-        htmlSerializer.processDocument(document);
-        htmlSerializer.fillHolesAsync(document, function(s) {
-          serializedDomArray = s.html;
-        });
-        ''')
-    print 'Snapshotting content of %s. This could take a while...' % url
-    tab.WaitForJavaScriptCondition('serializedDomArray !== undefined')
-    serialized_dom = ''.join(tab.EvaluateJavaScript('serializedDomArray'))
-    snapshot_file.write(serialized_dom)
+
+    with open(os.path.join(util.GetTelemetryThirdPartyDir(), 'snap-it',
+                           'popup.js')) as f:
+      dom_combining_script = f.read()
+
+    serialized_doms = []
+
+    # Serialize the dom in each frame.
+    for context_id in tab.EnableAllContexts():
+      tab.ExecuteJavaScript(snapit_script, context_id=context_id)
+      tab.ExecuteJavaScript(
+          '''
+          var serializedDom;
+          var htmlSerializer = new HTMLSerializer();
+          htmlSerializer.processDocument(document);
+          htmlSerializer.fillHolesAsync(document, function(s) {
+            serializedDom = s.asDict();
+          });
+          ''', context_id=context_id)
+      tab.WaitForJavaScriptCondition(
+          'serializedDom !== undefined', context_id=context_id)
+      serialized_doms.append(tab.EvaluateJavaScript(
+          'serializedDom', context_id=context_id))
+
+    # Execute doms combining code in blank page to minimize the chance of V8
+    # OOM.
+    tab.Navigate('about:blank')
+    tab.WaitForDocumentReadyStateToBeComplete()
+
+    # Sending all the serialized doms back to tab execution context.
+    tab.ExecuteJavaScript('var serializedDoms = [];')
+    for i in xrange(len(serialized_doms)):
+      sys.stdout.write('Processing dom of frame #%i / %i\r' %
+                       (i, len(serialized_doms)))
+      sys.stdout.flush()
+      sub_dom_string = json.dumps(serialized_doms[i])
+      tab.ExecuteJavaScript('var sub_dom = "";')
+
+      # To avoid crashing devtool connection (details in crbug.com/763119#c16),
+      # we break down the json string to chunks which each chunk has a maximum
+      # size of 100000 characters (100000 seems to not break the connection and
+      # makes sending data reasonably fast).
+      k = 0
+      step_size = 100000
+      while k < len(sub_dom_string):

[wkorman, 2017/09/27 17:15:13]

Worth adding a unit test to validate the chunking logic?

+        sub_dom_string_part = sub_dom_string[k: k + step_size]
+        k += step_size
+        tab.ExecuteJavaScript(
+            'sub_dom += {{sub_dom_string_part}};',
+            sub_dom_string_part=sub_dom_string_part)
+      tab.ExecuteJavaScript('serializedDoms.push(JSON.parse(sub_dom));')
+
+    # Combine all the doms to one HTML string.
+    tab.EvaluateJavaScript(dom_combining_script)
+    page_snapshot = tab.EvaluateJavaScript('outputHTMLString(serializedDoms);')
+
+    snapshot_file.write(page_snapshot)
   finally:
     browser.Close()