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
 
 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:
+
+    print 'Snapshotting content of %s. This could take a while...' % 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))
+
+    # Sending all the serialized doms back to tab execution context.
+    tab.ExecuteJavaScript('var serializedDoms = [];')
+    for sub_dom in serialized_doms:
+      print len(json.dumps(sub_dom))
+      tab.ExecuteJavaScript('serializedDoms.push({{sub_dom}})', sub_dom=sub_dom)
+
+    # Combine all the doms to one HTML string.
+    tab.EvaluateJavaScript(dom_combining_script)
+    page_snapshot = tab.ExecuteJavaScript(
+        'var domString = outputHTMLString(serialized_doms);')
+
+    snapshot_file.write(page_snapshot)
   finally:
     browser.Close()