michael@0: var rootDir = getRootDirectory(gTestPath);
michael@0: const gTestRoot = rootDir;
michael@0: const gHttpTestRoot = rootDir.replace("chrome://mochitests/content/", "http://127.0.0.1:8888/");
michael@0: 
michael@0: var gTestBrowser = null;
michael@0: var gNextTest = null;
michael@0: var gPluginHost = Components.classes["@mozilla.org/plugin/host;1"].getService(Components.interfaces.nsIPluginHost);
michael@0: var gRunNextTestAfterPluginRemoved = false;
michael@0: 
michael@0: Components.utils.import("resource://gre/modules/Services.jsm");
michael@0: 
michael@0: function test() {
michael@0:   waitForExplicitFinish();
michael@0:   registerCleanupFunction(function() {
michael@0:     clearAllPluginPermissions();
michael@0:     Services.prefs.clearUserPref("extensions.blocklist.suppressUI");
michael@0:     gTestBrowser.removeEventListener("PluginRemoved", handlePluginRemoved, true, true);
michael@0:   });
michael@0:   Services.prefs.setBoolPref("extensions.blocklist.suppressUI", true);
michael@0: 
michael@0:   var newTab = gBrowser.addTab();
michael@0:   gBrowser.selectedTab = newTab;
michael@0:   gTestBrowser = gBrowser.selectedBrowser;
michael@0:   gTestBrowser.addEventListener("load", pageLoad, true);
michael@0:   gTestBrowser.addEventListener("PluginRemoved", handlePluginRemoved, true, true);
michael@0: 
michael@0:   Services.prefs.setBoolPref("plugins.click_to_play", true);
michael@0:   setTestPluginEnabledState(Ci.nsIPluginTag.STATE_CLICKTOPLAY);
michael@0: 
michael@0:   prepareTest(runAfterPluginBindingAttached(test1), gHttpTestRoot + "plugin_two_types.html");
michael@0: }
michael@0: 
michael@0: function finishTest() {
michael@0:   clearAllPluginPermissions();
michael@0:   gTestBrowser.removeEventListener("load", pageLoad, true);
michael@0:   gBrowser.removeCurrentTab();
michael@0:   window.focus();
michael@0:   finish();
michael@0: }
michael@0: 
michael@0: function pageLoad() {
michael@0:   // The plugin events are async dispatched and can come after the load event
michael@0:   // This just allows the events to fire before we then go on to test the states
michael@0:   executeSoon(gNextTest);
michael@0: }
michael@0: 
michael@0: function prepareTest(nextTest, url) {
michael@0:   gNextTest = nextTest;
michael@0:   gTestBrowser.contentWindow.location = url;
michael@0: }
michael@0: 
michael@0: // Due to layout being async, "PluginBindAttached" may trigger later.
michael@0: // This wraps a function to force a layout flush, thus triggering it,
michael@0: // and schedules the function execution so they're definitely executed
michael@0: // afterwards.
michael@0: function runAfterPluginBindingAttached(func) {
michael@0:   return function() {
michael@0:     let doc = gTestBrowser.contentDocument;
michael@0:     let elems = doc.getElementsByTagName('embed');
michael@0:     if (elems.length < 1) {
michael@0:       elems = doc.getElementsByTagName('object');
michael@0:     }
michael@0:     elems[0].clientTop;
michael@0:     executeSoon(func);
michael@0:   };
michael@0: }
michael@0: 
michael@0: function handlePluginRemoved() {
michael@0:   if (gRunNextTestAfterPluginRemoved) {
michael@0:     executeSoon(gNextTest);
michael@0:     gRunNextTestAfterPluginRemoved = false;
michael@0:   }
michael@0: }
michael@0: 
michael@0: function runAfterPluginRemoved(func) {
michael@0:   gNextTest = func;
michael@0:   gRunNextTestAfterPluginRemoved = true;
michael@0: }
michael@0: 
michael@0: // Test that the click-to-play notification is not shown for non-plugin object elements
michael@0: 
michael@0: function test1() {
michael@0:   let popupNotification = PopupNotifications.getNotification("click-to-play-plugins", gTestBrowser);
michael@0:   ok(popupNotification, "Test 1, Should have a click-to-play notification");
michael@0: 
michael@0:   let plugin = gTestBrowser.contentDocument.getElementById("secondtestA");
michael@0:   plugin.parentNode.removeChild(plugin);
michael@0:   plugin = gTestBrowser.contentDocument.getElementById("secondtestB");
michael@0:   plugin.parentNode.removeChild(plugin);
michael@0: 
michael@0:   let image = gTestBrowser.contentDocument.createElement("object");
michael@0:   image.type = "image/png";
michael@0:   image.data = "moz.png";
michael@0:   gTestBrowser.contentDocument.body.appendChild(image);
michael@0: 
michael@0:   runAfterPluginRemoved(test2);
michael@0: }
michael@0: 
michael@0: function test2() {
michael@0:   let popupNotification = PopupNotifications.getNotification("click-to-play-plugins", gTestBrowser);
michael@0:   ok(popupNotification, "Test 2, Should have a click-to-play notification");
michael@0: 
michael@0:   let plugin = gTestBrowser.contentDocument.getElementById("test");
michael@0:   plugin.parentNode.removeChild(plugin);
michael@0: 
michael@0:   executeSoon(test3);
michael@0: }
michael@0: 
michael@0: function test3() {
michael@0:   let popupNotification = PopupNotifications.getNotification("click-to-play-plugins", gTestBrowser);
michael@0:   ok(popupNotification, "Test 3, Should still have a click-to-play notification");
michael@0: 
michael@0:   finishTest();
michael@0: }