circomlib/test/poseidoncontract.js

const ganache = require("ganache-cli");
const Web3 = require("web3");
const chai = require("chai");
const poseidonGenContract = require("../src/poseidon_gencontract.js");
const poseidon = require("../src/poseidon.js");

const assert = chai.assert;
const log = (msg) => { if (process.env.MOCHA_VERBOSE) console.log(msg); };

describe("Poseidon Smart contract test", function () {
    let testrpc;
    let web3;
    let poseidon2;
    let poseidon4;
    let accounts;
    this.timeout(100000);
    let C2;
    let C4;

    before(async () => {
        web3 = new Web3(ganache.provider(), null, { transactionConfirmationBlocks: 1 });
        accounts = await web3.eth.getAccounts();
    });

    it("Should deploy the contract", async () => {
        
        C2 = new web3.eth.Contract(poseidonGenContract.generateABI(2));
        poseidon2 = await C2.deploy({
            data: poseidonGenContract.createCode(2)
        }).send({
            gas: 5000000,
            from: accounts[0]
        });
        
        C4 = new web3.eth.Contract(poseidonGenContract.generateABI(4));
        poseidon4 = await C4.deploy({
            data: poseidonGenContract.createCode(4)
        }).send({
            gas: 5000000,
            from: accounts[0]
        });
    });

    it("Should calculate the poseidon correctly for 2 inputs", async () => {
        const res = await poseidon2.methods.poseidon([1, 2]).call();

        // console.log("Cir: " + bigInt(res.toString(16)).toString(16));

        const res2 = poseidon([1, 2]);
        // console.log("Ref: " + bigInt(res2).toString(16));

        assert.equal(res.toString(), res2.toString());
    });
    it("Should calculate the poseidon correctly for 4 inputs", async () => {

        const res = await poseidon4.methods.poseidon([1, 2, 3, 4]).call();

        // console.log("Cir: " + bigInt(res.toString(16)).toString(16));

        const res2 = poseidon([1, 2, 3, 4]);
        // console.log("Ref: " + bigInt(res2).toString(16));

        assert.equal(res.toString(), res2.toString());
    });

});