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: 
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:   });
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: 
michael@0:   Services.prefs.setBoolPref("plugins.click_to_play", true);
michael@0:   setTestPluginEnabledState(Ci.nsIPluginTag.STATE_CLICKTOPLAY);
michael@0:   setTestPluginEnabledState(Ci.nsIPluginTag.STATE_CLICKTOPLAY, "Second Test Plug-in");
michael@0: 
michael@0:   prepareTest(test1a, 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: // Test that the click-to-play doorhanger for multiple plugins shows the correct
michael@0: // state when re-opening without reloads or navigation.
michael@0: 
michael@0: function test1a() {
michael@0:   let doc = gTestBrowser.contentDocument;
michael@0:   let plugin = doc.getElementById("test");
michael@0:   let objLoadingContent = plugin.QueryInterface(Ci.nsIObjectLoadingContent);
michael@0:   ok(!objLoadingContent.activated, "Test1a, Plugin should not be activated");
michael@0: 
michael@0:   let notification = PopupNotifications.getNotification("click-to-play-plugins", gTestBrowser);
michael@0:   ok(notification, "Test 1a, Should have a click-to-play notification");
michael@0:   notification.reshow();
michael@0: 
michael@0:   is(notification.options.pluginData.size, 2,
michael@0:      "Test 1a, Should have two types of plugin in the notification");
michael@0: 
michael@0:   let pluginItem = null;
michael@0:   for (let item of PopupNotifications.panel.firstChild.childNodes) {
michael@0:     is(item.value, "block", "Test 1a, all plugins should start out blocked");
michael@0:     if (item.action.pluginName == "Test") {
michael@0:       pluginItem = item;
michael@0:     }
michael@0:   }
michael@0: 
michael@0:   // Choose "Allow now" for the test plugin
michael@0:   pluginItem.value = "allownow";
michael@0:   PopupNotifications.panel.firstChild._primaryButton.click();
michael@0: 
michael@0:   waitForCondition(() => objLoadingContent.activated, test1b,
michael@0:                    "Test 1a, Waited too long for plugin to activate");
michael@0: }
michael@0: 
michael@0: function test1b() {
michael@0:   let notification = PopupNotifications.getNotification("click-to-play-plugins", gTestBrowser);
michael@0:   ok(notification, "Test 1b, Should have a click-to-play notification");
michael@0:   notification.reshow();
michael@0: 
michael@0:   let pluginItem = null;
michael@0:   for (let item of PopupNotifications.panel.firstChild.childNodes) {
michael@0:     if (item.action.pluginName == "Test") {
michael@0:       is(item.value, "allownow", "Test 1b, Test plugin should now be set to 'Allow now'");
michael@0:     } else {
michael@0:       is(item.value, "block", "Test 1b, Second Test plugin should still be blocked");
michael@0:       pluginItem = item;
michael@0:     }
michael@0:   }
michael@0: 
michael@0:   // Choose "Allow and remember" for the Second Test plugin
michael@0:   pluginItem.value = "allowalways";
michael@0:   PopupNotifications.panel.firstChild._primaryButton.click();
michael@0: 
michael@0:   let doc = gTestBrowser.contentDocument;
michael@0:   let plugin = doc.getElementById("secondtestA");
michael@0:   let objLoadingContent = plugin.QueryInterface(Ci.nsIObjectLoadingContent);
michael@0:   waitForCondition(() => objLoadingContent.activated, test1c,
michael@0:                    "Test 1b, Waited too long for plugin to activate");
michael@0: }
michael@0: 
michael@0: function test1c() {
michael@0:   let notification = PopupNotifications.getNotification("click-to-play-plugins", gTestBrowser);
michael@0:   ok(notification, "Test 1c, Should have a click-to-play notification");
michael@0:   notification.reshow();
michael@0: 
michael@0:   for (let item of PopupNotifications.panel.firstChild.childNodes) {
michael@0:     if (item.action.pluginName == "Test") {
michael@0:       is(item.value, "allownow", "Test 1c, Test plugin should be set to 'Allow now'");
michael@0:     } else {
michael@0:       is(item.value, "allowalways", "Test 1c, Second Test plugin should be set to 'Allow always'");
michael@0:     }
michael@0:   }
michael@0: 
michael@0:   finishTest();
michael@0: }