Browserstack tests for session ID

rosahbruno · rosahbruno · commit b88c065bda36 · 2024-02-06T01:06:49.000-05:00
diff --git a/automation/compat/index.html b/automation/compat/index.html
@@ -7,7 +7,8 @@
   <title>Glean Benchmarks Sample</title>
 </head>
 <body>
-  <p id="msg"></p>
+  <p id="ping_msg"></p>
+  <p id="session_msg"></p>
   <script src="./dist/index.js"></script>
 </body>
 </html>
diff --git a/automation/compat/index.js b/automation/compat/index.js
@@ -8,19 +8,12 @@ import Glean from "@mozilla/glean/web";
 import { benchmark } from "./generated/pings.js";
 import * as metrics from "./generated/sample.js";
 
-Glean.setSourceTags(["automation"]);
-Glean.initialize("glean-compat-benchmark", true, {
-  enableAutoPageLoadEvents: true
-});
-
-metrics.pageLoaded.set();
-benchmark.submit();
-
 // !BIG HACK!
 //
 // Overwrite the console.info function in order to know when (and if) the benchmark ping was sent.
 // If a success ping message is logged we show that in the document.
 let pingSubmissionCount = 0;
+let sessionId = "";
 console.info = function () {
   var message = "";
   for (var i = 0; i < arguments.length; i++) {
@@ -30,6 +23,7 @@ console.info = function () {
       message += " ";
     }
   }
+
   console.log(message);
   if (/successfully sent 200.$/.test(message)) {
     pingSubmissionCount++;
@@ -38,8 +32,40 @@ console.info = function () {
     // 1. The built-in page_load event, which submits an events ping.
     // 2. The benchmark ping.
     if (pingSubmissionCount == 2) {
-      var elem = document.getElementById("msg");
+      var elem = document.getElementById("ping_msg");
       elem.innerHTML = "Pings submitted successfully.";
     }
   }
+
+  const sessionRegex = /"session_id": .+"/;
+  const sessionInfo = sessionRegex.exec(message);
+  if (!!sessionInfo) {
+    const currSessionId = sessionInfo?.[0].split(`"`)?.[3];
+    if (!!sessionId) {
+      if (currSessionId !== sessionId) {
+        var elem = document.getElementById("session_msg");
+        elem.innerHTML = "Session IDs updated successfully.";
+      } else {
+        console.log("Something went wrong...");
+      }
+    }
+
+    sessionId = currSessionId;
+  }
 }
+
+Glean.setSourceTags(["automation"]);
+
+// This needs to be set so we can pull the session ID from the log messages.
+Glean.setLogPings(true);
+
+Glean.initialize("glean-compat-benchmark", true, {
+  enableAutoPageLoadEvents: true,
+  // Setting the override to 0 means every action will trigger
+  // a new session. We use this to check that the session ID
+  // changes whenever a session has expired.
+  sessionLengthInMinutesOverride: 0
+});
+
+metrics.pageLoaded.set();
+benchmark.submit();
diff --git a/automation/compat/tests/utils.js b/automation/compat/tests/utils.js
@@ -73,7 +73,7 @@ export async function runWebTest(driver) {
     // will receive the text "Ping submitted successfully."
     await driver.get(`http://localhost:${PORT}/`);
     // Give it time to send the ping request.
-    const successTextContainer = await driver.findElement(By.id("msg"));
+    const pingTextContainer = await driver.findElement(By.id("ping_msg"));
 
     const areGleanWindowVarsSet = await driver.executeScript(() => {
       // Verify that all Glean `window` vars are properly set.
@@ -100,10 +100,17 @@ export async function runWebTest(driver) {
 
     await driver.wait(
       until.elementTextIs(
-        successTextContainer,
+        pingTextContainer,
         "Pings submitted successfully."
       ), 11_000); // 1s more than the default upload timeout in Glean.
 
+    const sessionTextContainer = await driver.findElement(By.id("session_msg"));
+    await driver.wait(
+      until.elementTextIs(
+        sessionTextContainer,
+        "Session IDs updated successfully."
+      ), 1000);
+
     console.log("Test passed.");
   } catch(e) {
     console.log("Test failed.", e);
