Some cleanup of app and extension CSP tests.

chrome/test/data/extensions/platform_apps/iframes/main.js

 // Copyright (c) 2012 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.
 
 chrome.test.getConfig(function(config) {
   var LOCAL_URL = 'local-iframe.html';
   var DATA_URL = 'data:text/plain,This frame should be displayed.';
   var REMOTE_URL = 'http://localhost:' + config.testServer.port
       '/extensions/platform_apps/iframes/remote-iframe.html';
 
+  function onEvent(target, event_name) {
+    return new Promise((resolve, reject) => {
+      target.addEventListener(event_name, resolve);
+    });
+  }
+
+  function onCspViolation(url) {
+    return onEvent(window, 'securitypolicyviolation').then(event => {
+      chrome.test.assertEq(url, event.blockedURI);
+    });
+  }
+
   chrome.test.runTests([
     function localIframe() {
       var iframe = document.createElement('iframe');
-      iframe.onload = chrome.test.callbackPass(function() {
+      iframe.src = LOCAL_URL;
+
+      // Expect this to load normally.
+      onEvent(iframe, 'load').then(() => {
         console.log('Local iframe loaded');
+        chrome.test.succeed();
       });
-      iframe.src = LOCAL_URL;
+
       document.body.appendChild(iframe);
     },
 
     function dataUrlIframe() {
       var iframe = document.createElement('iframe');
-      iframe.onload = chrome.test.callbackPass(function() {
+      iframe.src = DATA_URL;
+
+      // Expect this to load normally.
+      onEvent(iframe, 'load').then(() => {
         console.log('data: URL iframe loaded');
+        chrome.test.succeed();
       });
-      iframe.src = DATA_URL;
+
       document.body.appendChild(iframe);
     },
 
     function filesystemUrlIframe() {
       var iframe = document.createElement('iframe');
-      iframe.onload = chrome.test.callbackPass(function() {
-        console.log('filesystem: URL iframe loaded');
-      });
+
+      // The filesystem: document should load and postMessage back.
+      Promise.all([onEvent(iframe, 'load'), onEvent(window, 'message')])
+          .then(chrome.test.succeed);
+
+      var blob = new Blob(['This filesystem: frame should be displayed',
+        '<script src="', document.origin, '/post_to_parent.js"></scri',
+        'pt>'], {type: 'text/html'});
 
       webkitRequestFileSystem(
           window.TEMPORARY,
           1024,
           function(fs) {
             fs.root.getFile(
                 'test.html',
                 {create: true, exclusive: false},
                 function(fileEntry) {
                     fileEntry.createWriter(function(fileWriter) {
                       fileWriter.onwriteend = function(e) {
                         var url = fileEntry.toURL();
                         chrome.test.assertEq(0, url.indexOf('filesystem:'));
                         iframe.src = url;
                         document.body.appendChild(iframe);
                       };
 
-                      var blob = new Blob(['This frame should be displayed'],
-                                           {type: 'text/html'});
                       fileWriter.write(blob);
                     });
                 });
           });
     },
 
     function blobUrlIframe() {
-      var blob = new Blob(['This frame should be displayed'],
-                          {type: 'text/html'});
+      var iframe = document.createElement('iframe');
+
+      // The blob: document should load and postMessage back.
+      Promise.all([onEvent(iframe, 'load'), onEvent(window, 'message')])
+          .then(chrome.test.succeed);
+
+      var blob = new Blob(['This blob: frame should be displayed',
+        '<script src="', document.origin, '/post_to_parent.js"></scri',
+        'pt>'], {type: 'text/html'});
       var blobUrl = window.URL.createObjectURL(blob);
-      var iframe = document.createElement('iframe');
-      iframe.onload = chrome.test.callbackPass(function() {
-        console.log('blob: URL iframe loaded');
-      });
+
       iframe.src = blobUrl;
       document.body.appendChild(iframe);
     },
 
     function remoteIframe() {
       var iframe = document.createElement('iframe');
       iframe.src = REMOTE_URL;
+
+      // Expect this to be blocked by the CSP.
+      onCspViolation(REMOTE_URL).then(chrome.test.succeed);
+
       document.body.appendChild(iframe);
-
-      // Load failure should happen synchronously, but wait a bit before
-      // declaring success.
-      setTimeout(chrome.test.succeed, 500);
-    }
+    },
   ]);
 });