arnaucube
arnaucube/circomlibPublic Self-hosted repository
circomlib/test/poseidoncontract.js
test/poseidoncontract.jsRaw
 1const ganache = require("ganache-cli");
 2const Web3 = require("web3");
 3const chai = require("chai");
 4const poseidonGenContract = require("../src/poseidon_gencontract.js");
 5const poseidon = require("../src/poseidon.js");
 6
 7const assert = chai.assert;
 8const log = (msg) => { if (process.env.MOCHA_VERBOSE) console.log(msg); };
 9
10describe("Poseidon Smart contract test", function () {
11    let testrpc;
12    let web3;
13    let poseidon6;
14    let poseidon3;
15    let accounts;
16    this.timeout(100000);
17
18    before(async () => {
19        web3 = new Web3(ganache.provider(), null, { transactionConfirmationBlocks: 1 });
20        accounts = await web3.eth.getAccounts();
21    });
22
23    it("Should deploy the contract", async () => {
24        const C6 = new web3.eth.Contract(poseidonGenContract.generateABI(5));
25        const C3 = new web3.eth.Contract(poseidonGenContract.generateABI(2));
26
27        poseidon6 = await C6.deploy({
28            data: poseidonGenContract.createCode(5)
29        }).send({
30            gas: 5000000,
31            from: accounts[0]
32        });
33        poseidon3 = await C3.deploy({
34            data: poseidonGenContract.createCode(2)
35        }).send({
36            gas: 5000000,
37            from: accounts[0]
38        });
39    });
40
41    it("Should calculate the poseidon correctly t=6", async () => {
42
43        const res = await poseidon6.methods.poseidon([1,2, 0, 0, 0]).call();
44
45        // console.log("Cir: " + bigInt(res.toString(16)).toString(16));
46
47        const res2 = poseidon([1,2, 0, 0, 0]);
48        // console.log("Ref: " + bigInt(res2).toString(16));
49
50        assert.equal(res.toString(), res2.toString());
51    });
52    it("Should calculate the poseidon correctly t=3", async () => {
53
54        const res = await poseidon3.methods.poseidon([1,2]).call();
55
56        // console.log("Cir: " + bigInt(res.toString(16)).toString(16));
57
58        const res2 = poseidon([1,2]);
59        // console.log("Ref: " + bigInt(res2).toString(16));
60
61        assert.equal(res.toString(), res2.toString());
62    });
63
64});
65