netwerk/test/unit/test_bug248970_cookie.js

   1 /* ***** BEGIN LICENSE BLOCK *****
   2  * Version: MPL 1.1/GPL 2.0/LGPL 2.1
   3  *
   4  * The contents of this file are subject to the Mozilla Public License Version
   5  * 1.1 (the "License"); you may not use this file except in compliance with
   6  * the License. You may obtain a copy of the License at
   7  * http://www.mozilla.org/MPL/
   8  *
   9  * Software distributed under the License is distributed on an "AS IS" basis,
  10  * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
  11  * for the specific language governing rights and limitations under the
  12  * License.
  13  *
  14  * The Original Code is Private Browsing Tests.
  15  *
  16  * The Initial Developer of the Original Code is
  17  * Ehsan Akhgari.
  18  * Portions created by the Initial Developer are Copyright (C) 2008
  19  * the Initial Developer. All Rights Reserved.
  20  *
  21  * Contributor(s):
  22  *   Ehsan Akhgari <ehsan.akhgari@gmail.com> (Original Author)
  23  *
  24  * Alternatively, the contents of this file may be used under the terms of
  25  * either of the GNU General Public License Version 2 or later (the "GPL"),
  26  * or the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
  27  * in which case the provisions of the GPL or the LGPL are applicable instead
  28  * of those above. If you wish to allow use of your version of this file only
  29  * under the terms of either the GPL or the LGPL, and not to allow others to
  30  * use your version of this file under the terms of the MPL, indicate your
  31  * decision by deleting the provisions above and replace them with the notice
  32  * and other provisions required by the GPL or the LGPL. If you do not delete
  33  * the provisions above, a recipient may use your version of this file under
  34  * the terms of any one of the MPL, the GPL or the LGPL.
  35  *
  36  * ***** END LICENSE BLOCK ***** */
  37
  38 var _PBSvc = null;
  39 function get_PBSvc() {
  40   if (_PBSvc)
  41     return _PBSvc;
  42
  43   try {
  44     _PBSvc = Components.classes["@mozilla.org/privatebrowsing;1"].
  45              getService(Components.interfaces.nsIPrivateBrowsingService);
  46     return _PBSvc;
  47   } catch (e) {}
  48   return null;
  49 }
  50
  51 var _CMSvc = null;
  52 function get_CookieManager() {
  53   if (_CMSvc)
  54     return _CMSvc;
  55
  56   return _CMSvc = Components.classes["@mozilla.org/cookiemanager;1"].
  57                   getService(Components.interfaces.nsICookieManager2);
  58 }
  59
  60 function is_cookie_available1(domain, path, name, value,
  61                               secure, httponly, session, expires) {
  62   var cm = get_CookieManager();
  63   var enumerator = cm.enumerator;
  64   while (enumerator.hasMoreElements()) {
  65     var cookie = enumerator.getNext().QueryInterface(Components.interfaces.nsICookie);
  66     if (cookie.host == domain &&
  67         cookie.path == path &&
  68         cookie.name == name &&
  69         cookie.value == value &&
  70         cookie.isSecure == secure &&
  71         cookie.expires == expires)
  72       return true;
  73   }
  74   return false;
  75 }
  76
  77 function is_cookie_available2(domain, path, name, value,
  78                               secure, httponly, session, expires) {
  79   var cookie = {
  80     name: name,
  81     value: value,
  82     isDomain: true,
  83     host: domain,
  84     path: path,
  85     isSecure: secure,
  86     expires: expires,
  87     status: 0,
  88     policy: 0,
  89     isSession: session,
  90     expiry: expires,
  91     isHttpOnly: httponly,
  92     QueryInterface: function(iid) {
  93       var validIIDs = [Components.interfaces.nsISupports,
  94                        Components.interfaces.nsICookie,
  95                        Components.interfaces.nsICookie2];
  96       for (var i = 0; i < validIIDs.length; ++i) {
  97         if (iid == validIIDs[i])
  98           return this;
  99       }
 100       throw Components.results.NS_ERROR_NO_INTERFACE;
 101     }
 102   };
 103
 104   var cm = get_CookieManager();
 105   return cm.cookieExists(cookie);
 106 }
 107
 108 var cc_observer = null;
 109 function setup_cookie_changed_observer() {
 110   cc_observer = {
 111     gotReloaded: false,
 112     QueryInterface: function (iid) {
 113       const interfaces = [Components.interfaces.nsIObserver,
 114                           Components.interfaces.nsISupports];
 115       if (!interfaces.some(function(v) iid.equals(v)))
 116         throw Components.results.NS_ERROR_NO_INTERFACE;
 117       return this;
 118     },
 119     observe: function (subject, topic, data) {
 120       if (topic == "cookie-changed") {
 121         if (!subject) {
 122           if (data == "reload")
 123             this.gotReloaded = true;
 124         }
 125       }
 126     }
 127   };
 128   var os = Components.classes["@mozilla.org/observer-service;1"].
 129            getService(Components.interfaces.nsIObserverService);
 130   os.addObserver(cc_observer, "cookie-changed", false);
 131 }
 132
 133 function run_test() {
 134   var pb = get_PBSvc();
 135   if (pb) { // Private Browsing might not be available
 136     var prefBranch = Components.classes["@mozilla.org/preferences-service;1"].
 137                      getService(Components.interfaces.nsIPrefBranch);
 138     prefBranch.setBoolPref("browser.privatebrowsing.keep_current_session", true);
 139
 140     var cm = get_CookieManager();
 141     do_check_neq(cm, null);
 142
 143     setup_cookie_changed_observer();
 144     do_check_neq(cc_observer, null);
 145
 146     try {
 147       // create Cookie-A
 148       const time = (new Date("Jan 1, 2030")).getTime() / 1000;
 149       cm.add("pbtest.example.com", "/", "C1", "V1", false, true, false, time);
 150       // make sure Cookie-A is retrievable
 151       do_check_true(is_cookie_available1("pbtest.example.com", "/", "C1", "V1", false, true, false, time));
 152       do_check_true(is_cookie_available2("pbtest.example.com", "/", "C1", "V1", false, true, false, time));
 153       // enter private browsing mode
 154       pb.privateBrowsingEnabled = true;
 155       // make sure the "cleared" notification was fired
 156       do_check_true(cc_observer.gotReloaded);
 157       cc_observer.gotReloaded = false;
 158       // make sure Cookie-A is not retrievable
 159       do_check_false(is_cookie_available1("pbtest.example.com", "/", "C1", "V1", false, true, false, time));
 160       do_check_false(is_cookie_available2("pbtest.example.com", "/", "C1", "V1", false, true, false, time));
 161       // create Cookie-B
 162       const time2 = (new Date("Jan 2, 2030")).getTime() / 1000;
 163       cm.add("pbtest2.example.com", "/", "C2", "V2", false, true, false, time2);
 164       // make sure Cookie-B is retrievable
 165       do_check_true(is_cookie_available1("pbtest2.example.com", "/", "C2", "V2", false, true, false, time2));
 166       do_check_true(is_cookie_available2("pbtest2.example.com", "/", "C2", "V2", false, true, false, time2));
 167       // exit private browsing mode
 168       pb.privateBrowsingEnabled = false;
 169       // make sure the "reload" notification was fired
 170       do_check_true(cc_observer.gotReloaded);
 171       // make sure Cookie-B is not retrievable
 172       do_check_false(is_cookie_available1("pbtest2.example.com", "/", "C2", "V2", false, true, false, time2));
 173       do_check_false(is_cookie_available2("pbtest2.example.com", "/", "C2", "V2", false, true, false, time2));
 174       // make sure Cookie-A is retrievable
 175       do_check_true(is_cookie_available1("pbtest.example.com", "/", "C1", "V1", false, true, false, time));
 176       do_check_true(is_cookie_available2("pbtest.example.com", "/", "C1", "V1", false, true, false, time));
 177     } catch (e) {
 178       do_throw("Unexpected exception while testing cookies: " + e);
 179     }
 180
 181     prefBranch.clearUserPref("browser.privatebrowsing.keep_current_session");
 182   }
 183 }