summaryrefslogtreecommitdiffstats
path: root/dom/tests/mochitest/beacon/test_beaconContentPolicy.html
blob: 07f1080681771bfe6cffa56d5cc6c03fada343a9 (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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
<!DOCTYPE HTML>
<html>
<!--
https://bugzilla.mozilla.org/show_bug.cgi?id=936340
-->
<head>
  <title>Test that sendBeacon obeys content policy directives</title>
  <script src="/tests/SimpleTest/SimpleTest.js"></script>
  <link rel="stylesheet" type="text/css" href="/tests/SimpleTest/test.css" />
</head>
<body>
<a target="_blank" href="https://bugzilla.mozilla.org/show_bug.cgi?id=936340">Mozilla Bug 936340</a>
<p id="display"></p>
<div id="content" style="display: none">
  
</div>
<pre id="test">
<script class="testbody" type="text/javascript">

var beaconUrl = "http://mochi.test:8888/tests/dom/tests/mochitest/beacon/beacon-handler.sjs";

const Cc = SpecialPowers.Cc;
const Ci = SpecialPowers.Ci;

// not enabled by default yet.
SimpleTest.waitForExplicitFinish();

var policy;

SpecialPowers.pushPrefEnv({'set': [["beacon.enabled", true]]}, beginTest);

function setupPolicy() {
  var policyID = SpecialPowers.wrap(SpecialPowers.Components).ID("{b80e19d0-878f-d41b-2654-194714a4115c}");
  var policyName = "@mozilla.org/testpolicy;1";
  var policy = {
    // nsISupports implementation
    QueryInterface(iid) {
      iid = SpecialPowers.wrap(iid);
      if (iid.equals(Ci.nsISupports) ||
        iid.equals(Ci.nsIFactory) ||
        iid.equals(Ci.nsIContentPolicy))
        return this;
      throw SpecialPowers.Cr.NS_ERROR_NO_INTERFACE;
    },

    // nsIFactory implementation
    createInstance(iid) {
      return this.QueryInterface(iid);
    },

    // nsIContentPolicy implementation
    shouldLoad(contentLocation, loadInfo) {
      // Remember last content type seen for the test url
      let contentType = loadInfo.externalContentPolicyType;

      if (SpecialPowers.wrap(contentLocation).spec == beaconUrl) {
        is(contentType,  Ci.nsIContentPolicy.TYPE_BEACON, "Beacon content type should match expected.  is: " + contentType + " should be: " + Ci.nsIContentPolicy.TYPE_BEACON);
        teardownPolicy();
        SimpleTest.finish();
      }

      return Ci.nsIContentPolicy.ACCEPT;
    },

    shouldProcess(contentLocation, loadInfo) {
      return Ci.nsIContentPolicy.ACCEPT;
    }
  }
  policy = SpecialPowers.wrapCallbackObject(policy);

  // Register content policy
  var componentManager = SpecialPowers.wrap(SpecialPowers.Components).manager.QueryInterface(Ci.nsIComponentRegistrar);
  componentManager.registerFactory(policyID, "Test content policy", policyName, policy);

  var categoryManager = Cc["@mozilla.org/categorymanager;1"].getService(Ci.nsICategoryManager);
  categoryManager.addCategoryEntry("content-policy", policyName, policyName, false, true);

  return { 'policy': policy, 'policyID': policyID, 'policyName': policyName };
}

function teardownPolicy() {
  setTimeout(function() {
    // policy will not be removed from the category correctly
    var componentManager = SpecialPowers.wrap(SpecialPowers.Components).manager.QueryInterface(Ci.nsIComponentRegistrar);
    componentManager.unregisterFactory(policy.policyID, policy.policy);
    var categoryManager = Cc["@mozilla.org/categorymanager;1"].getService(Ci.nsICategoryManager);
    categoryManager.deleteCategoryEntry("content-policy", policy.policyName, false);
  }, 0);
}

function beginTest() {
  policy = setupPolicy();
  // Make sure to hit the event loop here in order to ensure that nsContentPolicy
  // has been notified of the newly registered policy.
  SimpleTest.executeSoon(function() {
    navigator.sendBeacon(beaconUrl, "bacon would have been a better name than beacon");
  });
}

</script>
</pre>
</body>
</html>