@arbitrum/nitro-contracts 1.0.0-beta.1

package/test/prover/one-step-proof.ts

@@ -0,0 +1,93 @@
+import { ethers, run, getNamedAccounts } from "hardhat"
+import fs from "fs";
+import assert from "assert";
+import readline from 'readline';
+
+const PARALLEL = 128;
+
+async function sendTestMessages() {
+  const { deployer } = await getNamedAccounts();
+  const inbox = await ethers.getContract("InboxStub", deployer);
+  const seqInbox = await ethers.getContract("SequencerInboxStub", deployer);
+  const msgRoot = "../arbitrator/prover/test-cases/rust/data/";
+  const gasOpts = { gasLimit: ethers.utils.hexlify(250000), gasPrice: ethers.utils.parseUnits('5', "gwei") };
+  for (let msgNum = 0; msgNum < 2; msgNum++) {
+    const path = msgRoot + "msg" + String(msgNum) + ".bin";
+    const buf = fs.readFileSync(path);
+    await inbox.sendL2MessageFromOrigin(buf, gasOpts);
+    await seqInbox.addSequencerL2BatchFromOrigin(msgNum, buf, 0, ethers.constants.AddressZero, gasOpts);
+  }
+}
+
+describe("OneStepProof", function () {
+  const root = "./test/prover/proofs/";
+  const dir = fs.readdirSync(root);
+
+  before(async function () {
+    await run("deploy", { "tags": "OneStepProofEntry" });
+    await run("deploy", { "tags": "SequencerInboxStub" });
+    await run("deploy", { "tags": "InboxStub" });
+    await sendTestMessages();
+  })
+
+  it("should deploy test harness", function() {})
+
+  for (let file of dir) {
+    if (!file.endsWith(".json")) continue;
+    it("Should pass " + file + " proofs", async function () {
+      let path = root + file;
+      let proofs = JSON.parse(fs.readFileSync(path).toString('utf8'));
+      const osp = await ethers.getContract("OneStepProofEntry");
+      const seqInbox = await ethers.getContract("SequencerInboxStub");
+      const bridge = await ethers.getContract("BridgeStub");
+
+      const promises = [];
+      const isdone = [];
+      for (let i = 0; i < proofs.length; i++) {
+        process.stdout.write("\rTesting " + file + " proof " + i + "/" + proofs.length + " ");
+        const proof = proofs[i];
+        isdone.push(false);
+        const inboxLimit = 1000000;
+        const promise = osp.proveOneStep([inboxLimit, seqInbox.address, bridge.address], i, [...Buffer.from(proof.before, "hex")], [...Buffer.from(proof.proof, "hex")])
+          .catch((err: any) => {
+            console.error("Error executing proof " + i);
+            throw err;
+          })
+          .then((after: any) => assert.equal(after, "0x" + proof.after, "After state doesn't match after proof " + i))
+          .finally((_: any) => {isdone[i] = true});
+        if (promises.length < PARALLEL) {
+          promises.push(promise);
+        } else {
+          const finished: any = await Promise.race(promises.map((p, k) => p.then((_: any) => k)));
+          promises[finished] = promise;
+        }
+      }
+
+      let stillWaiting = []
+      do {
+        const finished: any = await Promise.race(promises.map((p, k) => p.then((_: any) => k)));
+        if (finished == promises.length - 1) {
+          promises.pop()
+        } else {
+          promises[finished] = promises.pop()
+        }
+        stillWaiting = [];
+        for (let i = 0; i < isdone.length; i++) {
+          if (!isdone[i]) {
+            stillWaiting.push(i)
+          }
+        }
+        readline.clearLine(process.stdout, 0);
+        process.stdout.write("\rTesting " + file + " Waiting for: " + String(stillWaiting.length) + "/" + String(isdone.length));
+        if (stillWaiting.length < 10) {
+          process.stdout.write(": ")
+          for (let i = 0; i < stillWaiting.length; i++) {
+            process.stdout.write(String(stillWaiting[i]) + ",");
+          }
+        }
+      } while (stillWaiting.length > 0);
+      await Promise.all(promises);
+      process.stdout.write("\r");
+    });
+  }
+});

package/test/prover/value-arrays.ts

@@ -0,0 +1,11 @@
+import { ethers, run } from "hardhat";
+
+describe("ValueArray", function () {
+  it("Should pass ValueArrayTester", async function () {
+    await run("deploy", { tags: "ValueArrayTester" });
+
+    const valueArrayTester = await ethers.getContract("ValueArrayTester");
+
+    await valueArrayTester.test();
+  });
+});

package/tsconfig.json

@@ -0,0 +1,13 @@
+{
+    "compilerOptions": {
+      "target": "es2018",
+      "module": "commonjs",
+      "strict": true,
+      "esModuleInterop": true,
+      "outDir": "dist",
+      "resolveJsonModule": true,
+      "allowJs": true
+    },
+    "include": ["./scripts", "./test", "./typechain-types"],
+    "files": ["./hardhat.config.ts"]
+}