xahau.js/packages/ripple-binary-codec/test/shamap-test.js

const assert = require('assert');
const {ShaMap} = require('../src/shamap.js');
const {binary: {serializeObject}, Hash256, HashPrefix}
  = require('../src/coretypes');
const {loadFixture} = require('./utils');

function now() {
  return (Number(Date.now())) / 1000;
}

const ZERO =
  '0000000000000000000000000000000000000000000000000000000000000000';

function makeItem(indexArg) {
  let str = indexArg;
  while (str.length < 64) {
    str += '0';
  }
  const index = Hash256.from(str);
  const item = {
    toBytesSink(sink) {
      index.toBytesSink(sink);
    },
    hashPrefix() {
      return [1, 3, 3, 7];
    }
  };
  return [index, item];
}

describe('ShaMap', () => {
  now();

  it('hashes to zero when empty', () => {
    const map = new ShaMap();
    assert.equal(map.hash().toHex(), ZERO);
  });
  it('creates the same hash no matter which order items are added', () => {
    let map = new ShaMap();
    const items = [
      '0',
      '1',
      '11',
      '7000DE445E22CB9BB7E1717589FA858736BAA5FD192310E20000000000000000',
      '7000DE445E22CB9BB7E1717589FA858736BAA5FD192310E21000000000000000',
      '7000DE445E22CB9BB7E1717589FA858736BAA5FD192310E22000000000000000',
      '7000DE445E22CB9BB7E1717589FA858736BAA5FD192310E23000000000000000',
      '12',
      '122'
    ];
    items.forEach(i => map.addItem(...makeItem(i)));
    const h1 = map.hash();
    assert(h1.eq(h1));
    map = new ShaMap();
    items.reverse().forEach(i => map.addItem(...makeItem(i)));
    assert(map.hash().eq(h1));
  });
  function factory(fixture) {
    it(`recreate account state hash from ${fixture}`, () => {
      const map = new ShaMap();
      const ledger = loadFixture(fixture);
      // const t = now();
      const leafNodePrefix = HashPrefix.accountStateEntry;
      ledger.accountState.map((e, i) => {
        if (i > 1000 & (i % 1000) === 0) {
          console.log(e.index);
          console.log(i);
        }
        const bytes = serializeObject(e);
        return {
          index: Hash256.from(e.index),
          hashPrefix() {
            return leafNodePrefix;
          },
          toBytesSink(sink) {
            sink.put(bytes);
          }
        };
      }).forEach(so => map.addItem(so.index, so));
      assert.equal(map.hash().toHex(), ledger.account_hash);
      // console.log('took seconds: ', (now() - t));
    });
  }
  factory('ledger-full-38129.json');
  factory('ledger-full-40000.json');
  // factory('ledger-4320277.json');
  // factory('14280680.json');
});