Mypal68/browser/base/content/test/plugins/browser_bug820497.js

var gTestRoot = getRootDirectory(gTestPath).replace(
  "chrome://mochitests/content/",
  "http://127.0.0.1:8888/"
);
var gTestBrowser = null;
var gNumPluginBindingsAttached = 0;

add_task(async function() {
  registerCleanupFunction(function() {
    clearAllPluginPermissions();
    Services.prefs.clearUserPref("plugins.click_to_play");
    setTestPluginEnabledState(Ci.nsIPluginTag.STATE_ENABLED, "Test Plug-in");
    setTestPluginEnabledState(
      Ci.nsIPluginTag.STATE_ENABLED,
      "Second Test Plug-in"
    );
    gBrowser.removeCurrentTab();
    window.focus();
    gTestBrowser = null;
  });
});

add_task(async function() {
  Services.prefs.setBoolPref("plugins.click_to_play", true);

  gBrowser.selectedTab = BrowserTestUtils.addTab(gBrowser);
  gTestBrowser = gBrowser.selectedBrowser;

  setTestPluginEnabledState(Ci.nsIPluginTag.STATE_CLICKTOPLAY, "Test Plug-in");
  setTestPluginEnabledState(
    Ci.nsIPluginTag.STATE_CLICKTOPLAY,
    "Second Test Plug-in"
  );

  BrowserTestUtils.addContentEventListener(
    gTestBrowser,
    "PluginBindingAttached",
    function() {
      gNumPluginBindingsAttached++;
    },
    true,
    null,
    true
  );

  await promiseTabLoadEvent(
    gBrowser.selectedTab,
    gTestRoot + "plugin_bug820497.html"
  );

  await promiseForCondition(function() {
    return gNumPluginBindingsAttached == 1;
  });

  await ContentTask.spawn(gTestBrowser, null, () => {
    // Note we add the second plugin in the code farther down, so there's
    // no way we got here with anything but one plugin loaded.
    let doc = content.document;
    let testplugin = doc.getElementById("test");
    ok(testplugin, "should have test plugin");
    let secondtestplugin = doc.getElementById("secondtest");
    ok(!secondtestplugin, "should not yet have second test plugin");
  });

  await promisePopupNotification("click-to-play-plugins");
  let notification = PopupNotifications.getNotification(
    "click-to-play-plugins",
    gTestBrowser
  );
  ok(notification, "should have a click-to-play notification");

  await promiseForNotificationShown(notification);

  is(
    notification.options.pluginData.size,
    1,
    "should be 1 type of plugin in the popup notification"
  );

  await ContentTask.spawn(gTestBrowser, {}, async function() {
    XPCNativeWrapper.unwrap(content).addSecondPlugin();
  });

  await promiseForCondition(function() {
    return gNumPluginBindingsAttached == 2;
  });

  await ContentTask.spawn(gTestBrowser, null, () => {
    let doc = content.document;
    let testplugin = doc.getElementById("test");
    ok(testplugin, "should have test plugin");
    let secondtestplugin = doc.getElementById("secondtest");
    ok(secondtestplugin, "should have second test plugin");
  });

  notification = PopupNotifications.getNotification(
    "click-to-play-plugins",
    gTestBrowser
  );

  ok(notification, "should have popup notification");

  await promiseForNotificationShown(notification);

  is(
    notification.options.pluginData.size,
    2,
    "aited too long for 2 types of plugins in popup notification"
  );
});