Fix MimeHandlerView plugin loading for data: URLs

chrome/test/data/extensions/api_test/mime_handler_view/index.js

 // Copyright 2014 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.
 
 'use strict';
 
 // Tests don't start running until an async call to
 // chrome.mimeHandlerPrivate.getStreamInfo() completes, so queue any messages
 // received until that point.
 var queuedMessages = [];
@@ skipping 21 matching lines @@
     request.send();
   });
 }
 
 function expectSuccessfulRead(response) {
   chrome.test.assertEq(200, response.status);
   chrome.test.assertEq('content to read\n', response.data);
 }
 
 function checkStreamDetails(name, embedded) {
+  checkStreamDetailsNoFile();
+  chrome.test.assertEq(embedded, streamDetails.embedded);
   chrome.test.assertTrue(streamDetails.originalUrl.indexOf(name) != -1);
+  chrome.test.assertEq('text/csv',
+                       streamDetails.responseHeaders['Content-Type']);
+}
+
+function checkStreamDetailsNoFile() {
   chrome.test.assertEq('text/csv', streamDetails.mimeType);
   chrome.test.assertTrue(streamDetails.tabId != -1);
-  chrome.test.assertEq(embedded, streamDetails.embedded);
-  chrome.test.assertEq('text/csv',
-                       streamDetails.responseHeaders['Content-Type']);
 }
 
 var tests = [
   function testBasic() {
     checkStreamDetails('testBasic.csv', false);
     fetchUrl(streamDetails.streamUrl)
         .then(expectSuccessfulRead)
         .then(chrome.test.succeed);
   },
 
@@ skipping 51 matching lines @@
         messagesReceived++;
       } else {
         chrome.test.fail('unexpected message ' + event.data);
       }
     }
     window.addEventListener('message', handleMessage, false);
     while (queuedMessages.length) {
       handleMessage(queuedMessages.shift());
     }
 
+  },
+
+  function testDataUrl() {
+    // TODO(raymes): have separate checks for embedded/unembedded data URLs.
+    checkStreamDetailsNoFile();
+    fetchUrl(streamDetails.streamUrl)
+        .then(expectSuccessfulRead)
+        .then(chrome.test.succeed);
   }
 ];
 
 var testsByName = {};
 for (let i = 0; i < tests.length; i++) {
   testsByName[tests[i].name] = tests[i];
 }
 
 chrome.mimeHandlerPrivate.getStreamInfo(function(streamInfo) {
   if (!streamInfo)
     return;
 
   // If the name of the file we're handling matches the name of a test, run that
   // test.
   var urlComponents = streamInfo.originalUrl.split('/');
   var test = urlComponents[urlComponents.length - 1].split('.')[0];
   streamDetails = streamInfo;
   if (testsByName[test]) {
     window.removeEventListener('message', queueMessage);
     chrome.test.runTests([testsByName[test]]);
   }
+
+  // Run the test for data URLs.
+  if (streamInfo.originalUrl.indexOf("data:") === 0) {
+    window.removeEventListener('message', queueMessage);
+    chrome.test.runTests([testsByName['testDataUrl']]);
+  }
 });