The Tor Browser: toolkit/components/places/tests/cpp/places_test_harness

Conditionally enable double key logic according to:
private browsing mode or privacy.thirdparty.isolate preference and
implement in GetCookieStringCommon and FindCookie where it counts...
With some reservations of how to convince FindCookie users to test
condition and pass a nullptr when disabling double key logic.

     1 /* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*-

     2  * vim: sw=2 ts=2 et lcs=trail\:.,tab\:>~ :

     3  * This Source Code Form is subject to the terms of the Mozilla Public

     4  * License, v. 2.0. If a copy of the MPL was not distributed with this

     5  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

     7 #include "nsWidgetsCID.h"

     8 #include "nsIComponentRegistrar.h"

     9 #ifdef MOZ_CRASHREPORTER

    10 #include "nsICrashReporter.h"

    11 #endif

    13 #ifndef TEST_NAME

    14 #error "Must #define TEST_NAME before including places_test_harness_tail.h"

    15 #endif

    17 #ifndef TEST_FILE

    18 #error "Must #define TEST_FILE before include places_test_harness_tail.h"

    19 #endif

    21 int gTestsIndex = 0;

    23 #define TEST_INFO_STR "TEST-INFO | (%s) | "

    25 class RunNextTest : public nsRunnable

    26 {

    27 public:

    28   NS_IMETHOD Run()

    29   {

    30     NS_ASSERTION(NS_IsMainThread(), "Not running on the main thread?");

    31     if (gTestsIndex < int(mozilla::ArrayLength(gTests))) {

    32       do_test_pending();

    33       Test &test = gTests[gTestsIndex++];

    34       (void)fprintf(stderr, TEST_INFO_STR "Running %s.\n", TEST_FILE,

    35                     test.name);

    36       test.func();

    37     }

    39     do_test_finished();

    40     return NS_OK;

    41   }

    42 };

    44 void

    45 run_next_test()

    46 {

    47   nsCOMPtr<nsIRunnable> event = new RunNextTest();

    48   do_check_success(NS_DispatchToCurrentThread(event));

    49 }

    51 int gPendingTests = 0;

    53 void

    54 do_test_pending()

    55 {

    56   NS_ASSERTION(NS_IsMainThread(), "Not running on the main thread?");

    57   gPendingTests++;

    58 }

    60 void

    61 do_test_finished()

    62 {

    63   NS_ASSERTION(NS_IsMainThread(), "Not running on the main thread?");

    64   NS_ASSERTION(gPendingTests > 0, "Invalid pending test count!");

    65   gPendingTests--;

    66 }

    68 void

    69 disable_idle_service()

    70 {

    71   (void)fprintf(stderr, TEST_INFO_STR  "Disabling Idle Service.\n", TEST_FILE);

    72   static NS_DEFINE_IID(kIdleCID, NS_IDLE_SERVICE_CID);

    73   nsresult rv;

    74   nsCOMPtr<nsIFactory> idleFactory = do_GetClassObject(kIdleCID, &rv);

    75   do_check_success(rv);

    76   nsCOMPtr<nsIComponentRegistrar> registrar;

    77   rv = NS_GetComponentRegistrar(getter_AddRefs(registrar));

    78   do_check_success(rv);

    79   rv = registrar->UnregisterFactory(kIdleCID, idleFactory);

    80   do_check_success(rv);

    81 }

    83 int

    84 main(int aArgc,

    85      char** aArgv)

    86 {

    87   ScopedXPCOM xpcom(TEST_NAME);

    88   if (xpcom.failed())

    89     return -1;

    90   // Initialize a profile folder to ensure a clean shutdown.

    91   nsCOMPtr<nsIFile> profile = xpcom.GetProfileDirectory();

    92   if (!profile) {

    93     fail("Couldn't get the profile directory.");

    94     return -1;

    95   }

    97 #ifdef MOZ_CRASHREPORTER

    98     char* enabled = PR_GetEnv("MOZ_CRASHREPORTER");

    99     if (enabled && !strcmp(enabled, "1")) {

   100       // bug 787458: move this to an even-more-common location to use in all

   101       // C++ unittests

   102       nsCOMPtr<nsICrashReporter> crashreporter =

   103         do_GetService("@mozilla.org/toolkit/crash-reporter;1");

   104       if (crashreporter) {

   105         fprintf(stderr, "Setting up crash reporting\n");

   107         nsCOMPtr<nsIProperties> dirsvc =

   108           do_GetService(NS_DIRECTORY_SERVICE_CONTRACTID);

   109         if (!dirsvc)

   110           NS_RUNTIMEABORT("Couldn't get directory service");

   111         nsCOMPtr<nsIFile> cwd;

   112         nsresult rv = dirsvc->Get(NS_OS_CURRENT_WORKING_DIR,

   113                                   NS_GET_IID(nsIFile),

   114                                   getter_AddRefs(cwd));

   115         if (NS_FAILED(rv))

   116           NS_RUNTIMEABORT("Couldn't get CWD");

   117         crashreporter->SetEnabled(true);

   118         crashreporter->SetMinidumpPath(cwd);

   119       }

   120     }

   121 #endif

   123   nsRefPtr<WaitForConnectionClosed> spinClose = new WaitForConnectionClosed();

   125   // Tinderboxes are constantly on idle.  Since idle tasks can interact with

   126   // tests, causing random failures, disable the idle service.

   127   disable_idle_service();

   129   do_test_pending();

   130   run_next_test();

   132   // Spin the event loop until we've run out of tests to run.

   133   while (gPendingTests) {

   134     (void)NS_ProcessNextEvent();

   135   }

   137   // And let any other events finish before we quit.

   138   (void)NS_ProcessPendingEvents(nullptr);

   140   // Check that we have passed all of our tests, and output accordingly.

   141   if (gPassedTests == gTotalTests) {

   142     passed(TEST_FILE);

   143   }

   145   (void)fprintf(stderr, TEST_INFO_STR  "%u of %u tests passed\n",

   146                 TEST_FILE, unsigned(gPassedTests), unsigned(gTotalTests));

   148   return gPassedTests == gTotalTests ? 0 : -1;

   149 }

The Tor Browser / file revision

toolkit/components/places/tests/cpp/places_test_harness_tail.h@129ffea94266

toolkit/components/places/tests/cpp/places_test_harness_tail.h