In extension messaging, use the sender frame's URL as the sender URL rather

chrome/test/data/extensions/api_test/messaging/externally_connectable/sites/assertions.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.
 
-window.assertions = (function() {
+(function() {
 
 // We are going to kill all of the builtins, so hold onto the ones we need.
 var defineGetter = Object.prototype.__defineGetter__;
 var defineSetter = Object.prototype.__defineSetter__;
 var Error = window.Error;
 var forEach = Array.prototype.forEach;
 var hasOwnProperty = Object.prototype.hasOwnProperty;
 var getOwnPropertyNames = Object.getOwnPropertyNames;
 var stringify = JSON.stringify;
 
@@ skipping 52 matching lines @@
   FUNCTION_NOT_DEFINED: 2,
   COULD_NOT_ESTABLISH_CONNECTION_ERROR: 3,
   OTHER_ERROR: 4,
   INCORRECT_RESPONSE_SENDER: 5,
   INCORRECT_RESPONSE_MESSAGE: 6
 };
 
 // Make the messages sent vaguely complex, but unambiguously JSON-ifiable.
 var message = [{'a': {'b': 10}}, 20, 'c\x10\x11'];
 
+// Our tab's location. Normally this would be our document's location but if
+// we're an iframe it will be the location of the parent - in which case,
+// expect to be told.
+var tabLocationHref = null;
+
+if (parent == window) {
+  tabLocationHref = document.location.href;
+} else {
+  window.addEventListener('message', function listener(event) {
+    window.removeEventListener(listener);
+    tabLocationHref = event.data;
+  });
+}
+
 function checkLastError(reply) {
   if (!chrome.runtime.lastError)
     return true;
   var kCouldNotEstablishConnection =
     'Could not establish connection. Receiving end does not exist.';
   if (chrome.runtime.lastError.message == kCouldNotEstablishConnection)
     reply(results.COULD_NOT_ESTABLISH_CONNECTION_ERROR);
   else
     reply(results.OTHER_ERROR);
   return false;
 }
 
 function checkResponse(response, reply) {
   // The response will be an echo of both the original message *and* the
   // MessageSender (with the tab field stripped down).
   //
   // First check the sender was correct.
   var incorrectSender = false;
   if (!hasOwnProperty.call(response.sender, 'tab')) {
     console.warn('Expected a tab, got none');
     incorrectSender = true;
   }
-  if (response.sender.tab.url != document.location.href) {
-    console.warn('Expected tab url ' + document.location.href + ' got ' +
+  if (response.sender.tab.url != tabLocationHref) {
+    console.warn('Expected tab url ' + tabLocationHref + ' got ' +
                  response.sender.tab.url);
     incorrectSender = true;
   }
   if (hasOwnProperty.call(response.sender, 'id')) {
     console.warn('Expected no id, got "' + response.sender.id + '"');
     incorrectSender = true;
   }
   if (response.sender.url != document.location.href) {
     console.warn('Expected url ' + document.location.href + ' got ' +
                  response.sender.url);
@@ skipping 11 matching lines @@
     return true;
   console.warn('Expected message ' + expectedJson + ' got ' + actualJson);
   reply(results.INCORRECT_RESPONSE_MESSAGE);
   return false;
 }
 
 function sendToBrowser(msg) {
   domAutomationController.send(msg);
 }
 
-return {
+window.actions = {
+  appendIframe: function(src) {
+    var iframe = document.createElement('iframe');
+    // When iframe has loaded, notify it of our tab location (probably
+    // document.location) to use in its assertions, then continue.
+    iframe.addEventListener('load', function listener() {
+      iframe.removeEventListener('load', listener);
+      iframe.contentWindow.postMessage(tabLocationHref, '*');
+      sendToBrowser(true);
+    });
+    iframe.src = src;
+    document.body.appendChild(iframe);
+  }
+};
+
+window.assertions = {
   canConnectAndSendMessages: function(extensionId) {
     if (!chrome.runtime) {
       sendToBrowser(results.NAMESPACE_NOT_DEFINED);
       return;
     }
 
     if (!chrome.runtime.sendMessage || !chrome.runtime.connect) {
       sendToBrowser(results.FUNCTION_NOT_DEFINED);
       return;
     }
@@ skipping 38 matching lines @@
         if (chrome.runtime[name]) {
           console.log('runtime.' + name + ' is defined');
           result = true;
         }
       });
     }
     sendToBrowser(result);
   }
 };
 
-}());  // window.assertions
+}());