@platforma-sdk/model 1.58.5 → 1.58.9

package/src/block_migrations.test.ts

@@ -1,10 +1,15 @@
 import { describe, expect, it } from "vitest";
-import { DataModelBuilder, defaultRecover, makeDataVersioned } from "./block_migrations";
+import {
+  DataModelBuilder,
+  defaultRecover,
+  makeVersionedData,
+  type TransferTarget,
+} from "./block_migrations";
 import { DATA_MODEL_LEGACY_VERSION } from "./block_storage";
 
-describe("makeDataVersioned", () => {
+describe("makeVersionedData", () => {
   it("creates correct DataVersioned shape", () => {
-    const versioned = makeDataVersioned("v1", { count: 42 });
+    const versioned = makeVersionedData("v1", { count: 42 });
     expect(versioned).toStrictEqual({ version: "v1", data: { count: 42 } });
   });
 });
@@ -16,7 +21,7 @@ describe("DataModel migrations", () => {
       .migrate<{ count: number; label: string }>("v2", (v1) => ({ ...v1, label: "" }))
       .init(() => ({ count: 0, label: "" }));
 
-    const result = dataModel.migrate(makeDataVersioned("legacy", { count: 42 }));
+    const result = dataModel.migrate(makeVersionedData("legacy", { count: 42 }));
     expect(result.version).toBe("v2");
     expect(result.data).toStrictEqual({ count: 0, label: "" });
   });
@@ -42,7 +47,7 @@ describe("DataModel migrations", () => {
       })
       .init(() => ({ numbers: [], label: "" }));
 
-    expect(() => dataModel.migrate(makeDataVersioned("v1", { numbers: [666] }))).toThrow(
+    expect(() => dataModel.migrate(makeVersionedData("v1", { numbers: [666] }))).toThrow(
       "Forbidden number",
     );
   });
@@ -66,7 +71,7 @@ describe("DataModel migrations", () => {
         .init(() => ({ count: 0, label: "" }));
 
       // Legacy data is recovered as V1 then goes through v1→v2 migration
-      const result = dataModel.migrate(makeDataVersioned("legacy", { count: 5 }));
+      const result = dataModel.migrate(makeVersionedData("legacy", { count: 5 }));
       expect(result.version).toBe("v2");
       expect(result.data).toStrictEqual({ count: 5, label: "default" });
     });
@@ -92,7 +97,7 @@ describe("DataModel migrations", () => {
         .migrate<V3>("v3", (v2) => ({ ...v2, description: "added" }))
         .init(() => ({ count: 0, label: "", description: "" }));
 
-      const result = dataModel.migrate(makeDataVersioned("legacy", { count: 7 }));
+      const result = dataModel.migrate(makeVersionedData("legacy", { count: 7 }));
       expect(result.version).toBe("v3");
       expect(result.data).toStrictEqual({ count: 7, label: "recovered", description: "added" });
     });
@@ -116,7 +121,7 @@ describe("DataModel migrations", () => {
         })
         .init(() => ({ count: 0, label: "" }));
 
-      const result = dataModel.migrate(makeDataVersioned("legacy", { count: 9 }));
+      const result = dataModel.migrate(makeVersionedData("legacy", { count: 9 }));
       expect(result.version).toBe("v2");
       expect(result.data).toStrictEqual({ count: 9, label: "recovered" });
     });
@@ -128,7 +133,7 @@ describe("DataModel migrations", () => {
         .recover((version, data) => defaultRecover(version, data))
         .init(() => ({ count: 0, label: "" }));
 
-      const result = dataModel.migrate(makeDataVersioned("unknown", { count: 7 }));
+      const result = dataModel.migrate(makeVersionedData("unknown", { count: 7 }));
       expect(result.version).toBe("v2");
       expect(result.data).toStrictEqual({ count: 0, label: "" });
     });
@@ -156,7 +161,7 @@ describe("DataModel migrations", () => {
         })
         .init(() => ({ count: 0, label: "", description: "" }));
 
-      const result = dataModel.migrate(makeDataVersioned("legacy", { count: 7 }));
+      const result = dataModel.migrate(makeVersionedData("legacy", { count: 7 }));
       expect(result.version).toBe("v3");
       expect(result.data).toStrictEqual({ count: 0, label: "", description: "" });
     });
@@ -184,7 +189,7 @@ describe("DataModel migrations", () => {
 
       // Legacy data arrives at DATA_MODEL_LEGACY_VERSION (how normalizeBlockStorage wraps it)
       const result = dataModel.migrate(
-        makeDataVersioned(DATA_MODEL_LEGACY_VERSION, {
+        makeVersionedData(DATA_MODEL_LEGACY_VERSION, {
           args: { inputFile: "test.fa", threshold: 5 },
           uiState: { selectedTab: "results" },
         }),
@@ -209,7 +214,7 @@ describe("DataModel migrations", () => {
 
       // Non-legacy data at the user's version passes through unchanged
       const result = dataModel.migrate(
-        makeDataVersioned("v1", {
+        makeVersionedData("v1", {
           inputFile: "existing.fa",
           threshold: 10,
           selectedTab: "overview",
@@ -237,7 +242,7 @@ describe("DataModel migrations", () => {
         .init(() => ({ inputFile: "", threshold: 0, selectedTab: "main", description: "" }));
 
       const result = dataModel.migrate(
-        makeDataVersioned(DATA_MODEL_LEGACY_VERSION, {
+        makeVersionedData(DATA_MODEL_LEGACY_VERSION, {
           args: { inputFile: "test.fa", threshold: 3 },
           uiState: { selectedTab: "tab1" },
         }),
@@ -261,7 +266,7 @@ describe("DataModel migrations", () => {
 
       expect(() =>
         dataModel.migrate(
-          makeDataVersioned(DATA_MODEL_LEGACY_VERSION, {
+          makeVersionedData(DATA_MODEL_LEGACY_VERSION, {
             args: { inputFile: "test.fa", threshold: 5 },
             uiState: { selectedTab: "results" },
           }),
@@ -269,4 +274,169 @@ describe("DataModel migrations", () => {
       ).toThrow("bad legacy data");
     });
   });
+
+  describe("transfer()", () => {
+    // Helper: create a TransferTarget with given id and transferVersion
+    function makeTarget<Id extends string>(
+      id: Id,
+      transferVersion = "tv1",
+    ): TransferTarget<Id, unknown> {
+      return { id, transferVersion } as TransferTarget<Id, unknown>;
+    }
+
+    it("captures data at the current chain position via migrate()", () => {
+      type V1 = { count: number; tableState: string };
+      type V2 = { count: number };
+      const tableTarget = makeTarget("table1", "pv1");
+
+      const dataModel = new DataModelBuilder()
+        .from<V1>("v1")
+        .transfer(tableTarget, (v1) => ({ state: v1.tableState }))
+        .migrate<V2>("v2", ({ tableState: _, ...rest }) => rest)
+        .init(() => ({ count: 0 }));
+
+      const result = dataModel.migrate(
+        makeVersionedData("v1", { count: 5, tableState: "expanded" }),
+      );
+
+      expect(result.version).toBe("v2");
+      expect(result.data).toStrictEqual({ count: 5 });
+      expect(result.transfers).toHaveProperty("table1");
+      expect(result.transfers["table1"].data).toStrictEqual({ state: "expanded" });
+      expect(result.transfers["table1"].version).toBe("pv1");
+    });
+
+    it("captures data after last migration step when transfer is at the end", () => {
+      type V1 = { count: number };
+      type V2 = { count: number; label: string };
+      const target = makeTarget("endPlugin");
+
+      const dataModel = new DataModelBuilder()
+        .from<V1>("v1")
+        .migrate<V2>("v2", (v1) => ({ ...v1, label: "added" }))
+        .transfer(target, (v2) => ({ snapshot: v2.label }))
+        .init(() => ({ count: 0, label: "" }));
+
+      const result = dataModel.migrate(makeVersionedData("v1", { count: 1 }));
+
+      expect(result.version).toBe("v2");
+      expect(result.transfers["endPlugin"].data).toStrictEqual({ snapshot: "added" });
+    });
+
+    it("collects multiple transfers at different chain positions", () => {
+      type V1 = { a: number; b: string };
+      type V2 = { a: number };
+      type V3 = { a: number; c: boolean };
+      const target1 = makeTarget("plugin1");
+      const target2 = makeTarget("plugin2");
+
+      const dataModel = new DataModelBuilder()
+        .from<V1>("v1")
+        .transfer(target1, (v1) => ({ fromB: v1.b }))
+        .migrate<V2>("v2", ({ b: _, ...rest }) => rest)
+        .transfer(target2, (v2) => ({ fromA: v2.a }))
+        .migrate<V3>("v3", (v2) => ({ ...v2, c: true }))
+        .init(() => ({ a: 0, c: false }));
+
+      const result = dataModel.migrate(makeVersionedData("v1", { a: 42, b: "hello" }));
+
+      expect(result.data).toStrictEqual({ a: 42, c: true });
+      expect(result.transfers["plugin1"].data).toStrictEqual({ fromB: "hello" });
+      expect(result.transfers["plugin2"].data).toStrictEqual({ fromA: 42 });
+    });
+
+    it("migrate() still works unchanged (ignores transfers)", () => {
+      type V1 = { count: number; tableState: string };
+      type V2 = { count: number };
+      const tableTarget = makeTarget("table1");
+
+      const dataModel = new DataModelBuilder()
+        .from<V1>("v1")
+        .transfer(tableTarget, (v1) => ({ state: v1.tableState }))
+        .migrate<V2>("v2", ({ tableState: _, ...rest }) => rest)
+        .init(() => ({ count: 0 }));
+
+      const result = dataModel.migrate(
+        makeVersionedData("v1", { count: 5, tableState: "expanded" }),
+      );
+
+      expect(result.version).toBe("v2");
+      expect(result.data).toStrictEqual({ count: 5 });
+    });
+
+    it("returns empty transfers on fast-path (already at latest version)", () => {
+      type V1 = { count: number };
+      const target = makeTarget("s1");
+
+      const dataModel = new DataModelBuilder()
+        .from<V1>("v1")
+        .transfer(target, (v1) => ({ x: v1.count }))
+        .init(() => ({ count: 0 }));
+
+      const result = dataModel.migrate(makeVersionedData("v1", { count: 7 }));
+
+      expect(result.data).toStrictEqual({ count: 7 });
+      expect(result.transfers).toStrictEqual({});
+    });
+
+    it("returns empty transfers on recovery path", () => {
+      type V1 = { count: number };
+      type V2 = { count: number; label: string };
+      const target = makeTarget("s1");
+
+      const dataModel = new DataModelBuilder()
+        .from<V1>("v1")
+        .transfer(target, (v1) => ({ x: v1.count }))
+        .migrate<V2>("v2", (v1) => ({ ...v1, label: "" }))
+        .recover((_version, _data) => defaultRecover(_version, _data))
+        .init(() => ({ count: 0, label: "" }));
+
+      const result = dataModel.migrate(makeVersionedData("unknown", { count: 99 }));
+
+      // Recovery resets to initial data
+      expect(result.data).toStrictEqual({ count: 0, label: "" });
+      expect(result.transfers).toStrictEqual({});
+    });
+
+    it("rejects duplicate transfer plugin IDs at runtime", () => {
+      const target = makeTarget("dup");
+
+      expect(() => {
+        new DataModelBuilder()
+          .from<{ a: number }>("v1")
+          .transfer(target, (v1) => v1.a)
+          .transfer(target, (v1) => v1.a);
+      }).toThrow(/duplicate/i);
+    });
+
+    it("transfer with upgradeLegacy — transfer fires after upgrade step", () => {
+      type LegacyArgs = { inputFile: string };
+      type LegacyUiState = { tab: string };
+      type V1 = { inputFile: string; tab: string };
+      type V2 = { inputFile: string };
+      const target = makeTarget("tabPlugin", "pv1");
+
+      const dataModel = new DataModelBuilder()
+        .from<V1>("v1")
+        .upgradeLegacy<LegacyArgs, LegacyUiState>(({ args, uiState }) => ({
+          inputFile: args.inputFile,
+          tab: uiState.tab,
+        }))
+        .transfer(target, (v1) => ({ tab: v1.tab }))
+        .migrate<V2>("v2", ({ tab: _, ...rest }) => rest)
+        .init(() => ({ inputFile: "" }));
+
+      const result = dataModel.migrate(
+        makeVersionedData(DATA_MODEL_LEGACY_VERSION, {
+          args: { inputFile: "test.fa" },
+          uiState: { tab: "results" },
+        }),
+      );
+
+      expect(result.version).toBe("v2");
+      expect(result.data).toStrictEqual({ inputFile: "test.fa" });
+      expect(result.transfers["tabPlugin"].data).toStrictEqual({ tab: "results" });
+      expect(result.transfers["tabPlugin"].version).toBe("pv1");
+    });
+  });
 });