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