1const chai = require("chai");
 2const path = require("path");
 3const snarkjs = require("snarkjs");
 4const compiler = require("circom");
 5
 6const mimcjs = require("../src/mimcsponge.js");
 7
 8const assert = chai.assert;
 9
10describe("MiMC Sponge Circuit test", function () {
11    let circuit;
12
13    this.timeout(100000);
14
15    it("Should check permutation", async () => {
16        const cirDef = await compiler(path.join(__dirname, "circuits", "mimc_sponge_test.circom"));
17
18        circuit = new snarkjs.Circuit(cirDef);
19
20        console.log("MiMC Feistel constraints: " + circuit.nConstraints);
21
22        const w = circuit.calculateWitness({xL_in: 1, xR_in: 2, k: 3});
23
24        const xLout = w[circuit.getSignalIdx("main.xL_out")];
25        const xRout = w[circuit.getSignalIdx("main.xR_out")];
26
27        const out2 = mimcjs.hash(1,2,3);
28
29        assert.equal(xLout.toString(), out2.xL.toString());
30        assert.equal(xRout.toString(), out2.xR.toString());
31
32        assert(circuit.checkWitness(w));
33
34    });
35
36    it("Should check hash", async () => {
37        const cirDef = await compiler(path.join(__dirname, "circuits", "mimc_sponge_hash_test.circom"));
38
39        circuit = new snarkjs.Circuit(cirDef);
40
41        console.log("MiMC Sponge constraints: " + circuit.nConstraints);
42
43        const w = circuit.calculateWitness({ins: [1, 2], k: 0});
44
45        const o1 = w[circuit.getSignalIdx("main.outs[0]")];
46        const o2 = w[circuit.getSignalIdx("main.outs[1]")];
47        const o3 = w[circuit.getSignalIdx("main.outs[2]")];
48
49        const out2 = mimcjs.multiHash([1,2], 0, 3);
50
51        assert.equal(o1.toString(), out2[0].toString());
52        assert.equal(o2.toString(), out2[1].toString());
53        assert.equal(o3.toString(), out2[2].toString());
54
55        assert(circuit.checkWitness(w));
56
57    });
58});