Support for running tests in an iframe

pkg/unittest/lib/test_controller.js

 // Copyright (c) 2011, the Dart project authors.  Please see the AUTHORS file
 // for details. All rights reserved. Use of this source code is governed by a
 // BSD-style license that can be found in the LICENSE file.
 
 /**
  * Test controller logic - used by unit test harness to embed tests in
  * conent shell.
  */
 
 // Clear the console before every test run - this is Firebug specific code.
@@ skipping 15 matching lines @@
 if (navigator.webkitStartDart) {
   navigator.webkitStartDart();
 }
 
 // testRunner is provided by content shell.
 // It is not available in selenium tests.
 var testRunner = window.testRunner || window.layoutTestController;
 
 var waitForDone = false;
 
+// Returns the driving window object if available
+function getDriverWindow() {
+  if (window != window.parent) {
+    // We're running in an iframe.
+    return window.parent;
+  } else if (window.opener) {
+    // We were opened by another window.
+    return window.opener;
+  }
+  return null;
+}
+
 function notifyStart() {
-  if (window.opener) {
-    window.opener.postMessage("STARTING", "*");
+  var driver = getDriverWindow();
+  if (driver) {
+    driver.postMessage("STARTING", "*");
   }
 }
 
 function notifyDone() {
   if (testRunner) testRunner.notifyDone();
   // To support in browser launching of tests we post back start and result
   // messages to the window.opener.
-  if (window.opener) {
-    window.opener.postMessage(window.document.body.innerHTML, "*");
+  var driver = getDriverWindow();
+  if (driver) {
+    driver.postMessage(window.document.body.innerHTML, "*");
   }
 }
 
 function processMessage(msg) {
   if (typeof msg != 'string') return;
   if (msg == 'unittest-suite-done') {
     notifyDone();
   } else if (msg == 'unittest-suite-wait-for-done') {
     waitForDone = true;
     if (testRunner) testRunner.startedDartTest = true;
@@ skipping 94 matching lines @@
   try {
     main();
   } catch (e) {
     dartPrint(e);
     if (e.stack) dartPrint(e.stack);
     window.postMessage('unittest-suite-fail', '*');
     return;
   }
   window.postMessage('dart-main-done', '*');
 }