ethers.js/packages/tests/lib.esm/test-hdnode.js
2020-07-13 08:03:56 -04:00

181 lines
7.8 KiB
JavaScript

'use strict';
import assert from "assert";
import { ethers } from "ethers";
import { loadTests, randomNumber } from "@ethersproject/testcases";
function randomCase(seed, text) {
return text.split("").map(function (c, index) {
if (randomNumber(seed + "-" + index, 0, 2)) {
return c.toUpperCase();
}
return c;
}).join("");
}
// Too many test cases are caussing issues for the CI
// Only run random cases under random-128
function checkRandom(name) {
/*
if (name.substring(0, 7) === "random-") {
return (parseInt(name.substring(7)) <= 128);
}
*/
return true;
}
describe('Test HD Node Derivation is Case Agnostic', function () {
let tests = loadTests('hdnode');
tests.forEach((test) => {
if (!checkRandom(test.name)) {
return;
}
it("Normalizes case - " + test.name, function () {
this.timeout(10000);
let wordlist = (ethers.wordlists)[test.locale];
let rootNode = ethers.utils.HDNode.fromMnemonic(test.mnemonic, test.password || null, wordlist);
let altMnemonic = randomCase(test.name, test.mnemonic);
let altNode = ethers.utils.HDNode.fromMnemonic(altMnemonic, test.password || null, wordlist);
assert.equal(altNode.privateKey, rootNode.privateKey, altMnemonic);
});
});
});
describe('Test HD Node Derivation from Seed', function () {
let tests = loadTests('hdnode');
tests.forEach((test) => {
if (!checkRandom(test.name)) {
return;
}
// If there is nothing to derive, skip this portion of the test
if (test.hdnodes.length === 0) {
return;
}
it('Derives the HD nodes - ' + test.name, function () {
this.timeout(10000);
let rootNode = ethers.utils.HDNode.fromSeed(test.seed);
test.hdnodes.forEach((nodeTest) => {
let node = rootNode.derivePath(nodeTest.path);
assert.equal(node.privateKey, nodeTest.privateKey, 'Generates privateKey - ' + nodeTest.privateKey);
let wallet = new ethers.Wallet(node.privateKey);
assert.equal(wallet.address.toLowerCase(), nodeTest.address, 'Generates address - ' + nodeTest.privateKey);
});
});
});
});
describe('Test HD Node Derivation from Mnemonic', function () {
let tests = loadTests('hdnode');
tests.forEach((test) => {
if (!checkRandom(test.name)) {
return;
}
// If there is nothing to derive, skip this portion of the test
if (test.hdnodes.length === 0) {
return;
}
it('Derives the HD nodes - ' + test.name, function () {
this.timeout(10000);
let rootNode = ethers.utils.HDNode.fromMnemonic(test.mnemonic, test.password || null);
test.hdnodes.forEach((nodeTest) => {
let node = rootNode.derivePath(nodeTest.path);
assert.equal(node.privateKey, nodeTest.privateKey, 'Matches privateKey - ' + nodeTest.privateKey);
assert.equal(node.path, nodeTest.path, 'Matches path - ' + nodeTest.privateKey);
assert.equal(node.mnemonic.phrase, test.mnemonic, 'Matches mnemonic.phrase - ' + nodeTest.privateKey);
assert.equal(node.mnemonic.path, nodeTest.path, 'Matches mnemonic.path - ' + nodeTest.privateKey);
let wallet = new ethers.Wallet(node.privateKey);
assert.equal(wallet.address.toLowerCase(), nodeTest.address, 'Generates address - ' + nodeTest.privateKey);
});
});
});
});
describe('Test HD Mnemonic Phrases', function testMnemonic() {
let tests = loadTests('hdnode');
tests.forEach(function (test) {
if (!checkRandom(test.name)) {
return;
}
it(('converts mnemonic phrases - ' + test.name), function () {
this.timeout(1000000);
assert.equal(ethers.utils.mnemonicToSeed(test.mnemonic, test.password), test.seed, 'Converts mnemonic to seed - ' + test.mnemonic + ':' + test.password);
// Test default english
if (test.locale === "en") {
assert.equal(ethers.utils.entropyToMnemonic(test.entropy), test.mnemonic, "Converts entropy to mnemonic " + test.name + " (default en)");
assert.equal(ethers.utils.mnemonicToEntropy(test.mnemonic), test.entropy, "Converts mnemonic to entropy - " + test.mnemonic + " (default en)");
}
let wordlist = (ethers.wordlists)[test.locale];
let mnemonic = ethers.utils.entropyToMnemonic(test.entropy, wordlist);
assert.equal(mnemonic.normalize('NFKD'), test.mnemonic.normalize('NFKD'), 'Converts entropy to mnemonic ' + test.name);
assert.equal(ethers.utils.mnemonicToEntropy(test.mnemonic, wordlist), test.entropy, 'Converts mnemonic to entropy - ' + test.mnemonic);
});
});
});
describe("HD Extended Keys", function () {
const root = ethers.utils.HDNode.fromSeed("0xdeadbeefdeadbeefdeadbeefdeadbeef");
const root42 = root.derivePath("42");
it("exports and imports xpriv extended keys", function () {
const xpriv = root.extendedKey;
const node = ethers.utils.HDNode.fromExtendedKey(xpriv);
assert.equal(root.address, node.address, "address matches");
const node42 = node.derivePath("42");
assert.equal(root42.address, node42.address, "address matches");
});
it("exports and imports xpub extended keys", function () {
const xpub = root.neuter().extendedKey;
const node = ethers.utils.HDNode.fromExtendedKey(xpub);
assert.equal(root.address, node.address, "address matches");
const node42 = node.derivePath("42");
assert.equal(root42.address, node42.address, "address matches");
});
});
describe("HD error cases", function () {
const testInvalid = [
"",
"m/45/m",
"m/44/foobar"
];
const root = ethers.utils.HDNode.fromSeed("0xdeadbeefdeadbeefdeadbeefdeadbeef");
testInvalid.forEach((path) => {
it(`fails on path "${path}"`, function () {
assert.throws(() => {
root.derivePath(path);
}, (error) => {
return true;
});
});
});
it("fails to derive child of hardened key", function () {
// Deriving non-hardened should work...
const node = root.neuter().derivePath("44");
assert.throws(() => {
// Deriving hardened should fail...
node.derivePath("44'");
}, (error) => {
return true;
});
});
// The zero-mnemonic, with and without correct checksum
const zeroMnemonicCS = "abandon abandon abandon abandon abandon abandon abandon abandon abandon abandon abandon about";
const zeroMnemonic = "abandon abandon abandon abandon abandon abandon abandon abandon abandon abandon abandon abandon";
it("fails on invalid mnemonic length", function () {
const shortMnemonic = "abandon abandon abandon abandon";
// Test the validate functions
assert.ok(ethers.utils.isValidMnemonic(zeroMnemonicCS));
assert.ok(!ethers.utils.isValidMnemonic(zeroMnemonic));
assert.ok(!ethers.utils.isValidMnemonic(shortMnemonic));
assert.throws(() => {
ethers.utils.mnemonicToEntropy(shortMnemonic);
}, (error) => {
return true;
});
});
it("fails on invalid checksum", function () {
assert.throws(() => {
ethers.utils.mnemonicToEntropy(zeroMnemonic);
}, (error) => {
return true;
});
});
it("fails on unknown locale", function () {
assert.throws(() => {
ethers.utils.HDNode.fromMnemonic(zeroMnemonicCS, "foobar", "xx");
}, (error) => {
return true;
});
});
});
//# sourceMappingURL=test-hdnode.js.map