@typeberry/lib 0.5.10 → 0.5.11-10b197e

package/packages/jam/jam-host-calls/general/fetch.test.js

@@ -1,14 +1,14 @@
 import assert from "node:assert";
 import { describe, it } from "node:test";
 import { tryAsServiceId } from "#@typeberry/block";
-import { BytesBlob } from "#@typeberry/bytes";
+import { Bytes, BytesBlob } from "#@typeberry/bytes";
+import { HASH_SIZE } from "#@typeberry/hash";
 import { tryAsU64 } from "#@typeberry/numbers";
 import { HostCallMemory, HostCallRegisters, PvmExecution } from "#@typeberry/pvm-host-calls";
 import { tryAsGas } from "#@typeberry/pvm-interface";
 import { gasCounter, MemoryBuilder, tryAsMemoryIndex, tryAsSbrkIndex } from "#@typeberry/pvm-interpreter";
 import { PAGE_SIZE } from "#@typeberry/pvm-interpreter/memory/memory-consts.js";
-import { emptyRegistersBuffer } from "../utils.js";
-import { Fetch, FetchKind } from "./fetch.js";
+import { Fetch, FetchContext, FetchKind } from "./fetch.js";
 import { HostCallResult } from "./results.js";
 describe("Fetch", () => {
     const IN_OUT_REG = 7;
@@ -16,28 +16,31 @@ describe("Fetch", () => {
     it("should return PvmExecution.Panic if memory write fails", async () => {
         const currentServiceId = tryAsServiceId(10_000);
         const blob = BytesBlob.blobFromNumbers([1, 2, 3]);
-        const fetchMock = new FetchMock();
+        const fetchMock = new RefineFetchMock();
         fetchMock.constantsResponse = blob;
         const badOffset = tryAsU64(0xfffff);
-        const registers = new HostCallRegisters(emptyRegistersBuffer());
+        const registers = HostCallRegisters.empty();
         registers.set(IN_OUT_REG, badOffset);
         registers.set(8, tryAsU64(0));
         registers.set(9, tryAsU64(blob.length));
         registers.set(10, tryAsU64(FetchKind.Constants));
         const builder = new MemoryBuilder();
         // do not define any writable memory!
-        const memory = new HostCallMemory(builder.finalize(tryAsMemoryIndex(0), tryAsSbrkIndex(0)));
-        const fetch = new Fetch(currentServiceId, fetchMock);
+        const memory = HostCallMemory.new(builder.finalize(tryAsMemoryIndex(0), tryAsSbrkIndex(0)));
+        const fetch = Fetch.new(currentServiceId, fetchMock);
         const result = await fetch.execute(gas, registers, memory);
         assert.strictEqual(result, PvmExecution.Panic);
     });
     it("should write empty result and set IN_OUT_REG to NONE if fetch returns null", async () => {
         const currentServiceId = tryAsServiceId(10_000);
-        const fetchMock = new FetchMock();
-        fetchMock.entropyResponse = null;
+        const fetchMock = new RefineFetchMock();
+        // oneWorkItem returns null when the work item index has no mock response registered
         const blob = BytesBlob.blobFromNumbers([]);
-        const { registers, memory, readBack } = prepareRegsAndMemory(blob, FetchKind.Entropy);
-        const fetch = new Fetch(currentServiceId, fetchMock);
+        const { registers, memory, readBack } = prepareRegsAndMemory(blob, FetchKind.OneWorkItem);
+        // set work item index to one that has no response → oneWorkItem returns null
+        registers.set(11, tryAsU64(999));
+        fetchMock.oneWorkItemResponses.set("999", null);
+        const fetch = Fetch.new(currentServiceId, fetchMock);
         const result = await fetch.execute(gas, registers, memory);
         assert.strictEqual(result, undefined);
         assert.strictEqual(registers.get(IN_OUT_REG), HostCallResult.NONE);
@@ -47,10 +50,10 @@ describe("Fetch", () => {
     it("should write nothing if offset >= blob length", async () => {
         const currentServiceId = tryAsServiceId(10_000);
         const blob = BytesBlob.blobFromNumbers([1, 2, 3]);
-        const fetchMock = new FetchMock();
+        const fetchMock = new RefineFetchMock();
         fetchMock.constantsResponse = blob;
         const { registers, memory, readBack } = prepareRegsAndMemory(blob, FetchKind.Constants, 5, 2);
-        const fetch = new Fetch(currentServiceId, fetchMock);
+        const fetch = Fetch.new(currentServiceId, fetchMock);
         const result = await fetch.execute(gas, registers, memory);
         assert.strictEqual(result, undefined);
         assert.deepStrictEqual(registers.get(IN_OUT_REG), tryAsU64(blob.length));
@@ -59,10 +62,10 @@ describe("Fetch", () => {
     it("should clamp offset + length to blob end", async () => {
         const currentServiceId = tryAsServiceId(10_000);
         const blob = BytesBlob.blobFromNumbers([9, 8, 7, 6, 5]);
-        const fetchMock = new FetchMock();
+        const fetchMock = new RefineFetchMock();
         fetchMock.constantsResponse = blob;
         const { registers, memory, readBack } = prepareRegsAndMemory(blob, FetchKind.Constants, 3, 10);
-        const fetch = new Fetch(currentServiceId, fetchMock);
+        const fetch = Fetch.new(currentServiceId, fetchMock);
         const result = await fetch.execute(gas, registers, memory);
         assert.strictEqual(result, undefined);
         assert.deepStrictEqual(registers.get(IN_OUT_REG), tryAsU64(blob.length));
@@ -71,11 +74,11 @@ describe("Fetch", () => {
     it("should return NONE and write nothing if fetch kind is unknown", async () => {
         const currentServiceId = tryAsServiceId(10_000);
         const blob = BytesBlob.empty();
-        const fetchMock = new FetchMock();
+        const fetchMock = new RefineFetchMock();
         fetchMock.constantsResponse = blob;
         const { registers, memory, readBack } = prepareRegsAndMemory(blob, FetchKind.Constants);
         registers.set(10, tryAsU64(999));
-        const fetch = new Fetch(currentServiceId, fetchMock);
+        const fetch = Fetch.new(currentServiceId, fetchMock);
         const result = await fetch.execute(gas, registers, memory);
         assert.strictEqual(result, undefined);
         assert.strictEqual(registers.get(IN_OUT_REG), HostCallResult.NONE);
@@ -84,10 +87,10 @@ describe("Fetch", () => {
     it("should fetch constants and write result to memory", async () => {
         const currentServiceId = tryAsServiceId(10_000);
         const blob = BytesBlob.blobFromNumbers([1, 2, 3, 4, 5]);
-        const fetchMock = new FetchMock();
+        const fetchMock = new RefineFetchMock();
         fetchMock.constantsResponse = blob;
         const { registers, memory, readBack, expectedLength } = prepareRegsAndMemory(blob, FetchKind.Constants);
-        const fetch = new Fetch(currentServiceId, fetchMock);
+        const fetch = Fetch.new(currentServiceId, fetchMock);
         const result = await fetch.execute(gas, registers, memory);
         assert.strictEqual(result, undefined);
         assert.deepStrictEqual(registers.get(IN_OUT_REG), expectedLength);
@@ -95,11 +98,11 @@ describe("Fetch", () => {
     });
     it("should fetch entropy and write result to memory", async () => {
         const currentServiceId = tryAsServiceId(10_000);
-        const blob = BytesBlob.blobFromNumbers([10, 20, 30, 40]);
-        const fetchMock = new FetchMock();
+        const blob = Bytes.fill(HASH_SIZE, 10).asOpaque();
+        const fetchMock = new RefineFetchMock();
         fetchMock.entropyResponse = blob;
         const { registers, memory, readBack, expectedLength } = prepareRegsAndMemory(blob, FetchKind.Entropy);
-        const fetch = new Fetch(currentServiceId, fetchMock);
+        const fetch = Fetch.new(currentServiceId, fetchMock);
         const result = await fetch.execute(gas, registers, memory);
         assert.strictEqual(result, undefined);
         assert.deepStrictEqual(registers.get(IN_OUT_REG), expectedLength);
@@ -108,10 +111,10 @@ describe("Fetch", () => {
     it("should fetch authorizer trace and write result to memory", async () => {
         const currentServiceId = tryAsServiceId(10_000);
         const blob = BytesBlob.blobFromNumbers([9, 9, 9]);
-        const fetchMock = new FetchMock();
+        const fetchMock = new RefineFetchMock();
         fetchMock.authorizerTraceResponse = blob;
         const { registers, memory, readBack, expectedLength } = prepareRegsAndMemory(blob, FetchKind.AuthorizerTrace);
-        const fetch = new Fetch(currentServiceId, fetchMock);
+        const fetch = Fetch.new(currentServiceId, fetchMock);
         const result = await fetch.execute(gas, registers, memory);
         assert.strictEqual(result, undefined);
         assert.deepStrictEqual(registers.get(IN_OUT_REG), expectedLength);
@@ -120,7 +123,7 @@ describe("Fetch", () => {
     it("should fetch other work item extrinsics and write result to memory", async () => {
         const currentServiceId = tryAsServiceId(10_000);
         const blob = BytesBlob.blobFromNumbers([42, 43, 44]);
-        const fetchMock = new FetchMock();
+        const fetchMock = new RefineFetchMock();
         const workItem = tryAsU64(123);
         const index = tryAsU64(7);
         const key = `${workItem}:${index}`;
@@ -128,7 +131,7 @@ describe("Fetch", () => {
         const { registers, memory, readBack, expectedLength } = prepareRegsAndMemory(blob, FetchKind.OtherWorkItemExtrinsics);
         registers.set(11, workItem);
         registers.set(12, index);
-        const fetch = new Fetch(currentServiceId, fetchMock);
+        const fetch = Fetch.new(currentServiceId, fetchMock);
         const result = await fetch.execute(gas, registers, memory);
         assert.strictEqual(result, undefined);
         assert.deepStrictEqual(registers.get(IN_OUT_REG), expectedLength);
@@ -138,13 +141,13 @@ describe("Fetch", () => {
     it("should fetch my extrinsics and write result to memory", async () => {
         const currentServiceId = tryAsServiceId(10_000);
         const blob = BytesBlob.blobFromNumbers([11, 12, 13]);
-        const fetchMock = new FetchMock();
+        const fetchMock = new RefineFetchMock();
         const index = tryAsU64(5);
         const key = `null:${index}`;
         fetchMock.workItemExtrinsicResponses.set(key, blob);
         const { registers, memory, readBack, expectedLength } = prepareRegsAndMemory(blob, FetchKind.MyExtrinsics);
         registers.set(11, index); // only index; workItem is null
-        const fetch = new Fetch(currentServiceId, fetchMock);
+        const fetch = Fetch.new(currentServiceId, fetchMock);
         const result = await fetch.execute(gas, registers, memory);
         assert.strictEqual(result, undefined);
         assert.deepStrictEqual(registers.get(IN_OUT_REG), expectedLength);
@@ -154,7 +157,7 @@ describe("Fetch", () => {
     it("should fetch other work item imports and write result to memory", async () => {
         const currentServiceId = tryAsServiceId(10_000);
         const blob = BytesBlob.blobFromNumbers([21, 22, 23]);
-        const fetchMock = new FetchMock();
+        const fetchMock = new RefineFetchMock();
         const workItem = tryAsU64(42);
         const index = tryAsU64(3);
         const key = `${workItem}:${index}`;
@@ -162,7 +165,7 @@ describe("Fetch", () => {
         const { registers, memory, readBack, expectedLength } = prepareRegsAndMemory(blob, FetchKind.OtherWorkItemImports);
         registers.set(11, workItem);
         registers.set(12, index);
-        const fetch = new Fetch(currentServiceId, fetchMock);
+        const fetch = Fetch.new(currentServiceId, fetchMock);
         const result = await fetch.execute(gas, registers, memory);
         assert.strictEqual(result, undefined);
         assert.deepStrictEqual(registers.get(IN_OUT_REG), expectedLength);
@@ -172,13 +175,13 @@ describe("Fetch", () => {
     it("should fetch my imports and write result to memory", async () => {
         const currentServiceId = tryAsServiceId(10_000);
         const blob = BytesBlob.blobFromNumbers([31, 32, 33]);
-        const fetchMock = new FetchMock();
+        const fetchMock = new RefineFetchMock();
         const index = tryAsU64(8);
         const key = `null:${index}`;
         fetchMock.workItemImportResponses.set(key, blob);
         const { registers, memory, readBack, expectedLength } = prepareRegsAndMemory(blob, FetchKind.MyImports);
         registers.set(11, index); // workItem is implicitly null
-        const fetch = new Fetch(currentServiceId, fetchMock);
+        const fetch = Fetch.new(currentServiceId, fetchMock);
         const result = await fetch.execute(gas, registers, memory);
         assert.strictEqual(result, undefined);
         assert.deepStrictEqual(registers.get(IN_OUT_REG), expectedLength);
@@ -188,10 +191,10 @@ describe("Fetch", () => {
     it("should fetch work package and write result to memory", async () => {
         const currentServiceId = tryAsServiceId(10_000);
         const blob = BytesBlob.blobFromNumbers([100, 101, 102]);
-        const fetchMock = new FetchMock();
+        const fetchMock = new RefineFetchMock();
         fetchMock.workPackageResponse = blob;
         const { registers, memory, readBack, expectedLength } = prepareRegsAndMemory(blob, FetchKind.WorkPackage);
-        const fetch = new Fetch(currentServiceId, fetchMock);
+        const fetch = Fetch.new(currentServiceId, fetchMock);
         const result = await fetch.execute(gas, registers, memory);
         assert.strictEqual(result, undefined);
         assert.deepStrictEqual(registers.get(IN_OUT_REG), expectedLength);
@@ -200,10 +203,10 @@ describe("Fetch", () => {
     it("should fetch authorizer and write result to memory", async () => {
         const currentServiceId = tryAsServiceId(10_000);
         const blob = BytesBlob.blobFromNumbers([201, 202, 203]);
-        const fetchMock = new FetchMock();
+        const fetchMock = new RefineFetchMock();
         fetchMock.authorizerResponse = blob;
-        const { registers, memory, readBack, expectedLength } = prepareRegsAndMemory(blob, FetchKind.Authorizer);
-        const fetch = new Fetch(currentServiceId, fetchMock);
+        const { registers, memory, readBack, expectedLength } = prepareRegsAndMemory(blob, FetchKind.AuthConfiguration);
+        const fetch = Fetch.new(currentServiceId, fetchMock);
         const result = await fetch.execute(gas, registers, memory);
         assert.strictEqual(result, undefined);
         assert.deepStrictEqual(registers.get(IN_OUT_REG), expectedLength);
@@ -212,10 +215,10 @@ describe("Fetch", () => {
     it("should fetch authorization token and write result to memory", async () => {
         const currentServiceId = tryAsServiceId(10_000);
         const blob = BytesBlob.blobFromNumbers([210, 211, 212]);
-        const fetchMock = new FetchMock();
+        const fetchMock = new RefineFetchMock();
         fetchMock.authorizationTokenResponse = blob;
-        const { registers, memory, readBack, expectedLength } = prepareRegsAndMemory(blob, FetchKind.AuthorizationToken);
-        const fetch = new Fetch(currentServiceId, fetchMock);
+        const { registers, memory, readBack, expectedLength } = prepareRegsAndMemory(blob, FetchKind.AuthToken);
+        const fetch = Fetch.new(currentServiceId, fetchMock);
         const result = await fetch.execute(gas, registers, memory);
         assert.strictEqual(result, undefined);
         assert.deepStrictEqual(registers.get(IN_OUT_REG), expectedLength);
@@ -224,10 +227,10 @@ describe("Fetch", () => {
     it("should fetch refine context and write result to memory", async () => {
         const currentServiceId = tryAsServiceId(10_000);
         const blob = BytesBlob.blobFromNumbers([88, 89, 90]);
-        const fetchMock = new FetchMock();
+        const fetchMock = new RefineFetchMock();
         fetchMock.refineContextResponse = blob;
         const { registers, memory, readBack, expectedLength } = prepareRegsAndMemory(blob, FetchKind.RefineContext);
-        const fetch = new Fetch(currentServiceId, fetchMock);
+        const fetch = Fetch.new(currentServiceId, fetchMock);
         const result = await fetch.execute(gas, registers, memory);
         assert.strictEqual(result, undefined);
         assert.deepStrictEqual(registers.get(IN_OUT_REG), expectedLength);
@@ -236,10 +239,10 @@ describe("Fetch", () => {
     it("should fetch all work items and write result to memory", async () => {
         const currentServiceId = tryAsServiceId(10_000);
         const blob = BytesBlob.blobFromNumbers([70, 71, 72]);
-        const fetchMock = new FetchMock();
+        const fetchMock = new RefineFetchMock();
         fetchMock.allWorkItemsResponse = blob;
         const { registers, memory, readBack, expectedLength } = prepareRegsAndMemory(blob, FetchKind.AllWorkItems);
-        const fetch = new Fetch(currentServiceId, fetchMock);
+        const fetch = Fetch.new(currentServiceId, fetchMock);
         const result = await fetch.execute(gas, registers, memory);
         assert.strictEqual(result, undefined);
         assert.deepStrictEqual(registers.get(IN_OUT_REG), expectedLength);
@@ -248,12 +251,12 @@ describe("Fetch", () => {
     it("should fetch one work item and write result to memory", async () => {
         const currentServiceId = tryAsServiceId(10_000);
         const blob = BytesBlob.blobFromNumbers([33, 34, 35]);
-        const fetchMock = new FetchMock();
+        const fetchMock = new RefineFetchMock();
         const workItem = tryAsU64(55);
         fetchMock.oneWorkItemResponses.set(workItem.toString(), blob);
         const { registers, memory, readBack, expectedLength } = prepareRegsAndMemory(blob, FetchKind.OneWorkItem);
         registers.set(11, workItem);
-        const fetch = new Fetch(currentServiceId, fetchMock);
+        const fetch = Fetch.new(currentServiceId, fetchMock);
         const result = await fetch.execute(gas, registers, memory);
         assert.strictEqual(result, undefined);
         assert.deepStrictEqual(registers.get(IN_OUT_REG), expectedLength);
@@ -263,12 +266,12 @@ describe("Fetch", () => {
     it("should fetch work item payload and write result to memory", async () => {
         const currentServiceId = tryAsServiceId(10_000);
         const blob = BytesBlob.blobFromNumbers([60, 61, 62]);
-        const fetchMock = new FetchMock();
+        const fetchMock = new RefineFetchMock();
         const workItem = tryAsU64(77);
         fetchMock.workItemPayloadResponses.set(workItem.toString(), blob);
         const { registers, memory, readBack, expectedLength } = prepareRegsAndMemory(blob, FetchKind.WorkItemPayload);
         registers.set(11, workItem);
-        const fetch = new Fetch(currentServiceId, fetchMock);
+        const fetch = Fetch.new(currentServiceId, fetchMock);
         const result = await fetch.execute(gas, registers, memory);
         assert.strictEqual(result, undefined);
         assert.deepStrictEqual(registers.get(IN_OUT_REG), expectedLength);
@@ -278,10 +281,10 @@ describe("Fetch", () => {
     it("should fetch all transfers and operands and write result to memory", async () => {
         const currentServiceId = tryAsServiceId(10_000);
         const blob = BytesBlob.blobFromNumbers([101, 102, 103]);
-        const fetchMock = new FetchMock();
-        fetchMock.allTransfersAndOperandsResponses = blob;
+        const fetchMock = new AccumulateFetchMock();
+        fetchMock.allTransfersAndOperandsResponse = blob;
         const { registers, memory, readBack, expectedLength } = prepareRegsAndMemory(blob, FetchKind.AllTransfersAndOperands);
-        const fetch = new Fetch(currentServiceId, fetchMock);
+        const fetch = Fetch.new(currentServiceId, fetchMock);
         const result = await fetch.execute(gas, registers, memory);
         assert.strictEqual(result, undefined);
         assert.deepStrictEqual(registers.get(IN_OUT_REG), expectedLength);
@@ -290,30 +293,67 @@ describe("Fetch", () => {
     it("should fetch one operand or transfer and write result to memory", async () => {
         const currentServiceId = tryAsServiceId(10_000);
         const blob = BytesBlob.blobFromNumbers([115, 116, 117]);
-        const fetchMock = new FetchMock();
+        const fetchMock = new AccumulateFetchMock();
         const index = tryAsU64(9);
         fetchMock.oneTransferOrOperandResponses.set(index.toString(), blob);
         const { registers, memory, readBack, expectedLength } = prepareRegsAndMemory(blob, FetchKind.OneTransferOrOperand);
         registers.set(11, index);
-        const fetch = new Fetch(currentServiceId, fetchMock);
+        const fetch = Fetch.new(currentServiceId, fetchMock);
         const result = await fetch.execute(gas, registers, memory);
         assert.strictEqual(result, undefined);
         assert.deepStrictEqual(registers.get(IN_OUT_REG), expectedLength);
         assert.deepStrictEqual(readBack(), blob.raw);
         assert.deepStrictEqual(fetchMock.oneTransferOrOperandData, [[index]]);
     });
+    it("should return NONE for refine-only kinds in accumulate context", async () => {
+        const currentServiceId = tryAsServiceId(10_000);
+        const fetchMock = new AccumulateFetchMock();
+        const blob = BytesBlob.empty();
+        for (const kind of [
+            FetchKind.AuthorizerTrace,
+            FetchKind.OtherWorkItemExtrinsics,
+            FetchKind.MyExtrinsics,
+            FetchKind.OtherWorkItemImports,
+            FetchKind.MyImports,
+            FetchKind.WorkPackage,
+            FetchKind.AuthConfiguration,
+            FetchKind.AuthToken,
+            FetchKind.RefineContext,
+            FetchKind.AllWorkItems,
+            FetchKind.OneWorkItem,
+            FetchKind.WorkItemPayload,
+        ]) {
+            const { registers, memory } = prepareRegsAndMemory(blob, kind);
+            const fetch = Fetch.new(currentServiceId, fetchMock);
+            const result = await fetch.execute(gas, registers, memory);
+            assert.strictEqual(result, undefined, `Expected undefined for kind ${kind}`);
+            assert.strictEqual(registers.get(IN_OUT_REG), HostCallResult.NONE, `Expected NONE for kind ${kind}`);
+        }
+    });
+    it("should return NONE for accumulate-only kinds in refine context", async () => {
+        const currentServiceId = tryAsServiceId(10_000);
+        const fetchMock = new RefineFetchMock();
+        const blob = BytesBlob.empty();
+        for (const kind of [FetchKind.AllTransfersAndOperands, FetchKind.OneTransferOrOperand]) {
+            const { registers, memory } = prepareRegsAndMemory(blob, kind);
+            const fetch = Fetch.new(currentServiceId, fetchMock);
+            const result = await fetch.execute(gas, registers, memory);
+            assert.strictEqual(result, undefined, `Expected undefined for kind ${kind}`);
+            assert.strictEqual(registers.get(IN_OUT_REG), HostCallResult.NONE, `Expected NONE for kind ${kind}`);
+        }
+    });
     function prepareRegsAndMemory(blob, fetchKind, offset = 0, length = blob.length) {
         const pageStart = 2 ** 16;
         const memOffset = tryAsU64(pageStart + 1234);
         const blobLength = tryAsU64(blob.length);
-        const registers = new HostCallRegisters(emptyRegistersBuffer());
+        const registers = HostCallRegisters.empty();
         registers.set(IN_OUT_REG, memOffset);
         registers.set(8, tryAsU64(offset));
         registers.set(9, tryAsU64(length));
         registers.set(10, tryAsU64(fetchKind));
         const builder = new MemoryBuilder();
         builder.setWriteablePages(tryAsMemoryIndex(pageStart), tryAsMemoryIndex(pageStart + PAGE_SIZE));
-        const memory = new HostCallMemory(builder.finalize(tryAsMemoryIndex(0), tryAsSbrkIndex(0)));
+        const memory = HostCallMemory.new(builder.finalize(tryAsMemoryIndex(0), tryAsSbrkIndex(0)));
         const readBack = () => {
             const result = new Uint8Array(blob.length);
             assert.strictEqual(memory.loadInto(result, memOffset).isOk, true);
@@ -327,32 +367,24 @@ describe("Fetch", () => {
         };
     }
 });
-class FetchMock {
+class RefineFetchMock {
+    context = FetchContext.Refine;
     workItemExtrinsicData = [];
     workItemImportData = [];
     oneWorkItemData = [];
     workItemPayloadData = [];
-    oneOperandData = [];
-    oneTransferData = [];
-    oneTransferOrOperandData = [];
     constantsResponse = null;
     entropyResponse = null;
-    authorizerTraceResponse = null;
+    authorizerTraceResponse = BytesBlob.empty();
     workItemExtrinsicResponses = new Map();
     workItemImportResponses = new Map();
-    workPackageResponse = null;
-    authorizerResponse = null;
-    authorizationTokenResponse = null;
-    refineContextResponse = null;
-    allWorkItemsResponse = null;
+    workPackageResponse = BytesBlob.empty();
+    authorizerResponse = BytesBlob.empty();
+    authorizationTokenResponse = BytesBlob.empty();
+    refineContextResponse = BytesBlob.empty();
+    allWorkItemsResponse = BytesBlob.empty();
     oneWorkItemResponses = new Map();
     workItemPayloadResponses = new Map();
-    allOperandsResponse = null;
-    oneOperandResponses = new Map();
-    allTransfersResponse = null;
-    oneTransferResponses = new Map();
-    allTransfersAndOperandsResponses = null;
-    oneTransferOrOperandResponses = new Map();
     constants() {
         if (this.constantsResponse === null) {
             throw new Error("Unexpected call to constants.");
@@ -360,6 +392,9 @@ class FetchMock {
         return this.constantsResponse;
     }
     entropy() {
+        if (this.entropyResponse === null) {
+            throw new Error("Unexpected call to entropy.");
+        }
         return this.entropyResponse;
     }
     authorizerTrace() {
@@ -384,10 +419,10 @@ class FetchMock {
     workPackage() {
         return this.workPackageResponse;
     }
-    authorizer() {
+    authConfiguration() {
         return this.authorizerResponse;
     }
-    authorizationToken() {
+    authToken() {
         return this.authorizationTokenResponse;
     }
     refineContext() {
@@ -412,30 +447,28 @@ class FetchMock {
         }
         return this.workItemPayloadResponses.get(key) ?? null;
     }
-    allOperands() {
-        return this.allOperandsResponse;
-    }
-    oneOperand(operandIndex) {
-        this.oneOperandData.push([operandIndex]);
-        const key = operandIndex.toString();
-        if (!this.oneOperandResponses.has(key)) {
-            throw new Error(`Missing mock response for oneOperand(${key})`);
+}
+class AccumulateFetchMock {
+    context = FetchContext.Accumulate;
+    oneTransferOrOperandData = [];
+    constantsResponse = null;
+    entropyResponse = null;
+    allTransfersAndOperandsResponse = null;
+    oneTransferOrOperandResponses = new Map();
+    constants() {
+        if (this.constantsResponse === null) {
+            throw new Error("Unexpected call to constants.");
         }
-        return this.oneOperandResponses.get(key) ?? null;
-    }
-    allTransfers() {
-        return this.allTransfersResponse;
+        return this.constantsResponse;
     }
-    oneTransfer(transferIndex) {
-        this.oneTransferData.push([transferIndex]);
-        const key = transferIndex.toString();
-        if (!this.oneTransferResponses.has(key)) {
-            throw new Error(`Missing mock response for oneTransfer(${key})`);
+    entropy() {
+        if (this.entropyResponse === null) {
+            throw new Error("Unexpected call to entropy.");
         }
-        return this.oneTransferResponses.get(key) ?? null;
+        return this.entropyResponse;
     }
     allTransfersAndOperands() {
-        return this.allTransfersAndOperandsResponses;
+        return this.allTransfersAndOperandsResponse;
     }
     oneTransferOrOperand(index) {
         this.oneTransferOrOperandData.push([index]);

package/packages/jam/jam-host-calls/general/gas.d.ts

@@ -14,7 +14,8 @@ export declare class GasHostCall implements HostCallHandler {
     index: number & import("@typeberry/numbers").WithBytesRepresentation<4> & import("@typeberry/utils").WithOpaque<"HostCallIndex[U32]">;
     basicGasCost: number & import("@typeberry/numbers").WithBytesRepresentation<4> & import("@typeberry/utils").WithOpaque<"SmallGas[U32]">;
     tracedRegisters: (number & import("@typeberry/utils").WithOpaque<"register index">)[];
-    constructor(currentServiceId: ServiceId);
+    static new(currentServiceId: ServiceId): GasHostCall;
+    private constructor();
     execute(gas: IGasCounter, regs: HostCallRegisters): Promise<undefined | PvmExecution>;
 }
 //# sourceMappingURL=gas.d.ts.map

package/packages/jam/jam-host-calls/general/gas.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"gas.d.ts","sourceRoot":"","sources":["../../../../../../packages/jam/jam-host-calls/general/gas.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAElD,OAAO,KAAK,EAAE,eAAe,EAAE,iBAAiB,EAAE,MAAM,2BAA2B,CAAC;AACpF,OAAO,EAAE,KAAK,YAAY,EAAsC,MAAM,2BAA2B,CAAC;AAClG,OAAO,EAAE,KAAK,WAAW,EAAiB,MAAM,0BAA0B,CAAC;AAG3E;;;;;;GAMG;AACH,qBAAa,WAAY,YAAW,eAAe;aAKrB,gBAAgB,EAAE,SAAS;IAJvD,KAAK,iIAAyB;IAC9B,YAAY,4HAAqB;IACjC,eAAe,uEAAqB;gBAER,gBAAgB,EAAE,SAAS;IAEvD,OAAO,CAAC,GAAG,EAAE,WAAW,EAAE,IAAI,EAAE,iBAAiB,GAAG,OAAO,CAAC,SAAS,GAAG,YAAY,CAAC;CAMtF"}
+{"version":3,"file":"gas.d.ts","sourceRoot":"","sources":["../../../../../../packages/jam/jam-host-calls/general/gas.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAElD,OAAO,KAAK,EAAE,eAAe,EAAE,iBAAiB,EAAE,MAAM,2BAA2B,CAAC;AACpF,OAAO,EAAE,KAAK,YAAY,EAAsC,MAAM,2BAA2B,CAAC;AAClG,OAAO,EAAE,KAAK,WAAW,EAAiB,MAAM,0BAA0B,CAAC;AAG3E;;;;;;GAMG;AACH,qBAAa,WAAY,YAAW,eAAe;aASb,gBAAgB,EAAE,SAAS;IAR/D,KAAK,iIAAyB;IAC9B,YAAY,4HAAqB;IACjC,eAAe,uEAAqB;IAEpC,MAAM,CAAC,GAAG,CAAC,gBAAgB,EAAE,SAAS;IAItC,OAAO;IAEP,OAAO,CAAC,GAAG,EAAE,WAAW,EAAE,IAAI,EAAE,iBAAiB,GAAG,OAAO,CAAC,SAAS,GAAG,YAAY,CAAC;CAMtF"}

package/packages/jam/jam-host-calls/general/gas.js

@@ -14,6 +14,9 @@ export class GasHostCall {
     index = tryAsHostCallIndex(0);
     basicGasCost = tryAsSmallGas(10);
     tracedRegisters = traceRegisters(7);
+    static new(currentServiceId) {
+        return new GasHostCall(currentServiceId);
+    }
     constructor(currentServiceId) {
         this.currentServiceId = currentServiceId;
     }

package/packages/jam/jam-host-calls/general/gas.test.js

@@ -4,15 +4,14 @@ import { tryAsServiceId } from "#@typeberry/block";
 import { HostCallRegisters } from "#@typeberry/pvm-host-calls";
 import { tryAsGas } from "#@typeberry/pvm-interface";
 import { gasCounter } from "#@typeberry/pvm-interpreter";
-import { emptyRegistersBuffer } from "../utils.js";
 import { GasHostCall } from "./gas.js";
 const REGISTER = 7;
 describe("HostCalls: Gas", () => {
     it("should write U32 gas to register", () => {
         const currentServiceId = tryAsServiceId(10_000);
-        const gas = new GasHostCall(currentServiceId);
+        const gas = GasHostCall.new(currentServiceId);
         const counter = gasCounter(tryAsGas(10_000));
-        const regs = new HostCallRegisters(emptyRegistersBuffer());
+        const regs = HostCallRegisters.empty();
         assert.deepStrictEqual(regs.get(REGISTER), 0n);
         // when
         gas.execute(counter, regs);
@@ -21,9 +20,9 @@ describe("HostCalls: Gas", () => {
     });
     it("should write U64 gas to register", () => {
         const currentServiceId = tryAsServiceId(10_000);
-        const gas = new GasHostCall(currentServiceId);
+        const gas = GasHostCall.new(currentServiceId);
         const counter = gasCounter(tryAsGas(2n ** 64n - 1n));
-        const regs = new HostCallRegisters(emptyRegistersBuffer());
+        const regs = HostCallRegisters.empty();
         assert.deepStrictEqual(regs.get(REGISTER), 0n);
         // when
         gas.execute(counter, regs);

package/packages/jam/jam-host-calls/general/info.d.ts

@@ -33,7 +33,8 @@ export declare class Info implements HostCallHandler {
     index: number & import("@typeberry/numbers").WithBytesRepresentation<4> & import("@typeberry/utils").WithOpaque<"HostCallIndex[U32]">;
     basicGasCost: number & import("@typeberry/numbers").WithBytesRepresentation<4> & import("@typeberry/utils").WithOpaque<"SmallGas[U32]">;
     tracedRegisters: (number & import("@typeberry/utils").WithOpaque<"register index">)[];
-    constructor(currentServiceId: ServiceId, account: AccountsInfo);
+    static new(currentServiceId: ServiceId, account: AccountsInfo): Info;
+    private constructor();
     execute(_gas: IGasCounter, regs: HostCallRegisters, memory: HostCallMemory): Promise<undefined | PvmExecution>;
 }
 /**

package/packages/jam/jam-host-calls/general/info.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"info.d.ts","sourceRoot":"","sources":["../../../../../../packages/jam/jam-host-calls/general/info.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,SAAS,EAAkD,MAAM,kBAAkB,CAAC;AAKlG,OAAO,KAAK,EAAE,eAAe,EAAE,cAAc,EAAE,iBAAiB,EAAE,MAAM,2BAA2B,CAAC;AACpG,OAAO,EAAE,YAAY,EAAsC,MAAM,2BAA2B,CAAC;AAC7F,OAAO,EAAE,KAAK,WAAW,EAAiB,MAAM,0BAA0B,CAAC;AAC3E,OAAO,EAAE,kBAAkB,EAAE,MAAM,kBAAkB,CAAC;AAMtD,kDAAkD;AAClD,MAAM,WAAW,YAAY;IAC3B,wBAAwB;IACxB,cAAc,CAAC,SAAS,EAAE,SAAS,GAAG,IAAI,GAAG,kBAAkB,GAAG,IAAI,CAAC;CACxE;AAMD,eAAO,MAAM,OAAO,QACsF,CAAC;AAE3G;;;;;;;;;;;;;;;;;GAiBG;AACH,qBAAa,IAAK,YAAW,eAAe;aAMxB,gBAAgB,EAAE,SAAS;IAC3C,OAAO,CAAC,QAAQ,CAAC,OAAO;IAN1B,KAAK,iIAAyB;IAC9B,YAAY,4HAAqB;IACjC,eAAe,uEAAsD;gBAGnD,gBAAgB,EAAE,SAAS,EAC1B,OAAO,EAAE,YAAY;IAGlC,OAAO,CAAC,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,iBAAiB,EAAE,MAAM,EAAE,cAAc,GAAG,OAAO,CAAC,SAAS,GAAG,YAAY,CAAC;CA8CrH;AAED;;;;;;GAMG;AACH,eAAO,MAAM,2CAA2C;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;IAevD,CAAC"}
+{"version":3,"file":"info.d.ts","sourceRoot":"","sources":["../../../../../../packages/jam/jam-host-calls/general/info.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,SAAS,EAAkD,MAAM,kBAAkB,CAAC;AAKlG,OAAO,KAAK,EAAE,eAAe,EAAE,cAAc,EAAE,iBAAiB,EAAE,MAAM,2BAA2B,CAAC;AACpG,OAAO,EAAE,YAAY,EAAsC,MAAM,2BAA2B,CAAC;AAC7F,OAAO,EAAE,KAAK,WAAW,EAAiB,MAAM,0BAA0B,CAAC;AAC3E,OAAO,EAAE,kBAAkB,EAAE,MAAM,kBAAkB,CAAC;AAMtD,kDAAkD;AAClD,MAAM,WAAW,YAAY;IAC3B,wBAAwB;IACxB,cAAc,CAAC,SAAS,EAAE,SAAS,GAAG,IAAI,GAAG,kBAAkB,GAAG,IAAI,CAAC;CACxE;AAMD,eAAO,MAAM,OAAO,QACsF,CAAC;AAE3G;;;;;;;;;;;;;;;;;GAiBG;AACH,qBAAa,IAAK,YAAW,eAAe;aAUxB,gBAAgB,EAAE,SAAS;IAC3C,OAAO,CAAC,QAAQ,CAAC,OAAO;IAV1B,KAAK,iIAAyB;IAC9B,YAAY,4HAAqB;IACjC,eAAe,uEAAsD;IAErE,MAAM,CAAC,GAAG,CAAC,gBAAgB,EAAE,SAAS,EAAE,OAAO,EAAE,YAAY;IAI7D,OAAO;IAKD,OAAO,CAAC,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,iBAAiB,EAAE,MAAM,EAAE,cAAc,GAAG,OAAO,CAAC,SAAS,GAAG,YAAY,CAAC;CA8CrH;AAED;;;;;;GAMG;AACH,eAAO,MAAM,2CAA2C;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;IAevD,CAAC"}

package/packages/jam/jam-host-calls/general/info.js

@@ -37,6 +37,9 @@ export class Info {
     index = tryAsHostCallIndex(5);
     basicGasCost = tryAsSmallGas(10);
     tracedRegisters = traceRegisters(IN_OUT_REG, 8, OFFSET_REG, LEN_REG);
+    static new(currentServiceId, account) {
+        return new Info(currentServiceId, account);
+    }
     constructor(currentServiceId, account) {
         this.currentServiceId = currentServiceId;
         this.account = account;

package/packages/jam/jam-host-calls/general/info.test.js

@@ -12,7 +12,6 @@ import { tryAsSbrkIndex } from "#@typeberry/pvm-interpreter/memory/memory-index.
 import { PAGE_SIZE } from "#@typeberry/pvm-interpreter/spi-decoder/memory-conts.js";
 import { ServiceAccountInfo } from "#@typeberry/state";
 import { TestAccounts } from "../externalities/test-accounts.js";
-import { emptyRegistersBuffer } from "../utils.js";
 import { codecServiceAccountInfoWithThresholdBalance, Info, LEN_REG } from "./info.js";
 import { HostCallResult } from "./results.js";
 const SERVICE_ID_REG = 7;
@@ -23,13 +22,13 @@ const serviceAccountInfoSize = tryAsExactBytes(codecServiceAccountInfoWithThresh
 function prepareRegsAndMemory(serviceId, accountInfoLength = serviceAccountInfoSize) {
     const pageStart = 2 ** 16;
     const memStart = pageStart + PAGE_SIZE - accountInfoLength - 1;
-    const registers = new HostCallRegisters(emptyRegistersBuffer());
+    const registers = HostCallRegisters.empty();
     registers.set(SERVICE_ID_REG, tryAsU64(serviceId));
     registers.set(DEST_START_REG, tryAsU64(memStart));
     registers.set(LEN_REG, tryAsU64(serviceAccountInfoSize));
     const builder = new MemoryBuilder();
     builder.setWriteablePages(tryAsMemoryIndex(pageStart), tryAsMemoryIndex(pageStart + PAGE_SIZE));
-    const memory = new HostCallMemory(builder.finalize(tryAsMemoryIndex(0), tryAsSbrkIndex(0)));
+    const memory = HostCallMemory.new(builder.finalize(tryAsMemoryIndex(0), tryAsSbrkIndex(0)));
     const readRaw = () => {
         const result = new Uint8Array(Number(registers.get(LEN_REG)));
         assert.strictEqual(memory.loadInto(result, tryAsU64(memStart)).isOk, true);
@@ -56,7 +55,7 @@ describe("HostCalls: Info", () => {
         const serviceId = tryAsServiceId(10_000);
         const currentServiceId = serviceId;
         const accounts = new TestAccounts(currentServiceId);
-        const info = new Info(currentServiceId, accounts);
+        const info = Info.new(currentServiceId, accounts);
         const { registers, memory, readInfo } = prepareRegsAndMemory(serviceId);
         const storageUtilisationBytes = tryAsU64(10_000);
         const storageUtilisationCount = tryAsU32(1_000);
@@ -84,7 +83,7 @@ describe("HostCalls: Info", () => {
         const serviceId = tryAsServiceId(10_000);
         const currentServiceId = serviceId;
         const accounts = new TestAccounts(currentServiceId);
-        const info = new Info(currentServiceId, accounts);
+        const info = Info.new(currentServiceId, accounts);
         const { registers, memory, readRaw } = prepareRegsAndMemory(serviceId);
         registers.set(LEN_REG, tryAsU64(10));
         const storageUtilisationBytes = tryAsU64(10_000);
@@ -108,7 +107,7 @@ describe("HostCalls: Info", () => {
     it("should write none if account info is missing", async () => {
         const currentServiceId = tryAsServiceId(15_000);
         const accounts = new TestAccounts(currentServiceId);
-        const info = new Info(currentServiceId, accounts);
+        const info = Info.new(currentServiceId, accounts);
         const serviceId = tryAsServiceId(10_000);
         const { registers, memory } = prepareRegsAndMemory(serviceId);
         // when
@@ -121,7 +120,7 @@ describe("HostCalls: Info", () => {
         const serviceId = tryAsServiceId(10_000);
         const currentServiceId = serviceId;
         const accounts = new TestAccounts(currentServiceId);
-        const info = new Info(serviceId, accounts);
+        const info = Info.new(serviceId, accounts);
         const { registers, memory } = prepareRegsAndMemory(serviceId, 10);
         const storageUtilisationBytes = tryAsU64(10_000);
         const storageUtilisationCount = tryAsU32(1_000);