@proto-kit/protocol 0.1.1-develop.1086

package/test/StateTransition.test.ts

@@ -0,0 +1,174 @@
+import "reflect-metadata";
+import { InMemoryMerkleTreeStorage } from "@proto-kit/common";
+import { Bool, Field } from "o1js";
+
+import { Option, ProvableStateTransition } from "../src/index";
+import { RollupMerkleTree } from "../../common/src/trees/RollupMerkleTree.js";
+
+// TODO Not worth fixing rn because we will revamp the STProver very soon
+
+describe.skip("stateTransition", () => {
+  async function checkTransitions(
+    tree: RollupMerkleTree,
+    transitions: ProvableStateTransition[]
+  ) {
+    // const batch = StateTransitionProvableBatch.fromTransitions(transitions, []);
+    //
+    // const temporaryTree = new RollupMerkleTree(
+    //   new CachedMerkleTreeStore(tree.store)
+    // );
+    // const startRoot = temporaryTree.getRoot();
+    //
+    // const hashList = new DefaultProvableHashList(ProvableStateTransition);
+    //
+    // batch.batch.forEach((item) => {
+    //   if (item.to.isSome.toBoolean()) {
+    //     temporaryTree.setLeaf(item.path.toBigInt(), item.to.value);
+    //   }
+    //   hashList.push(item);
+    // });
+    //
+    // const endRoot = temporaryTree.getRoot();
+    //
+    // class DummySTWP implements StateTransitionWitnessProvider {
+    //   private i = 0;
+    //
+    //   public constructor(private readonly witnessTree: RollupMerkleTree) {}
+    //
+    //   public getWitness(key: Field): RollupMerkleTreeWitness {
+    //     const witness = this.witnessTree.getWitness(key.toBigInt());
+    //     const set = batch.batch[this.i];
+    //     if (set.to.isSome.toBoolean()) {
+    //       this.witnessTree.setLeaf(key.toBigInt(), set.to.value);
+    //     }
+    //     this.i += 1;
+    //     return witness;
+    //   }
+    // }
+    //
+    // const childContainer = container.createChildContainer();
+    // childContainer.registerInstance(
+    //   "StateTransitionWitnessProvider",
+    //   new DummySTWP(tree)
+    // );
+    // const prover = childContainer.resolve(StateTransitionProver);
+    //
+    // const state = prover.applyTransitions(startRoot, Field(0), batch);
+    //
+    // expect(state.stateRoot).toStrictEqual(endRoot);
+    // expect(state.stateTransitionList.commitment).toStrictEqual(
+    //   hashList.commitment
+    // );
+    //
+    // await childContainer.dispose();
+  }
+
+  it.each([
+    [
+      [
+        new ProvableStateTransition({
+          from: Option.fromValue(Field(1), Field).toProvable(),
+          to: Option.fromValue(Field(14), Field).toProvable(),
+          path: Field(1),
+        }),
+        new ProvableStateTransition({
+          from: Option.fromValue(Field(14), Field).toProvable(),
+          to: Option.fromValue(Field(4), Field).toProvable(),
+          path: Field(1),
+        }),
+      ],
+    ],
+    [
+      [
+        new ProvableStateTransition({
+          from: Option.none().toProvable(),
+          to: Option.from(Bool(true), Field(4), Field).toProvable(),
+          path: Field(1),
+        }),
+        new ProvableStateTransition({
+          from: Option.from(Bool(true), Field(5), Field).toProvable(),
+          to: Option.from(Bool(true), Field(2), Field).toProvable(),
+          path: Field(2),
+        }),
+        new ProvableStateTransition({
+          from: Option.from(Bool(true), Field(2), Field).toProvable(),
+          to: Option.none().toProvable(),
+          path: Field(2),
+        }),
+      ],
+    ],
+  ])("should pass without throwing", async (transitions) => {
+    // expect.assertions(2);
+
+    const tree = new RollupMerkleTree(new InMemoryMerkleTreeStorage());
+
+    // Is ignored because overwritten by first transition
+    tree.setLeaf(1n, Option.fromValue(Field(1), Field).treeValue);
+    tree.setLeaf(2n, Option.fromValue(Field(5), Field).treeValue);
+
+    await checkTransitions(tree, transitions);
+  });
+
+  it.each([
+    [
+      [
+        new ProvableStateTransition({
+          // fail
+          from: Option.from(Bool(true), Field(2), Field).toProvable(),
+          to: Option.none().toProvable(),
+          path: Field(1),
+        }),
+      ],
+      0,
+    ],
+    [
+      [
+        new ProvableStateTransition({
+          // success
+          from: Option.from(Bool(true), Field(1), Field).toProvable(),
+          to: Option.from(Bool(true), Field(14), Field).toProvable(),
+          path: Field(1),
+        }),
+        new ProvableStateTransition({
+          // fail
+          from: Option.from(Bool(true), Field(6), Field).toProvable(),
+          to: Option.none().toProvable(),
+          path: Field(2),
+        }),
+      ],
+      1,
+    ],
+    [
+      [
+        new ProvableStateTransition({
+          // success
+          from: Option.from(Bool(true), Field(1), Field).toProvable(),
+          to: Option.from(Bool(true), Field(14), Field).toProvable(),
+          path: Field(1),
+        }),
+        new ProvableStateTransition({
+          // fail
+          from: Option.from(Bool(true), Field(15), Field).toProvable(),
+          to: Option.none().toProvable(),
+          path: Field(1),
+        }),
+      ],
+      1,
+    ],
+  ])(
+    "should throw because of failing precondition",
+    async (transitions, index) => {
+      expect.assertions(1);
+
+      const tree = new RollupMerkleTree(new InMemoryMerkleTreeStorage());
+
+      // Is ignored because overwritten by first transition
+      tree.setLeaf(1n, Option.fromValue(Field(1), Field).treeValue);
+      tree.setLeaf(2n, Option.fromValue(Field(5), Field).treeValue);
+
+      await expect(checkTransitions(tree, transitions)).rejects.toThrow(
+        `MerkleWitness not valid for StateTransition (${index})`
+      );
+    }
+  );
+});

package/test/TestingProtocol.ts

@@ -0,0 +1,47 @@
+import { WithZkProgrammable, ZkProgrammable } from "@proto-kit/common";
+import { container } from "tsyringe";
+
+import {
+  AccountStateHook,
+  BlockHeightHook,
+  BlockProver,
+  LastStateRootBlockHook,
+  MethodPublicOutput,
+  NoOpStateTransitionWitnessProvider,
+  Protocol,
+  StateTransitionProver,
+} from "../src";
+
+class RuntimeMock implements WithZkProgrammable<undefined, MethodPublicOutput> {
+  zkProgrammable: ZkProgrammable<undefined, MethodPublicOutput> =
+    undefined as unknown as ZkProgrammable<undefined, MethodPublicOutput>;
+}
+
+export function createAndInitTestingProtocol() {
+  const ProtocolClass = Protocol.from({
+    modules: {
+      StateTransitionProver: StateTransitionProver,
+      BlockProver: BlockProver,
+      AccountState: AccountStateHook,
+      BlockHeight: BlockHeightHook,
+      LastStateRoot: LastStateRootBlockHook,
+    },
+  });
+  const protocol = new ProtocolClass();
+
+  protocol.configure({
+    BlockProver: {},
+    AccountState: {},
+    BlockHeight: {},
+    StateTransitionProver: {},
+    LastStateRoot: {},
+  });
+  protocol.create(() => container.createChildContainer());
+
+  protocol.registerValue({
+    StateTransitionWitnessProvider: new NoOpStateTransitionWitnessProvider(),
+    Runtime: new RuntimeMock(),
+  });
+
+  return protocol;
+}

package/test/model/Option.test.ts

@@ -0,0 +1,72 @@
+import "reflect-metadata";
+
+import { Bool, Field, Poseidon } from "o1js";
+
+import { Option, ProvableOption } from "../../src";
+
+describe("option", () => {
+  it.each([
+    [Bool(true), Field(0), Field, true],
+    [Bool(false), Field(0), Field, false],
+  ])(
+    "should serialize to the correct tree value",
+    (isSome, value, valueType, shouldHash) => {
+      expect.assertions(1);
+
+      const option = Option.from(isSome, value, valueType);
+      const treeValue = shouldHash ? Poseidon.hash(value.toFields()) : Field(0);
+
+      expect(option.treeValue.toString()).toStrictEqual(treeValue.toString());
+    }
+  );
+
+  it.each([
+    [
+      Option.from(Bool(true), Field(5), Field),
+      { isSome: true, value: Poseidon.hash([Field(5)]) },
+      false,
+    ],
+    [
+      Option.from(Bool(true), Field(0), Field),
+      { isSome: true, value: Poseidon.hash([Field(0)]) },
+      false,
+    ],
+    [
+      Option.from(Bool(false), Field(0), Field),
+      { isSome: true, value: Field(0) },
+      true,
+    ],
+    [
+      Option.from(Bool(false), Field(1), Field),
+      { isSome: false, value: Field(0) },
+      false,
+    ],
+  ])(
+    "should encode to correct provable",
+    (option, provableTemplate, forceSome) => {
+      expect.assertions(4);
+
+      if (forceSome) {
+        option.forceSome();
+      }
+
+      const provableInput = new ProvableOption({
+        isSome: Bool(provableTemplate.isSome),
+        value: provableTemplate.value,
+      });
+
+      const provable = option.toProvable();
+      const fields = ProvableOption.toFields(provable);
+
+      expect(provable.value.toBigInt()).toStrictEqual(
+        provableInput.value.toBigInt()
+      );
+      expect(provable.isSome.toBoolean()).toStrictEqual(
+        provableInput.isSome.toBoolean()
+      );
+
+      expect(fields[0]).toStrictEqual(provableInput.isSome.toFields()[0]);
+      expect(fields[1]).toStrictEqual(provableInput.value);
+    }
+  );
+});

package/test/state/assert/assert.test.ts

@@ -0,0 +1,56 @@
+import "reflect-metadata";
+
+import { Bool } from "o1js";
+import { container } from "tsyringe";
+
+import { assert } from "../../../src/state/assert/assert";
+import { RuntimeMethodExecutionContext } from "../../../src/state/context/RuntimeMethodExecutionContext";
+import { RuntimeTransaction } from "../../../src/model/transaction/RuntimeTransaction";
+import { NetworkState } from "../../../src/model/network/NetworkState";
+
+describe("assert", () => {
+  const defaultStatusMessage = "something went wrong";
+  const executionContext = container.resolve(RuntimeMethodExecutionContext);
+
+  beforeEach(() => {
+    executionContext.beforeMethod("testConstructor", "test", []);
+
+    executionContext.setup({
+      transaction: undefined as unknown as RuntimeTransaction,
+      networkState: undefined as unknown as NetworkState,
+    });
+  });
+
+  afterEach(() => {
+    executionContext.afterMethod();
+  });
+
+  describe.each([
+    [true, undefined],
+    [false, undefined],
+    [false, defaultStatusMessage],
+  ])("status and message propagation", (status, statusMessage) => {
+    it("should propagate the assertion status and message", () => {
+      expect.assertions(2);
+
+      assert(Bool(status), statusMessage);
+
+      const { status: resultStatus, statusMessage: resultStatusMessage } =
+        executionContext.current().result;
+
+      expect(status).toBe(resultStatus.toBoolean());
+      expect(statusMessage).toBe(resultStatusMessage);
+    });
+  });
+
+  it("should keep a false status, once it was already set", () => {
+    expect.assertions(1);
+
+    assert(Bool(false));
+    assert(Bool(true));
+
+    const { status } = executionContext.current().result;
+
+    expect(status.toBoolean()).toBe(false);
+  });
+});

package/test/tsconfig.json

@@ -0,0 +1,7 @@
+{
+  "extends": "./../../../tsconfig.json",
+  "compilerOptions": {
+    "experimentalDecorators": true
+  },
+  "include": ["../src/**/*.test.ts", "./**/*.ts", "./*.ts", "../src/**/*.ts"]
+}

package/test/utils/ProvableReductionHashList.test.ts

@@ -0,0 +1,117 @@
+import "reflect-metadata";
+import { Bool, Field, Poseidon } from "o1js";
+
+import {
+  StateTransitionReductionList,
+  reduceStateTransitions,
+} from "../../src/utils/StateTransitionReductionList";
+import {
+  DefaultProvableHashList,
+  ProvableStateTransition,
+  StateTransition,
+  Option,
+} from "../../src";
+
+interface UnprovableStateTransition {
+  path: number;
+  from: number;
+  to: { isSome: boolean; value: number };
+}
+
+describe("provableReductionHashList", () => {
+  function mapToProvable(
+    sts: UnprovableStateTransition[]
+  ): StateTransition<unknown>[] {
+    return sts.map(
+      (st) =>
+        new StateTransition(
+          Field(st.path),
+          new Option(Bool(true), Field(st.from), Field, Bool(false)),
+          new Option(Bool(st.to.isSome), Field(st.to.value), Field, Bool(false))
+        ) as StateTransition<unknown>
+    );
+  }
+
+  describe.each<[UnprovableStateTransition[], number]>([
+    [
+      [
+        { path: 1, from: 5, to: { isSome: true, value: 10 } },
+        { path: 1, from: 10, to: { isSome: true, value: 15 } },
+      ],
+      1,
+    ],
+    [
+      [
+        { path: 1, from: 5, to: { isSome: true, value: 10 } },
+        { path: 1, from: 10, to: { isSome: false, value: 0 } },
+      ],
+      1,
+    ],
+    [
+      [
+        { path: 1, from: 5, to: { isSome: true, value: 10 } },
+        { path: 1, from: 10, to: { isSome: false, value: 0 } },
+        { path: 1, from: 10, to: { isSome: false, value: 0 } },
+        { path: 5, from: 15, to: { isSome: false, value: 0 } },
+        { path: 5, from: 15, to: { isSome: true, value: 20 } },
+        { path: 1, from: 10, to: { isSome: false, value: 0 } },
+        { path: 5, from: 20, to: { isSome: true, value: 25 } },
+        { path: 5, from: 25, to: { isSome: true, value: 30 } },
+      ],
+      4,
+    ],
+  ])("should reduce and append correctly", (sts, reducedLength) => {
+    it("start from empty list", () => {
+      const reductionList = new StateTransitionReductionList(
+        ProvableStateTransition
+      );
+      const normalList = new DefaultProvableHashList(ProvableStateTransition);
+
+      const provableSTs = mapToProvable(sts);
+
+      provableSTs.forEach((st) => {
+        reductionList.push(st.toProvable());
+      });
+
+      const reduced = reduceStateTransitions(provableSTs);
+      reduced.forEach((st) => {
+        normalList.push(st.toProvable());
+      });
+
+      expect(reduced).toHaveLength(reducedLength);
+      expect(normalList.commitment.toBigInt()).toStrictEqual(
+        reductionList.commitment.toBigInt()
+      );
+      expect(reductionList.unconstrainedList).toHaveLength(reduced.length);
+    });
+
+    it("start from non-zero commitment - skip first reduction", () => {
+      const commitment = Poseidon.hash([Field(123_456)]);
+      const reductionList = new StateTransitionReductionList(
+        ProvableStateTransition,
+        commitment
+      );
+      const normalList = new DefaultProvableHashList(
+        ProvableStateTransition,
+        commitment
+      );
+
+      const provableSTs = mapToProvable(sts);
+
+      provableSTs.forEach((st) => {
+        reductionList.push(st.toProvable());
+      });
+
+      const reduced = reduceStateTransitions(provableSTs);
+      reduced.forEach((st) => {
+        normalList.push(st.toProvable());
+      });
+
+      expect(reduced).toHaveLength(reducedLength);
+      expect(normalList.commitment.toBigInt()).toStrictEqual(
+        reductionList.commitment.toBigInt()
+      );
+      expect(reductionList.unconstrainedList).toHaveLength(reduced.length);
+    });
+  });
+});

package/test/utils.test.ts

@@ -0,0 +1,27 @@
+import "reflect-metadata";
+
+import { stringToField } from "../src";
+
+describe("stringToField", () => {
+  const stringToFieldInputs: string[] = [];
+
+  [31, 32, 33, 63, 64, 65, 1000].forEach((length) => {
+    stringToFieldInputs.push(
+      Array.from({ length })
+        .map(() => "A")
+        .reduce((a, b) => a + b)
+    );
+  });
+
+  it.each(
+    ["", "a", "Helloasdlsdaglsdiousdioagiosadgoisaudogiusadogusoadgds"].concat(
+      stringToFieldInputs
+    )
+  )("should encode without errors", (input) => {
+    expect.assertions(1);
+
+    const field = stringToField(input);
+
+    expect(field.toBigInt()).toBeGreaterThan(0n);
+  });
+});

package/tsconfig.json

@@ -0,0 +1,8 @@
+{
+  "extends": "./../../tsconfig.json",
+  "compilerOptions": {
+    "outDir": "./dist"
+  },
+  "include": ["./src/index.ts"],
+  "exclude": ["./dist/**/*.ts"]
+}