62 lines
2.0 KiB
JavaScript
62 lines
2.0 KiB
JavaScript
const { contract } = require('@openzeppelin/test-environment');
|
|
|
|
require('@openzeppelin/test-helpers');
|
|
|
|
const { MerkleTree } = require('../helpers/merkleTree.js');
|
|
const { keccakFromString, bufferToHex } = require('ethereumjs-util');
|
|
|
|
const { expect } = require('chai');
|
|
|
|
const MerkleProofWrapper = contract.fromArtifact('MerkleProofWrapper');
|
|
|
|
describe('MerkleProof', function () {
|
|
beforeEach(async function () {
|
|
this.merkleProof = await MerkleProofWrapper.new();
|
|
});
|
|
|
|
describe('verify', function () {
|
|
it('returns true for a valid Merkle proof', async function () {
|
|
const elements = ['a', 'b', 'c', 'd'];
|
|
const merkleTree = new MerkleTree(elements);
|
|
|
|
const root = merkleTree.getHexRoot();
|
|
|
|
const proof = merkleTree.getHexProof(elements[0]);
|
|
|
|
const leaf = bufferToHex(keccakFromString(elements[0]));
|
|
|
|
expect(await this.merkleProof.verify(proof, root, leaf)).to.equal(true);
|
|
});
|
|
|
|
it('returns false for an invalid Merkle proof', async function () {
|
|
const correctElements = ['a', 'b', 'c'];
|
|
const correctMerkleTree = new MerkleTree(correctElements);
|
|
|
|
const correctRoot = correctMerkleTree.getHexRoot();
|
|
|
|
const correctLeaf = bufferToHex(keccakFromString(correctElements[0]));
|
|
|
|
const badElements = ['d', 'e', 'f'];
|
|
const badMerkleTree = new MerkleTree(badElements);
|
|
|
|
const badProof = badMerkleTree.getHexProof(badElements[0]);
|
|
|
|
expect(await this.merkleProof.verify(badProof, correctRoot, correctLeaf)).to.equal(false);
|
|
});
|
|
|
|
it('returns false for a Merkle proof of invalid length', async function () {
|
|
const elements = ['a', 'b', 'c'];
|
|
const merkleTree = new MerkleTree(elements);
|
|
|
|
const root = merkleTree.getHexRoot();
|
|
|
|
const proof = merkleTree.getHexProof(elements[0]);
|
|
const badProof = proof.slice(0, proof.length - 5);
|
|
|
|
const leaf = bufferToHex(keccakFromString(elements[0]));
|
|
|
|
expect(await this.merkleProof.verify(badProof, root, leaf)).to.equal(false);
|
|
});
|
|
});
|
|
});
|