mirror of
https://github.com/arnaucube/miksi-core.git
synced 2026-02-06 19:16:40 +01:00
Add SmartContracts test flow with 3 deposits
This commit is contained in:
arnaucube
@@ -61,6 +61,7 @@ describe("deposit test", function () {
|
||||
"amount": amount,
|
||||
"secret": secret,
|
||||
"nullifier": nullifier,
|
||||
"oldKey": "1",
|
||||
"siblingsOld": siblingsOld,
|
||||
"siblingsNew": siblingsNew,
|
||||
"rootOld": rootOld,
|
||||
|
||||
@@ -13,6 +13,29 @@ const WitnessCalculatorBuilder = require("circom_runtime").WitnessCalculatorBuil
|
||||
const circomlib = require("circomlib");
|
||||
const smt = require("circomlib").smt;
|
||||
|
||||
let insVerifier;
|
||||
let insMiksi;
|
||||
|
||||
const nLevels = 5;
|
||||
const secret = ["1234567890", "987654321", "123"];
|
||||
|
||||
const coinCode = "0"; // refearing to ETH
|
||||
const ethAmount = '1';
|
||||
const amount = web3.utils.toWei(ethAmount, 'ether');
|
||||
const nullifier = ["567891234", "432198765", "321"];
|
||||
let commitment = [];
|
||||
let tree;
|
||||
let oldKey = [];
|
||||
let siblingsOld = [];
|
||||
let siblingsNew = [];
|
||||
let rootOld = [];
|
||||
let rootNew = [];
|
||||
// let commitment = [];
|
||||
let proof = [];
|
||||
let publicSignals = [];
|
||||
let commitmentsArray = [];
|
||||
let u = 0;
|
||||
|
||||
contract("miksi", (accounts) => {
|
||||
|
||||
|
||||
@@ -21,27 +44,9 @@ contract("miksi", (accounts) => {
|
||||
1: addr1, // used for the deposit
|
||||
2: addr2, // used for the withdraw
|
||||
3: addr3,
|
||||
4: addr4,
|
||||
} = accounts;
|
||||
|
||||
let insVerifier;
|
||||
let insMiksi;
|
||||
|
||||
const nLevels = 5;
|
||||
const secret = "1234567890";
|
||||
|
||||
const coinCode = "0"; // refearing to ETH
|
||||
const ethAmount = '1';
|
||||
const amount = web3.utils.toWei(ethAmount, 'ether');
|
||||
const nullifier = "567891234";
|
||||
let tree;
|
||||
let siblingsOld;
|
||||
let siblingsNew;
|
||||
let rootOld;
|
||||
let rootNew;
|
||||
let commitment;
|
||||
let proof;
|
||||
let publicSignals;
|
||||
let commitmentsArray;
|
||||
|
||||
before(async () => {
|
||||
insDepositVerifier = await DepositVerifier.new();
|
||||
@@ -54,35 +59,126 @@ contract("miksi", (accounts) => {
|
||||
// console.log("Balance at " + addr1, web3.utils.fromWei(balance_wei, 'ether'));
|
||||
expect(balance_wei).to.be.equal('100000000000000000000');
|
||||
|
||||
const poseidon = circomlib.poseidon.createHash(6, 8, 57);
|
||||
commitment = poseidon([coinCode, amount, secret, nullifier]).toString();
|
||||
|
||||
// deposit
|
||||
// add commitment into SMT
|
||||
tree = await smt.newMemEmptyTrie();
|
||||
await tree.insert(1, 0);
|
||||
|
||||
rootOld = tree.root;
|
||||
const resC = await tree.find(commitment);
|
||||
assert(!resC.found);
|
||||
siblingsOld = resC.siblings;
|
||||
while (siblingsOld.length < nLevels) {
|
||||
siblingsOld.push("0");
|
||||
};
|
||||
await computeTree(0);
|
||||
|
||||
await tree.insert(commitment, 0);
|
||||
rootNew = tree.root;
|
||||
|
||||
expect(rootOld.toString()).to.be.equal('11499909227292257605992378629333104385616480982267969744564817844870636870870');
|
||||
expect(rootNew.toString()).to.be.equal('9328869343897770565751281504295758914771207504252217956739346620422361279598');
|
||||
expect(rootOld[0].toString()).to.be.equal('11499909227292257605992378629333104385616480982267969744564817844870636870870');
|
||||
expect(rootNew[0].toString()).to.be.equal('9328869343897770565751281504295758914771207504252217956739346620422361279598');
|
||||
});
|
||||
|
||||
it("Make the deposit", async () => {
|
||||
const resC = await tree.find(commitment);
|
||||
it("Make first deposit", async () => {
|
||||
await makeDeposit(0, addr1);
|
||||
balance_wei = await web3.eth.getBalance(addr1);
|
||||
// console.log("Balance at " + addr1, web3.utils.fromWei(balance_wei, 'ether'));
|
||||
// expect(balance_wei).to.be.equal('98993526980000000000');
|
||||
});
|
||||
it("Make second deposit", async () => {
|
||||
await computeTree(1);
|
||||
await makeDeposit(1, addr3);
|
||||
});
|
||||
it("Make 3rd deposit", async () => {
|
||||
await computeTree(2);
|
||||
await makeDeposit(2, addr3);
|
||||
});
|
||||
|
||||
it("Get the commitments data", async () => {
|
||||
// getCommitments data
|
||||
let res = await insMiksi.getCommitments();
|
||||
expect(res[0][0].toString()).to.be.equal('189025084074544266465422070282645213792582195466360448472858620722286781863');
|
||||
// expect(res[1].toString()).to.be.equal('9328869343897770565751281504295758914771207504252217956739346620422361279598');
|
||||
console.log(res[0]);
|
||||
commitmentsArray[0] = res[0];
|
||||
});
|
||||
|
||||
it("Rebuild the tree from sc commitments", async () => {
|
||||
let treeTmp = await smt.newMemEmptyTrie();
|
||||
await treeTmp.insert(1, 0);
|
||||
for (let i=0; i<commitmentsArray[0].length; i++) {
|
||||
await treeTmp.insert(commitmentsArray[0][i], 0);
|
||||
}
|
||||
expect(treeTmp.root).to.be.equal(tree.root);
|
||||
});
|
||||
|
||||
it("Calculate witness and generate the zkProof", async () => {
|
||||
await genZKProof(0, addr2);
|
||||
await genZKProof(1, addr4);
|
||||
await genZKProof(2, addr4);
|
||||
});
|
||||
|
||||
it("Try to use the zkProof with another address and get revert", async () => {
|
||||
// console.log("Try to reuse the zkproof and expect revert");
|
||||
await truffleAssert.fails(
|
||||
withdrawSC(0, addr1),
|
||||
truffleAssert.ErrorType.REVERT,
|
||||
"zkProof withdraw could not be verified"
|
||||
);
|
||||
});
|
||||
|
||||
it("Withdraw 1 ETH with the zkProof of the 1st deposit to addr2", async () => {
|
||||
// withdraw
|
||||
// console.log("Withdraw of " + ethAmount + " ETH to " + addr2);
|
||||
let resW = await withdrawSC(0, addr2);
|
||||
// console.log("resW", resW);
|
||||
|
||||
balance_wei = await web3.eth.getBalance(addr2);
|
||||
// console.log("Balance at " + addr2, web3.utils.fromWei(balance_wei, 'ether'));
|
||||
expect(balance_wei).to.be.equal('101000000000000000000');
|
||||
});
|
||||
|
||||
it("Try to reuse the zkProof and get revert", async () => {
|
||||
// console.log("Try to reuse the zkproof and expect revert");
|
||||
await truffleAssert.fails(
|
||||
withdrawSC(0, addr2),
|
||||
truffleAssert.ErrorType.REVERT,
|
||||
"nullifier already used"
|
||||
);
|
||||
balance_wei = await web3.eth.getBalance(addr2);
|
||||
expect(balance_wei).to.be.equal('101000000000000000000');
|
||||
});
|
||||
it("Withdraw 1 ETH with the zkProof of the 2nd deposit to addr4", async () => {
|
||||
let resW = await withdrawSC(1, addr4);
|
||||
balance_wei = await web3.eth.getBalance(addr4);
|
||||
expect(balance_wei).to.be.equal('101000000000000000000');
|
||||
});
|
||||
it("Withdraw 1 ETH with the zkProof of the 3rd deposit to addr4", async () => {
|
||||
let resW = await withdrawSC(2, addr4);
|
||||
balance_wei = await web3.eth.getBalance(addr4);
|
||||
expect(balance_wei).to.be.equal('102000000000000000000');
|
||||
});
|
||||
});
|
||||
|
||||
|
||||
async function computeTree(u) {
|
||||
const poseidon = circomlib.poseidon.createHash(6, 8, 57);
|
||||
commitment[u] = poseidon([coinCode, amount, secret[u], nullifier[u]]).toString();
|
||||
|
||||
// deposit
|
||||
// add commitment into SMT
|
||||
|
||||
rootOld[u] = tree.root;
|
||||
const resC = await tree.find(commitment[u]);
|
||||
assert(!resC.found);
|
||||
oldKey[u] = "1";
|
||||
if (!resC.found) {
|
||||
oldKey[u] = resC.notFoundKey.toString();
|
||||
}
|
||||
siblingsOld[u] = resC.siblings;
|
||||
while (siblingsOld[u].length < nLevels) {
|
||||
siblingsOld[u].push("0");
|
||||
};
|
||||
|
||||
await tree.insert(commitment[u], 0);
|
||||
rootNew[u] = tree.root;
|
||||
}
|
||||
|
||||
async function makeDeposit(u, addr) {
|
||||
const resC = await tree.find(commitment[u]);
|
||||
assert(resC.found);
|
||||
siblingsNew = resC.siblings;
|
||||
while (siblingsNew.length < nLevels) {
|
||||
siblingsNew.push("0");
|
||||
siblingsNew[u] = resC.siblings;
|
||||
while (siblingsNew[u].length < nLevels) {
|
||||
siblingsNew[u].push("0");
|
||||
};
|
||||
|
||||
// calculate witness
|
||||
@@ -90,13 +186,14 @@ contract("miksi", (accounts) => {
|
||||
const input = unstringifyBigInts({
|
||||
"coinCode": coinCode,
|
||||
"amount": amount,
|
||||
"secret": secret,
|
||||
"nullifier": nullifier,
|
||||
"siblingsOld": siblingsOld,
|
||||
"siblingsNew": siblingsNew,
|
||||
"rootOld": rootOld,
|
||||
"rootNew": rootNew,
|
||||
"commitment": commitment
|
||||
"secret": secret[u],
|
||||
"nullifier": nullifier[u],
|
||||
"oldKey": oldKey[u],
|
||||
"siblingsOld": siblingsOld[u],
|
||||
"siblingsNew": siblingsNew[u],
|
||||
"rootOld": rootOld[u],
|
||||
"rootNew": rootNew[u],
|
||||
"commitment": commitment[u]
|
||||
});
|
||||
const options = {};
|
||||
// console.log("Calculate witness");
|
||||
@@ -109,50 +206,29 @@ contract("miksi", (accounts) => {
|
||||
|
||||
// console.log("Generate zkSNARK proof");
|
||||
const res = groth.genProof(provingKey, witness);
|
||||
proof = res.proof;
|
||||
publicSignals = res.publicSignals;
|
||||
proof[u] = res.proof;
|
||||
publicSignals[u] = res.publicSignals;
|
||||
|
||||
const verificationKey = unstringifyBigInts(JSON.parse(fs.readFileSync("./build/deposit-verification_key.json", "utf8")));
|
||||
let pubI = unstringifyBigInts([coinCode, amount, rootOld.toString(), rootNew.toString(), commitment]);
|
||||
let validCheck = groth.isValid(verificationKey, proof, pubI);
|
||||
let pubI = unstringifyBigInts([coinCode, amount, rootOld[u].toString(), rootNew[u].toString(), commitment[u]]);
|
||||
let validCheck = groth.isValid(verificationKey, proof[u], pubI);
|
||||
assert(validCheck);
|
||||
await insMiksi.deposit(
|
||||
commitment,
|
||||
commitment[u],
|
||||
tree.root.toString(),
|
||||
[proof.pi_a[0].toString(), proof.pi_a[1].toString()],
|
||||
[proof[u].pi_a[0].toString(), proof[u].pi_a[1].toString()],
|
||||
[
|
||||
[proof.pi_b[0][1].toString(), proof.pi_b[0][0].toString()],
|
||||
[proof.pi_b[1][1].toString(), proof.pi_b[1][0].toString()]
|
||||
[proof[u].pi_b[0][1].toString(), proof[u].pi_b[0][0].toString()],
|
||||
[proof[u].pi_b[1][1].toString(), proof[u].pi_b[1][0].toString()]
|
||||
],
|
||||
[proof.pi_c[0].toString(), proof.pi_c[1].toString()],
|
||||
{from: addr1, value: amount}
|
||||
[proof[u].pi_c[0].toString(), proof[u].pi_c[1].toString()],
|
||||
{from: addr, value: amount}
|
||||
);
|
||||
|
||||
balance_wei = await web3.eth.getBalance(addr1);
|
||||
// console.log("Balance at " + addr1, web3.utils.fromWei(balance_wei, 'ether'));
|
||||
// expect(balance_wei).to.be.equal('98993526980000000000');
|
||||
});
|
||||
}
|
||||
|
||||
it("Get the commitments data", async () => {
|
||||
// getCommitments data
|
||||
let res = await insMiksi.getCommitments();
|
||||
expect(res[0][0].toString()).to.be.equal('189025084074544266465422070282645213792582195466360448472858620722286781863');
|
||||
expect(res[1].toString()).to.be.equal('9328869343897770565751281504295758914771207504252217956739346620422361279598');
|
||||
console.log(res[0]);
|
||||
commitmentsArray = res[0];
|
||||
});
|
||||
|
||||
it("Rebuild the tree from sc commitments", async () => {
|
||||
let treeTmp = await smt.newMemEmptyTrie();
|
||||
await treeTmp.insert(1, 0);
|
||||
for (let i=0; i<commitmentsArray.length; i++) {
|
||||
await treeTmp.insert(commitmentsArray[i], 0);
|
||||
}
|
||||
expect(treeTmp.root).to.be.equal(tree.root);
|
||||
});
|
||||
|
||||
it("Calculate witness and generate the zkProof", async () => {
|
||||
const resC = await tree.find(commitment);
|
||||
async function genZKProof(u, addr) {
|
||||
const resC = await tree.find(commitment[u]);
|
||||
assert(resC.found);
|
||||
let siblings = resC.siblings;
|
||||
while (siblings.length < nLevels) {
|
||||
@@ -165,11 +241,11 @@ contract("miksi", (accounts) => {
|
||||
const input = unstringifyBigInts({
|
||||
"coinCode": coinCode,
|
||||
"amount": amount,
|
||||
"secret": secret,
|
||||
"nullifier": nullifier,
|
||||
"secret": secret[u],
|
||||
"nullifier": nullifier[u],
|
||||
"siblings": siblings,
|
||||
"root": tree.root,
|
||||
"address": addr2
|
||||
"address": addr
|
||||
});
|
||||
const options = {};
|
||||
// console.log("Calculate witness");
|
||||
@@ -182,65 +258,20 @@ contract("miksi", (accounts) => {
|
||||
|
||||
// console.log("Generate zkSNARK proof");
|
||||
const res = groth.genProof(provingKey, witness);
|
||||
proof = res.proof;
|
||||
publicSignals = res.publicSignals;
|
||||
});
|
||||
proof[u] = res.proof;
|
||||
publicSignals[u] = res.publicSignals;
|
||||
}
|
||||
|
||||
it("Try to use the zkProof with another address and get revert", async () => {
|
||||
// console.log("Try to reuse the zkproof and expect revert");
|
||||
await truffleAssert.fails(
|
||||
insMiksi.withdraw(
|
||||
addr1,
|
||||
nullifier,
|
||||
[proof.pi_a[0].toString(), proof.pi_a[1].toString()],
|
||||
async function withdrawSC(u, addr) {
|
||||
return insMiksi.withdraw(
|
||||
addr,
|
||||
nullifier[u],
|
||||
[proof[u].pi_a[0].toString(), proof[u].pi_a[1].toString()],
|
||||
[
|
||||
[proof.pi_b[0][1].toString(), proof.pi_b[0][0].toString()],
|
||||
[proof.pi_b[1][1].toString(), proof.pi_b[1][0].toString()]
|
||||
[proof[u].pi_b[0][1].toString(), proof[u].pi_b[0][0].toString()],
|
||||
[proof[u].pi_b[1][1].toString(), proof[u].pi_b[1][0].toString()]
|
||||
],
|
||||
[proof.pi_c[0].toString(), proof.pi_c[1].toString()]
|
||||
),
|
||||
truffleAssert.ErrorType.REVERT,
|
||||
"zkProof withdraw could not be verified"
|
||||
);
|
||||
});
|
||||
[proof[u].pi_c[0].toString(), proof[u].pi_c[1].toString()]
|
||||
);
|
||||
}
|
||||
|
||||
it("Withdraw 1 ETH with the zkProof", async () => {
|
||||
// withdraw
|
||||
// console.log("Withdraw of " + ethAmount + " ETH to " + addr2);
|
||||
let resW = await insMiksi.withdraw(
|
||||
addr2,
|
||||
nullifier,
|
||||
[proof.pi_a[0].toString(), proof.pi_a[1].toString()],
|
||||
[
|
||||
[proof.pi_b[0][1].toString(), proof.pi_b[0][0].toString()],
|
||||
[proof.pi_b[1][1].toString(), proof.pi_b[1][0].toString()]
|
||||
],
|
||||
[proof.pi_c[0].toString(), proof.pi_c[1].toString()]
|
||||
);
|
||||
// console.log("resW", resW);
|
||||
|
||||
balance_wei = await web3.eth.getBalance(addr2);
|
||||
// console.log("Balance at " + addr2, web3.utils.fromWei(balance_wei, 'ether'));
|
||||
expect(balance_wei).to.be.equal('101000000000000000000');
|
||||
});
|
||||
|
||||
it("Try to reuse the zkProof and get revert", async () => {
|
||||
// console.log("Try to reuse the zkproof and expect revert");
|
||||
await truffleAssert.fails(
|
||||
insMiksi.withdraw(
|
||||
addr2,
|
||||
nullifier,
|
||||
[proof.pi_a[0].toString(), proof.pi_a[1].toString()],
|
||||
[
|
||||
[proof.pi_b[0][1].toString(), proof.pi_b[0][0].toString()],
|
||||
[proof.pi_b[1][1].toString(), proof.pi_b[1][0].toString()]
|
||||
],
|
||||
[proof.pi_c[0].toString(), proof.pi_c[1].toString()]
|
||||
),
|
||||
truffleAssert.ErrorType.REVERT,
|
||||
"nullifier already used"
|
||||
);
|
||||
balance_wei = await web3.eth.getBalance(addr2);
|
||||
expect(balance_wei).to.be.equal('101000000000000000000');
|
||||
});
|
||||
});
|
||||
|
||||
Reference in New Issue
Block a user