fixup! add test

Author: vicb

packages/cloudflare/src/cli/commands/skew-protection.spec.ts

@@ -0,0 +1,69 @@
+import { describe, expect, test, vi } from "vitest";
+
+import { CURRENT_VERSION_ID } from "../templates/skew-protection";
+import { listWorkerVersions, updateDeploymentMapping } from "./skew-protection";
+
+describe("skew protection", () => {
+	describe("listWorkerVersions", () => {
+		test("listWorkerVersions return versions ordered by time DESC", async () => {
+			// eslint-disable-next-line @typescript-eslint/no-explicit-any
+			const client: any = {
+				workers: {
+					scripts: {
+						versions: {
+							list: () => [],
+						},
+					},
+				},
+			};
+
+			const now = Date.now();
+
+			client.workers.scripts.versions.list = vi.fn().mockReturnValue([
+				{
+					id: "HEAD",
+					metadata: { created_on: new Date(now) },
+				},
+				{
+					id: "HEAD~2",
+					metadata: { created_on: new Date(now - 2000) },
+				},
+				{
+					id: "HEAD~1",
+					metadata: { created_on: new Date(now - 1000) },
+				},
+			]);
+
+			expect(await listWorkerVersions("scriptName", { client, accountId: "accountId" })).toMatchObject([
+				{
+					createdOnMs: now,
+					id: "HEAD",
+				},
+				{
+					createdOnMs: now - 1000,
+					id: "HEAD~1",
+				},
+				{
+					createdOnMs: now - 2000,
+					id: "HEAD~2",
+				},
+			]);
+		});
+	});
+});
+
+describe("updateDeploymentMapping", () => {
+	test("Update", () => {
+		const mapping = {
+			N: CURRENT_VERSION_ID,
+			"N-1": "vN-1",
+			"N-2": "vN-2",
+		};
+		const versions = [{ id: "vN" }, { id: "vN-1" }]; // "vN-2" is deleted
+		expect(updateDeploymentMapping(mapping, versions, "N+1")).toMatchObject({
+			"N+1": CURRENT_VERSION_ID,
+			N: "vN",
+			"N-1": "vN-1",
+		});
+	});
+});

packages/cloudflare/src/cli/commands/skew-protection.ts

@@ -126,7 +126,7 @@ export async function getDeploymentMapping(
  * @param deploymentId Deployment ID
  * @returns The updated mapping
  */
-function updateDeploymentMapping(
+export function updateDeploymentMapping(
 	mapping: Record<string, string>,
 	versions: { id: string }[],
 	deploymentId: string