ethers.js/packages/tests/src.ts/utils.ts
2019-05-14 18:42:45 -04:00

69 lines
1.8 KiB
TypeScript

'use strict';
import { ethers } from "ethers";
function randomBytes(seed: string, lower: number, upper?: number): Uint8Array {
if (!upper) { upper = lower; }
if (upper === 0 && upper === lower) { return new Uint8Array(0); }
let result = ethers.utils.arrayify(ethers.utils.keccak256(ethers.utils.toUtf8Bytes(seed)));
while (result.length < upper) {
result = ethers.utils.concat([result, ethers.utils.keccak256(ethers.utils.concat([seed, result]))]);
}
let top = ethers.utils.arrayify(ethers.utils.keccak256(result));
let percent = ((top[0] << 16) | (top[1] << 8) | top[2]) / 0x01000000;
return result.slice(0, lower + Math.floor((upper - lower) * percent));
}
function randomHexString(seed: string, lower: number, upper?: number): string {
return ethers.utils.hexlify(randomBytes(seed, lower, upper));
}
function randomNumber(seed: string, lower: number, upper: number): number {
let top = randomBytes(seed, 3);
let percent = ((top[0] << 16) | (top[1] << 8) | top[2]) / 0x01000000;
return lower + Math.floor((upper - lower) * percent);
}
function equals(a: any, b: any): boolean {
// Array (treat recursively)
if (Array.isArray(a)) {
if (!Array.isArray(b) || a.length !== b.length) { return false; }
for (let i = 0; i < a.length; i++) {
if (!equals(a[i], b[i])) { return false; }
}
return true;
}
// BigNumber
if (a.eq) {
if (!b.eq || !a.eq(b)) { return false; }
return true;
}
// Uint8Array
if (a.buffer) {
if (!b.buffer || a.length !== b.length) { return false; }
for (let i = 0; i < a.length; i++) {
if (a[i] !== b[i]) { return false; }
}
return true;
}
// Something else
return a === b;
}
export {
randomBytes,
randomHexString,
randomNumber,
equals
}