summaryrefslogtreecommitdiffstats
path: root/netwerk/test/unit/test_cookies_profile_close.js
blob: 6ea9ab23f3a5fb069a5817feaccc80de8fa46ba6 (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
104
105
106
107
108
109
110
111
112
113
114
/* Any copyright is dedicated to the Public Domain.
   http://creativecommons.org/publicdomain/zero/1.0/ */

// Test that the cookie APIs behave sanely after 'profile-before-change'.

"use strict";

add_task(async () => {
  // Set up a profile.
  do_get_profile();

  // Allow all cookies.
  Services.prefs.setIntPref("network.cookie.cookieBehavior", 0);
  Services.prefs.setBoolPref(
    "network.cookieJarSettings.unblocked_for_testing",
    true
  );
  Services.prefs.setBoolPref("dom.security.https_first", false);

  // Start the cookieservice.
  Services.cookies;

  CookieXPCShellUtils.createServer({ hosts: ["foo.com"] });

  // Set a cookie.
  let uri = NetUtil.newURI("http://foo.com");
  let channel = NetUtil.newChannel({
    uri,
    loadUsingSystemPrincipal: true,
    contentPolicyType: Ci.nsIContentPolicy.TYPE_DOCUMENT,
  });

  Services.scriptSecurityManager.createContentPrincipal(uri, {});

  await CookieXPCShellUtils.setCookieToDocument(
    uri.spec,
    "oh=hai; max-age=1000"
  );

  let cookies = Services.cookies.cookies;
  Assert.ok(cookies.length == 1);
  let cookie = cookies[0];

  // Fire 'profile-before-change'.
  do_close_profile();

  let promise = new _promise_observer("cookie-db-closed");

  // Check that the APIs behave appropriately.
  Assert.equal(
    await CookieXPCShellUtils.getCookieStringFromDocument("http://foo.com/"),
    ""
  );

  Assert.equal(Services.cookies.getCookieStringFromHttp(uri, channel), "");

  await CookieXPCShellUtils.setCookieToDocument(uri.spec, "oh2=hai");

  Services.cookies.setCookieStringFromHttp(uri, "oh3=hai", channel);
  Assert.equal(
    await CookieXPCShellUtils.getCookieStringFromDocument("http://foo.com/"),
    ""
  );

  do_check_throws(function () {
    Services.cookies.removeAll();
  }, Cr.NS_ERROR_NOT_AVAILABLE);

  do_check_throws(function () {
    Services.cookies.cookies;
  }, Cr.NS_ERROR_NOT_AVAILABLE);

  do_check_throws(function () {
    Services.cookies.add(
      "foo.com",
      "",
      "oh4",
      "hai",
      false,
      false,
      false,
      0,
      {},
      Ci.nsICookie.SAMESITE_NONE,
      Ci.nsICookie.SCHEME_HTTPS
    );
  }, Cr.NS_ERROR_NOT_AVAILABLE);

  do_check_throws(function () {
    Services.cookies.remove("foo.com", "", "oh4", {});
  }, Cr.NS_ERROR_NOT_AVAILABLE);

  do_check_throws(function () {
    Services.cookies.cookieExists(cookie.host, cookie.path, cookie.name, {});
  }, Cr.NS_ERROR_NOT_AVAILABLE);

  do_check_throws(function () {
    Services.cookies.countCookiesFromHost("foo.com");
  }, Cr.NS_ERROR_NOT_AVAILABLE);

  do_check_throws(function () {
    Services.cookies.getCookiesFromHost("foo.com", {});
  }, Cr.NS_ERROR_NOT_AVAILABLE);

  // Wait for the database to finish closing.
  await promise;

  // Load the profile and check that the API is available.
  do_load_profile();
  Assert.ok(
    Services.cookies.cookieExists(cookie.host, cookie.path, cookie.name, {})
  );
  Services.prefs.clearUserPref("dom.security.https_first");
});