Deal with frame removal by content scripts

chrome/test/data/extensions/api_test/executescript/destructive/remove_self.js

+// Copyright 2016 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.
+
+if (top === window) {
+  testTop();
+} else {
+  testChild();
+}
+
+function testTop() {
+  var testMessage = {};
+  function reportFrames() {
+    if (reportFrames.didRun)
+      return;
+    reportFrames.didRun = true;
+    testMessage.frameCount = frames.length;
+    testMessage.frameHTML = window.frameHTML;
+    chrome.runtime.sendMessage(testMessage);
+  }
+  window.onmessage = function(event) {
+    if (event.data === 'toBeRemoved')
+      reportFrames();
+  };
+  window.onload = function() {
+    // Fall back, in case the content script never ran.
+    setTimeout(reportFrames, 3000);
+  };
+
+  if (window.frameHTML) {  // Set by child frame...
+    // about:blank frames are synchronously parsed, so their document_end script
+    // injection happens before the main frame's injection.
+    var expectChildBeforeMain = location.search.includes('?blankend');
+    if (!expectChildBeforeMain) {
+      // Add a message to the test notification to cause the test to fail,
+      // with some useful information for diagnostics.
+      testMessage.warning = 'Content script in child frame was executed ' +
+        'before the main frame\'s content script!';
+    }
+    reportFrames();
+  }
+  window.frameHTML = '(not set)';
+}
+
+function testChild() {
+  var TEST_HOST = parent.location.hostname;
+
+  if (TEST_HOST === 'synchronous') {
+    doRemove();
+  } else if (TEST_HOST === 'microtask') {
+    Promise.resolve().then(doRemove);
+  } else if (TEST_HOST === 'macrotask') {
+    setTimeout(doRemove, 0);
+  } else if (TEST_HOST.startsWith('domnodeinserted')) {
+    removeOnEvent('DOMNodeInserted');
+  } else if (TEST_HOST.startsWith('domsubtreemodified')) {
+    removeOnEvent('DOMSubtreeModified');
+  } else {
+    console.error('Unexpected test: ' + TEST_HOST);
+    chrome.test.fail();
+  }
+
+  function doRemove() {
+    parent.frameHTML = document.documentElement.outerHTML || '(no outerHTML)';
+    parent.postMessage('toBeRemoved', '*');
+    // frameElement = <iframe> element in parent document.
+    frameElement.remove();
+  }
+
+  function removeOnEvent(eventName) {
+    // For document_start scripts, DOM mutation events are likely, so pick a
+    // generous time-out delay to make sure that all parser-initiated DOM
+    // mutations have been processed.
+    // DOM mutation events are not really expected in document_end scripts, so
+    // use a very small time-out delay.
+    var delay = parent.location.search.includes('?start') ? 1000 : 50;
+    var expected = parseInt(TEST_HOST.match(/\d+/)[0]);
+    // Fallback in case the mutation events are not triggered.
+    setTimeout(function() {
+      if (expected > 0) {
+        expected = 0;
+        doRemove();
+      }
+    }, delay);
+    document.addEventListener(eventName, function() {
+      if (--expected === 0)
+        doRemove();
+    });
+  }
+}