1const chai = require("chai");
 2const path = require("path");
 3var blake2b = require("blake2b");
 4const tester = require("circom").tester;
 5
 6const poseidon = require("../src/poseidon.js");
 7
 8const assert = chai.assert;
 9
10describe("Blake2b version test", function() {
11    it("Should give the expected output for blake2b version", async () => {
12        var output = new Uint8Array(32);
13        var input = Buffer.from("poseidon_constants");
14        const h = blake2b(output.length).update(input).digest("hex");
15        assert.equal("e57ba154fb2c47811dc1a2369b27e25a44915b4e4ece4eb8ec74850cb78e01b1", h);
16    });
17});
18
19describe("Poseidon Circuit test", function () {
20    let circuit6;
21    let circuit3;
22
23    this.timeout(100000);
24
25    before( async () => {
26        circuit6 = await tester(path.join(__dirname, "circuits", "poseidon6_test.circom"));
27        circuit3 = await tester(path.join(__dirname, "circuits", "poseidon3_test.circom"));
28    });
29
30    it("Should check constrain of hash([1, 2]) t=6", async () => {
31        const w = await circuit6.calculateWitness({inputs: [1, 2]}, true);
32
33        const hash = poseidon.createHash(6, 8, 57);
34
35        const res2 = hash([1,2]);
36        assert.equal("12242166908188651009877250812424843524687801523336557272219921456462821518061", res2.toString());
37        await circuit6.assertOut(w, {out : res2});
38        await circuit6.checkConstraints(w);
39    });
40
41    it("Should check constrain of hash([3, 4]) t=6", async () => {
42        const w = await circuit6.calculateWitness({inputs: [3, 4]});
43
44        const hash = poseidon.createHash(6, 8, 57);
45
46        const res2 = hash([3, 4]);
47
48        assert.equal("17185195740979599334254027721507328033796809509313949281114643312710535000993", res2.toString());
49        await circuit6.assertOut(w, {out : res2});
50        await circuit6.checkConstraints(w);
51    });
52
53
54    it("Should check constrain of hash([1, 2]) t=3", async () => {
55        const w = await circuit3.calculateWitness({inputs: [1, 2]});
56
57        const hash = poseidon.createHash(3, 8, 57);
58
59        const res2 = hash([1,2]);
60        assert.equal("2104035019328376391822106787753454168168617545136592089411833517434990977743", res2.toString());
61        await circuit3.assertOut(w, {out : res2});
62        await circuit3.checkConstraints(w);
63    });
64
65    it("Should check constrain of hash([3, 4]) t=3", async () => {
66        const w = await circuit3.calculateWitness({inputs: [3, 4]});
67
68        const hash = poseidon.createHash(3, 8, 57);
69
70        const res2 = hash([3, 4]);
71        assert.equal("12456141564250880945411182508630957604732712316993112736876413121277158512223", res2.toString());
72        await circuit3.assertOut(w, {out : res2});
73        await circuit3.checkConstraints(w);
74    });
75});