summaryrefslogtreecommitdiffstats
path: root/netwerk/test/unit/test_cookies_partition_counting.js
diff options
context:
space:
mode:
authorDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-19 00:47:55 +0000
committerDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-19 00:47:55 +0000
commit26a029d407be480d791972afb5975cf62c9360a6 (patch)
treef435a8308119effd964b339f76abb83a57c29483 /netwerk/test/unit/test_cookies_partition_counting.js
parentInitial commit. (diff)
downloadfirefox-26a029d407be480d791972afb5975cf62c9360a6.tar.xz
firefox-26a029d407be480d791972afb5975cf62c9360a6.zip
Adding upstream version 124.0.1.upstream/124.0.1
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'netwerk/test/unit/test_cookies_partition_counting.js')
-rw-r--r--netwerk/test/unit/test_cookies_partition_counting.js183
1 files changed, 183 insertions, 0 deletions
diff --git a/netwerk/test/unit/test_cookies_partition_counting.js b/netwerk/test/unit/test_cookies_partition_counting.js
new file mode 100644
index 0000000000..26bd56e1f6
--- /dev/null
+++ b/netwerk/test/unit/test_cookies_partition_counting.js
@@ -0,0 +1,183 @@
+/* Any copyright is dedicated to the Public Domain.
+ * http://creativecommons.org/publicdomain/zero/1.0/ */
+
+add_setup(function test_setup() {
+ // FOG needs a profile directory to put its data in.
+ do_get_profile();
+
+ // FOG needs to be initialized in order for data to flow.
+ Services.fog.initializeFOG();
+});
+
+add_task(async function test_purge_counting() {
+ let profile = do_get_profile();
+ let dbFile = do_get_cookie_file(profile);
+ Assert.ok(!dbFile.exists());
+
+ let schema12db = new CookieDatabaseConnection(dbFile, 12);
+ let now = Date.now() * 1000; // date in microseconds
+
+ // add some non-partitioned cookies for key
+ let cookieNum1 = 1;
+ let hostNonPartitioned = "cookie-host-non-partitioned.com";
+ for (let i = 0; i < cookieNum1; i++) {
+ let cookie = new Cookie(
+ "cookie-name" + i,
+ "cookie-value" + i,
+ hostNonPartitioned,
+ "/", // path
+ now, // expiry
+ now, // needed to get the cookie by the db init
+ now, // creation time
+ false, // session
+ false, // secure
+ false, // http-only
+ false, // inBrowserElement
+ {} // OA
+ );
+ schema12db.insertCookie(cookie);
+ }
+
+ // add some non-partitioned cookies different key
+ let cookieNum2 = 10;
+ for (let i = 0; i < cookieNum2; i++) {
+ let cookie = new Cookie(
+ "cookie-name" + i,
+ "cookie-value" + i,
+ hostNonPartitioned,
+ "/", // path
+ now, // expiry
+ now, // needed to get the cookie by the db init
+ now, // creation time
+ false, // session
+ false, // secure
+ false, // http-only
+ false, // inBrowserElement
+ { userContextId: 8 } // OA
+ );
+ schema12db.insertCookie(cookie);
+ }
+
+ // add some partitioned cookies
+ let hostPartitioned = "host-partitioned.com";
+ let cookieNum3 = 3;
+ for (let i = 0; i < cookieNum3; i++) {
+ let cookie = new Cookie(
+ "cookie-name" + i,
+ "cookie-value" + i,
+ hostPartitioned,
+ "/", // path
+ now, // expiry
+ now, // needed to get the cookie by the db init
+ now, // creation time
+ false, // session
+ false, // secure
+ false, // http-only
+ false, // inBrowserElement
+ { partitionKey: "(https,example.com)" }
+ );
+ schema12db.insertCookie(cookie);
+ }
+
+ // add some partitioned with different OA
+ let cookieNum4 = 35;
+ for (let i = 0; i < cookieNum4; i++) {
+ let cookie = new Cookie(
+ "cookie-name" + i,
+ "cookie-value" + i,
+ hostPartitioned,
+ "/", // path
+ now, // expiry
+ now, // needed to get the cookie by the db init
+ now, // creation time
+ false, // session
+ false, // secure
+ false, // http-only
+ false, // inBrowserElement
+ { partitionKey: "(https,example.com)", userContextId: 7 }
+ );
+ schema12db.insertCookie(cookie);
+ }
+
+ let allCookieCount = cookieNum1 + cookieNum2 + cookieNum3 + cookieNum4;
+ Assert.equal(do_count_cookies_in_db(schema12db.db), allCookieCount);
+
+ // startup the cookie service and check the cookie counts
+ let cookieCountNonPart =
+ Services.cookies.countCookiesFromHost(hostNonPartitioned); // includes expired cookies
+ Assert.equal(cookieCountNonPart, cookieNum1);
+ let cookieCountNonPartOA = Services.cookies.getCookiesFromHost(
+ hostNonPartitioned,
+ { userContextId: 8 }
+ ).length; // includes expired cookies
+ Assert.equal(cookieCountNonPartOA, cookieNum2);
+ let cookieCountPart = Services.cookies.getCookiesFromHost(hostPartitioned, {
+ partitionKey: "(https,example.com)",
+ }).length; // includes expired cookies
+ Assert.equal(cookieCountPart, cookieNum3);
+ let cookieCountPartOA = Services.cookies.getCookiesFromHost(hostPartitioned, {
+ partitionKey: "(https,example.com)",
+ userContextId: 7,
+ }).length; // includes expired cookies
+ Assert.equal(cookieCountPartOA, cookieNum4);
+
+ // trigger the collection
+ Services.obs.notifyObservers(null, "idle-daily");
+
+ // check telem fired for all cookie count
+ let cct = await Glean.networking.cookieCountTotal.testGetValue();
+ Assert.equal(cct.sum, allCookieCount, "All cookies telem");
+
+ // check telem for all un/partitioned counts
+ let ccp = await Glean.networking.cookieCountPartitioned.testGetValue();
+ Assert.equal(
+ ccp.sum,
+ cookieNum3 + cookieNum4,
+ "All partitioned cookies telem"
+ );
+
+ let ccu = await Glean.networking.cookieCountUnpartitioned.testGetValue();
+ Assert.equal(
+ ccu.sum,
+ cookieNum1 + cookieNum2,
+ "All unpartitioned cookies telem"
+ );
+
+ // check telem for part by key (host+OA)
+ // Note: With the decided histogram layout we see the buckets
+ // (used for indexing the histogram's buckets)
+ let histPartByKey =
+ await Glean.networking.cookieCountPartByKey.testGetValue();
+ Assert.equal(histPartByKey.values[2], 1, "Partitioned bucket 2 has 1 value");
+ Assert.equal(
+ histPartByKey.values[31],
+ 1,
+ "Partitioned bucket 31 has 1 value"
+ );
+ Assert.equal(
+ histPartByKey.sum,
+ cookieNum3 + cookieNum4,
+ "Partitioned bucket sums correctly"
+ );
+
+ // check telem for unpart by key (host+OA)
+ let histUnpartByKey =
+ await Glean.networking.cookieCountUnpartByKey.testGetValue();
+ Assert.equal(
+ histUnpartByKey.values[1],
+ 1,
+ "Unpartitioned bucket 1 has 1 value"
+ );
+ Assert.equal(
+ histUnpartByKey.values[8],
+ 1,
+ "Unpartitioned bucket 8 has 1 value"
+ );
+ Assert.equal(
+ histUnpartByKey.sum,
+ cookieNum1 + cookieNum2,
+ "Unpartitioned bucket sums correctly"
+ );
+
+ schema12db.close();
+});