[Local NTP] Add an integration test for the most visited iframe

chrome/test/data/local_ntp_browsertest.js

 // Copyright 2013 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.
 
 
 /**
  * @fileoverview Tests the local NTP.
  */
 
 
@@ skipping 121 matching lines @@
   assert(!window.chrome.embeddedSearch.newTabPage.getMostVisitedItemData(
       window.chrome.embeddedSearch.newTabPage.mostVisited[0].rid));
 }
 
 
 
 // ****************************** ADVANCED TESTS ******************************
 // Advanced tests are controlled from the native side. The helpers here are
 // called from native code to set up the page and to check results.
 
-function setupAdvancedTest() {
+function handlePostMessage(event) {
+  if (event.data.cmd == 'loaded') {
+    domAutomationController.setAutomationId(0);
+    domAutomationController.send('loaded');
+  }
+}
+
+function setupAdvancedTest(opt_waitForIframeLoaded) {
+  if (opt_waitForIframeLoaded) {
+    window.addEventListener('message', handlePostMessage);
+  }
+
   setUp();
   initLocalNTP(/*isGooglePage=*/true);
 
   assert(elementIsVisible($('fakebox')));
 
   return true;
 }
 
 function getFakeboxPositionX() {
   assert(elementIsVisible($('fakebox')));
   var rect = $('fakebox').getBoundingClientRect();
   return rect.left;
 }
 
 function getFakeboxPositionY() {
   assert(elementIsVisible($('fakebox')));
   var rect = $('fakebox').getBoundingClientRect();
   return rect.top;
 }
 
 function fakeboxIsVisible() {
   return elementIsVisible($('fakebox'));
 }
 
 function fakeboxIsFocused() {
   return fakeboxIsVisible() &&
       document.body.classList.contains('fakebox-focused');
 }