summaryrefslogtreecommitdiffstats
path: root/browser/base/content/test/newtab/browser_newtab_background_captures.js
blob: 5e838196e2032e1198ea9ef5ee588297a315f6e2 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
/* Any copyright is dedicated to the Public Domain.
   http://creativecommons.org/publicdomain/zero/1.0/ */

/**
 * Verifies that hidden, pre-loaded newtabs don't allow background captures, and
 * when unhidden, do allow background captures.
 */

const CAPTURE_PREF = "browser.pagethumbnails.capturing_disabled";

add_task(function* () {
  let imports = {};
  Cu.import("resource://gre/modules/PageThumbs.jsm", imports);

  // Disable captures.
  yield pushPrefs([CAPTURE_PREF, false]);

  // Make sure the thumbnail doesn't exist yet.
  let url = "http://example.com/";
  let path = imports.PageThumbsStorage.getFilePathForURL(url);
  let file = Cc["@mozilla.org/file/local;1"].createInstance(Ci.nsIFile);
  file.initWithPath(path);
  try {
    file.remove(false);
  }
  catch (err) {}

  // Add a top site.
  yield setLinks("-1");

  // We need a handle to a hidden, pre-loaded newtab so we can verify that it
  // doesn't allow background captures. Ensure we have a preloaded browser.
  gBrowser._createPreloadBrowser();

  // Wait for the preloaded browser to load.
  if (gBrowser._preloadedBrowser.contentDocument.readyState != "complete") {
    yield BrowserTestUtils.waitForEvent(gBrowser._preloadedBrowser, "load", true);
  }

  // We're now ready to use the preloaded browser.
  BrowserOpenTab();
  let tab = gBrowser.selectedTab;

  let thumbnailCreatedPromise = new Promise(resolve => {
    // Showing the preloaded tab should trigger thumbnail capture.
    Services.obs.addObserver(function onCreate(subj, topic, data) {
      if (data != url)
        return;
      Services.obs.removeObserver(onCreate, "page-thumbnail:create");
      ok(true, "thumbnail created after preloaded tab was shown");

      resolve();
    }, "page-thumbnail:create", false);
  });

  // Enable captures.
  yield pushPrefs([CAPTURE_PREF, false]);

  yield thumbnailCreatedPromise;

  // Test finished!
  gBrowser.removeTab(tab);
  file.remove(false);
});