1const chai = require("chai");
 2const path = require("path");
 3const zkSnark = require("zksnark");
 4const compiler = require("circom");
 5
 6const assert = chai.assert;
 7
 8const bigInt = require("big-integer");
 9
10
11describe("Baby Jub test", () => {
12    it("Should add point (0,1) and (0,1)", async () => {
13
14        const cirDef = await compiler(path.join(__dirname, "circuits", "babyadd_tester.circom"));
15
16//        console.log(JSON.stringify(cirDef, null, 1));
17
18//        assert.equal(cirDef.nVars, 2);
19
20        const circuit = new zkSnark.Circuit(cirDef);
21
22        console.log("NConstrains: " + circuit.nConstraints);
23
24        const input={
25            x1: zkSnark.bigInt(0),
26            y1: zkSnark.bigInt(1),
27            x2: zkSnark.bigInt(0),
28            y2: zkSnark.bigInt(1)
29        }
30
31        const w = circuit.calculateWitness(input);
32
33        const xout = w[circuit.getSignalIdx("main.xout")];
34        const yout = w[circuit.getSignalIdx("main.yout")];
35
36        assert(xout.equals(0));
37        assert(yout.equals(1));
38    });
39
40    it("Should add 2 same numbers", async () => {
41
42        const cirDef = await compiler(path.join(__dirname, "circuits", "babyadd_tester.circom"));
43
44//        console.log(JSON.stringify(cirDef, null, 1));
45
46//        assert.equal(cirDef.nVars, 2);
47
48        const circuit = new zkSnark.Circuit(cirDef);
49
50        console.log("NConstrains: " + circuit.nConstraints);
51
52        const input={
53            x1: zkSnark.bigInt("17777552123799933955779906779655732241715742912184938656739573121738514868268"),
54            y1: zkSnark.bigInt("2626589144620713026669568689430873010625803728049924121243784502389097019475"),
55            x2: zkSnark.bigInt("17777552123799933955779906779655732241715742912184938656739573121738514868268"),
56            y2: zkSnark.bigInt("2626589144620713026669568689430873010625803728049924121243784502389097019475")
57        }
58
59        const w = circuit.calculateWitness(input);
60
61        const xout = w[circuit.getSignalIdx("main.xout")];
62        const yout = w[circuit.getSignalIdx("main.yout")];
63
64        assert(xout.equals(zkSnark.bigInt("6890855772600357754907169075114257697580319025794532037257385534741338397365")));
65        assert(yout.equals(zkSnark.bigInt("4338620300185947561074059802482547481416142213883829469920100239455078257889")));
66    });
67
68    it("Should add 2 different numbers", async () => {
69
70        const cirDef = await compiler(path.join(__dirname, "circuits", "babyadd_tester.circom"));
71
72//        console.log(JSON.stringify(cirDef, null, 1));
73
74//        assert.equal(cirDef.nVars, 2);
75
76        const circuit = new zkSnark.Circuit(cirDef);
77
78        console.log("NConstrains: " + circuit.nConstraints);
79
80        const input={
81            x1: zkSnark.bigInt("17777552123799933955779906779655732241715742912184938656739573121738514868268"),
82            y1: zkSnark.bigInt("2626589144620713026669568689430873010625803728049924121243784502389097019475"),
83            x2: zkSnark.bigInt("16540640123574156134436876038791482806971768689494387082833631921987005038935"),
84            y2: zkSnark.bigInt("20819045374670962167435360035096875258406992893633759881276124905556507972311")
85        }
86
87        const w = circuit.calculateWitness(input);
88
89        const xout = w[circuit.getSignalIdx("main.xout")];
90        const yout = w[circuit.getSignalIdx("main.yout")];
91
92        console.log(xout.toString());
93        console.log(yout.toString());
94
95        assert(xout.equals(zkSnark.bigInt("7916061937171219682591368294088513039687205273691143098332585753343424131937")));
96        assert(yout.equals(zkSnark.bigInt("14035240266687799601661095864649209771790948434046947201833777492504781204499")));
97    });
98});