test: fix remaining release protocol handshake fixtures (#2909)

Author: miguelg719

packages/extension/tests/rpc-client.test.ts

@@ -3,11 +3,15 @@ import { StackContextManager } from "@opentelemetry/sdk-trace-web";
 import { describe, expect, it, vi } from "vitest";
 import { JSONRPCRequestSchema } from "@browserbasehq/stagehand-protocol/json-rpc/schemas";
 import { StagehandMethods } from "@browserbasehq/stagehand-protocol/schema-registry";
+import { STAGEHAND_PROTOCOL_VERSION } from "@browserbasehq/stagehand-protocol/schemas";
 import { ChromeRuntimeClient } from "../clients/chromeRuntimeClient.ts";
 import { RPCClient } from "../clients/rpcClient.ts";
 import { createStagehandRuntime } from "../runtime.ts";
 import { RPCRouter } from "../rpcRouter.ts";
 
+const [protocolMajor, protocolMinor, protocolPatch] =
+  STAGEHAND_PROTOCOL_VERSION.split(".").map(Number);
+
 describe("worker RPCClient", () => {
   function createRuntime() {
     return createStagehandRuntime(
@@ -315,10 +319,10 @@ describe("worker RPCClient", () => {
   });
 
   it.each([
-    ["1.0.9", true, undefined],
-    ["1.1.0", false, "protocol-server-too-old"],
-    ["2.0.0", false, "protocol-major-mismatch"],
-    ["1.0.0-beta.1", false, "protocol-prerelease-mismatch"],
+    [`${protocolMajor}.${protocolMinor}.${protocolPatch + 1}`, true, undefined],
+    [`${protocolMajor}.${protocolMinor + 1}.0`, false, "protocol-server-too-old"],
+    [`${protocolMajor + 1}.0.0`, false, "protocol-major-mismatch"],
+    [`${STAGEHAND_PROTOCOL_VERSION}-beta.1`, false, "protocol-prerelease-mismatch"],
   ] as const)(
     "negotiates client protocol %s through the stagehand.init wire handshake",
     async (protocolVersion, compatible, reason) => {

packages/extension/tests/rpc-router.test.ts

@@ -21,6 +21,7 @@ import type { CdpWebSocketCloseEvent, CdpWebSocketTransport } from "../understud
 import type { Page } from "../understudy/page.ts";
 
 const EMPTY_METRICS = new StagehandMetricsAccumulator().snapshot();
+const [protocolMajor, protocolMinor] = STAGEHAND_PROTOCOL_VERSION.split(".").map(Number);
 
 afterEach(() => vi.restoreAllMocks());
 
@@ -328,9 +329,9 @@ describe("Stagehand RPC router", () => {
   });
 
   it.each([
-    ["1.1.0", "protocol-server-too-old"],
-    ["2.0.0", "protocol-major-mismatch"],
-    ["1.0.0-beta.1", "protocol-prerelease-mismatch"],
+    [`${protocolMajor}.${protocolMinor + 1}.0`, "protocol-server-too-old"],
+    [`${protocolMajor + 1}.0.0`, "protocol-major-mismatch"],
+    [`${STAGEHAND_PROTOCOL_VERSION}-beta.1`, "protocol-prerelease-mismatch"],
   ] as const)("rejects incompatible client protocol %s", async (protocolVersion, reason) => {
     const initializeStagehand = vi.fn(async () => ({ initialized: true as const, pages: [] }));
     const router = new RPCRouter(createStagehandRuntime(), { initializeStagehand });