The Tor Browser: extensions/cookie/test/unit/test_cookies_read.js@6474c204b198 (annotated)

extensions/cookie/test/unit/test_cookies_read.js@6474c204b198 (annotated)

extensions/cookie/test/unit/test_cookies_read.js

Wed, 31 Dec 2014 06:09:35 +0100

author: Michael Schloh von Bennewitz <michael@schloh.com>
date: Wed, 31 Dec 2014 06:09:35 +0100
changeset 0: 6474c204b198
permissions: -rw-r--r--

Cloned upstream origin tor-browser at tor-browser-31.3.0esr-4.5-1-build1
revision ID fc1c9ff7c1b2defdbc039f12214767608f46423f for hacking purpose.

 /* Any copyright is dedicated to the Public Domain.
    http://creativecommons.org/publicdomain/zero/1.0/ */
 // test cookie database asynchronous read operation.
 let test_generator = do_run_test();
 let CMAX = 1000;    // # of cookies to create
 function run_test() {
   do_test_pending();
   test_generator.next();
 }
 function finish_test() {
   do_execute_soon(function() {
     test_generator.close();
     do_test_finished();
   });
 }
 function do_run_test() {
   // Set up a profile.
   let profile = do_get_profile();
   // Allow all cookies.
   Services.prefs.setIntPref("network.cookie.cookieBehavior", 0);
   // Start the cookieservice, to force creation of a database.
   Services.cookies;
   // Open a database connection now, after synchronous initialization has
   // completed. We may not be able to open one later once asynchronous writing
   // begins.
   do_check_true(do_get_cookie_file(profile).exists());
   let db = new CookieDatabaseConnection(do_get_cookie_file(profile), 4);
   for (let i = 0; i < CMAX; ++i) {
     let uri = NetUtil.newURI("http://" + i + ".com/");
     Services.cookies.setCookieString(uri, null, "oh=hai; max-age=1000", null);
   }
   do_check_eq(do_count_cookies(), CMAX);
   // Wait until all CMAX cookies have been written out to the database.
   while (do_count_cookies_in_db(db.db) < CMAX) {
     do_execute_soon(function() {
       do_run_generator(test_generator);
     });
     yield;
   }
   // Check the WAL file size. We set it to 16 pages of 32k, which means it
   // should be around 500k.
   let file = db.db.databaseFile;
   do_check_true(file.exists());
   do_check_true(file.fileSize < 1e6);
   db.close();
   // fake a profile change
   do_close_profile(test_generator);
   yield;
   do_load_profile();
   // test a few random cookies
   do_check_eq(Services.cookiemgr.countCookiesFromHost("999.com"), 1);
   do_check_eq(Services.cookiemgr.countCookiesFromHost("abc.com"), 0);
   do_check_eq(Services.cookiemgr.countCookiesFromHost("100.com"), 1);
   do_check_eq(Services.cookiemgr.countCookiesFromHost("400.com"), 1);
   do_check_eq(Services.cookiemgr.countCookiesFromHost("xyz.com"), 0);
   // force synchronous load of everything
   do_check_eq(do_count_cookies(), CMAX);
   // check that everything's precisely correct
   for (let i = 0; i < CMAX; ++i) {
     let host = i.toString() + ".com";
     do_check_eq(Services.cookiemgr.countCookiesFromHost(host), 1);
   }
   // reload again, to make sure the additions were written correctly
   do_close_profile(test_generator);
   yield;
   do_load_profile();
   // remove some of the cookies, in both reverse and forward order
   for (let i = 100; i-- > 0; ) {
     let host = i.toString() + ".com";
     Services.cookiemgr.remove(host, "oh", "/", false);
   }
   for (let i = CMAX - 100; i < CMAX; ++i) {
     let host = i.toString() + ".com";
     Services.cookiemgr.remove(host, "oh", "/", false);
   }
   // check the count
   do_check_eq(do_count_cookies(), CMAX - 200);
   // reload again, to make sure the removals were written correctly
   do_close_profile(test_generator);
   yield;
   do_load_profile();
   // check the count
   do_check_eq(do_count_cookies(), CMAX - 200);
   // reload again, but wait for async read completion
   do_close_profile(test_generator);
   yield;
   do_load_profile(test_generator);
   yield;
   // check that everything's precisely correct
   do_check_eq(do_count_cookies(), CMAX - 200);
   for (let i = 100; i < CMAX - 100; ++i) {
     let host = i.toString() + ".com";
     do_check_eq(Services.cookiemgr.countCookiesFromHost(host), 1);
   }
   finish_test();
 }