copay/test/test.TxProposals.js

'use strict';

var chai = chai || require('chai');
var should = chai.should();
var bitcore = bitcore || require('bitcore');
var Transaction = bitcore.Transaction;
var buffertools = bitcore.buffertools;
var WalletKey = bitcore.WalletKey;
var Key = bitcore.Key;
var bignum = bitcore.Bignum;
var Script = bitcore.Script;
var Builder = bitcore.TransactionBuilder;
var util = bitcore.util;
var networks = bitcore.networks;
try {
  var copay = require('copay'); //browser
} catch (e) {
  var copay = require('../copay'); //node
}
var fakeStorage = copay.FakeStorage;
var PrivateKey = copay.PrivateKey || require('../js/models/PrivateKey');
var TxProposals = copay.TxProposals || require('../js/models/TxProposal');
var is_browser = (typeof process == 'undefined' || typeof process.versions === 'undefined')
var PublicKeyRing = is_browser ? copay.PublicKeyRing :
  require('soop').load('../js/models/core/PublicKeyRing', {
    Storage: fakeStorage
  });

var config = {
  networkName: 'testnet',
};

var unspentTest = [{
  "address": "dummy",
  "scriptPubKey": "dummy",
  "txid": "2ac165fa7a3a2b535d106a0041c7568d03b531e58aeccdd3199d7289ab12cfc1",
  "vout": 1,
  "amount": 10,
  "confirmations": 7
}];

var createPKR = function(bip32s) {
  var w = new PublicKeyRing(config);
  should.exist(w);

  for (var i = 0; i < 5; i++) {
    if (bip32s && i < bip32s.length) {
      var b = bip32s[i];
      w.addCopayer(b.deriveBIP45Branch().extendedPublicKeyString());
    } else {
      w.addCopayer();
    }
  }

  var pubkey = bip32s[0].publicHex;

  w.generateAddress(false, pubkey);
  w.generateAddress(false, pubkey);
  w.generateAddress(false, pubkey);
  w.generateAddress(true, pubkey);
  w.generateAddress(true, pubkey);
  w.generateAddress(true, pubkey);

  return w;
};

var vopts = {
  verifyP2SH: true,
  dontVerifyStrictEnc: true
};


describe('TxProposals model', function() {

  var isChange = false;
  var index = 0;

  it('verify TXs', function(done) {

    var priv = new PrivateKey(config);
    var priv2 = new PrivateKey(config);
    var priv3 = new PrivateKey(config);
    var pub = priv.publicHex;

    var ts = Date.now();
    var pkr = createPKR([priv, priv2, priv3]);
    var opts = {
      remainderOut: {
        address: pkr.generateAddress(true, pub).toString()
      }
    };

    var w = new TxProposals({
      networkName: config.networkName,
    });

    unspentTest[0].address = pkr.getAddress(index, isChange, pub).toString();
    unspentTest[0].scriptPubKey = pkr.getScriptPubKeyHex(index, isChange, pub);
    w.add(createTx(
      '15q6HKjWHAksHcH91JW23BJEuzZgFwydBt',
      '123456789',
      unspentTest,
      opts,
      priv,
      pkr
    ));
    var ntxid = Object.keys(w.txps)[0];
    var b = w.txps[ntxid].builder;
    var tx = b.build();
    tx.isComplete().should.equal(false);

    var ringIndex = pkr.getIndex(pub);
    b.sign(priv2.getAll(ringIndex.getReceiveIndex(), ringIndex.getChangeIndex(), ringIndex.cosigner));
    b.sign(priv3.getAll(ringIndex.getReceiveIndex(), ringIndex.getChangeIndex(), ringIndex.cosigner));
    tx = b.build();
    tx.isComplete().should.equal(true);

    var s = new Script(new bitcore.Buffer(unspentTest[0].scriptPubKey, 'hex'));

    tx.verifyInput(0, s, {
      verifyP2SH: true,
      dontVerifyStrictEnc: true
    }, function(err, results) {
      should.not.exist(err);
      results.should.equal(true);
      done();
    });
  });


  it('should create an instance', function() {
    var w = new TxProposals({
      networkName: config.networkName
    });
    should.exist(w);
    w.network.name.should.equal(config.networkName);
  });

  var createTx = function(toAddress, amountSatStr, utxos, opts, priv, pkr) {
    opts = opts || {};

    var pub = priv.publicHex;

    if (!pkr.isComplete()) {
      throw new Error('publicKeyRing is not complete');
    }

    if (!opts.remainderOut) {
      opts.remainderOut = {
        address: pkr.generateAddress(true, pub).toString()
      };
    };

    var b = new Builder(opts)
      .setUnspent(utxos)
      .setOutputs([{
        address: toAddress,
        amountSatStr: amountSatStr,
      }]);
    var selectedUtxos = b.getSelectedUnspent();
    var inputChainPaths = selectedUtxos.map(function(utxo) {
      return pkr.pathForAddress(utxo.address);
    });

    var selectedUtxos = b.getSelectedUnspent();
    var inputChainPaths = selectedUtxos.map(function(utxo) {
      return pkr.pathForAddress(utxo.address);
    });
    b.setHashToScriptMap(pkr.getRedeemScriptMap(inputChainPaths));

    var signRet;
    if (priv) {
      var pkeys = priv.getForPaths(inputChainPaths);
      b.sign(pkeys);
    }
    var me = {};
    if (priv) me[priv.getId()] = Date.now();

    var tx = b.build();

    return {
      inputChainPaths: inputChainPaths,
      creator: priv.getId(),
      createdTs: new Date(),
      signedBy: priv && tx.countInputSignatures(0) ? me : {},
      seenBy: priv ? me : {},
      builder: b,
    };
  };


  it('#getUsedUnspend', function() {
    var priv = new PrivateKey(config);
    var pub = priv.publicHex;

    var w = new TxProposals({
      networkName: config.networkName,
    });
    var start = new Date().getTime();
    var pkr = createPKR([priv]);
    var ts = Date.now();
    unspentTest[0].address = pkr.getAddress(index, isChange, pub).toString();
    unspentTest[0].scriptPubKey = pkr.getScriptPubKeyHex(index, isChange, pub);
    w.add(createTx(
      '15q6HKjWHAksHcH91JW23BJEuzZgFwydBt',
      '123456789',
      unspentTest, {},
      priv,
      pkr
    ));
    var uu = w.getUsedUnspent();
    var uuk = Object.keys(uu);
    uuk.length.should.equal(1);
    uuk[0].split(',')[0].should.equal(unspentTest[0].txid);
  });

  it('#merge with self', function() {
    var priv = new PrivateKey(config);
    var pub = priv.publicHex;

    var w = new TxProposals({
      networkName: config.networkName,
    });
    var start = new Date().getTime();
    var pkr = createPKR([priv]);
    var ts = Date.now();

    unspentTest[0].address = pkr.getAddress(index, isChange, pub).toString();
    unspentTest[0].scriptPubKey = pkr.getScriptPubKeyHex(index, isChange, pub);
    w.add(createTx(
      '15q6HKjWHAksHcH91JW23BJEuzZgFwydBt',
      '123456789',
      unspentTest, {},
      priv,
      pkr
    ));
    var ntxid = Object.keys(w.txps)[0];
    var tx = w.txps[ntxid].builder.build();
    tx.isComplete().should.equal(false);
    tx.countInputMissingSignatures(0).should.equal(2);

    var x = priv.getId();
    (w.txps[ntxid].signedBy[priv.getId()] - ts > 0).should.equal(true);
    (w.txps[ntxid].seenBy[priv.id] - ts > 0).should.equal(true);

    var info = w.merge(w.txps[ntxid], pkr.getCopayerId(0));
    info.events.length.should.equal(0);

    Object.keys(w.txps).length.should.equal(1);

    tx.isComplete().should.equal(false);
    tx.countInputMissingSignatures(0).should.equal(2);

    (w.txps[ntxid].signedBy[priv.id] - ts > 0).should.equal(true);
    (w.txps[ntxid].seenBy[priv.id] - ts > 0).should.equal(true);
  });



  it('#merge, merge signatures case 1', function() {
    var priv2 = new PrivateKey(config);
    var priv = new PrivateKey(config);
    var pub = priv.publicHex;

    var ts = Date.now();
    var pkr = createPKR([priv]);
    var opts = {
      remainderOut: {
        address: pkr.generateAddress(true, pub).toString()
      }
    };


    var w = new TxProposals({
      networkName: config.networkName,
    });
    unspentTest[0].address = pkr.getAddress(index, isChange, pub).toString();
    unspentTest[0].scriptPubKey = pkr.getScriptPubKeyHex(index, isChange, pub);
    w.add(createTx(
      '15q6HKjWHAksHcH91JW23BJEuzZgFwydBt',
      '123456789',
      unspentTest,
      opts,
      priv2,
      pkr
    ));

    var ntxid = Object.keys(w.txps)[0];
    var tx = w.txps[ntxid].builder.build();
    tx.isComplete().should.equal(false);
    tx.countInputSignatures(0).should.equal(0);
    tx.countInputMissingSignatures(0).should.equal(1);

    Object.keys(w.txps[ntxid].signedBy).length.should.equal(0);
    Object.keys(w.txps[ntxid].seenBy).length.should.equal(1);


    var w2 = new TxProposals({
      networkName: config.networkName,
      publicKeyRing: w.publicKeyRing,
    });
    unspentTest[0].address = pkr.getAddress(index, isChange, pub).toString();
    unspentTest[0].scriptPubKey = pkr.getScriptPubKeyHex(index, isChange, pub);
    w2.add(createTx(
      '15q6HKjWHAksHcH91JW23BJEuzZgFwydBt',
      '123456789',
      unspentTest,
      opts,
      priv,
      pkr
    ));

    var ntxid = Object.keys(w.txps)[0];
    var tx = w2.txps[ntxid].builder.build();
    tx.isComplete().should.equal(false);
    tx.countInputMissingSignatures(0).should.equal(2);


    (w2.txps[ntxid].signedBy[priv.id] - ts > 0).should.equal(true, 'asdsd');
    (w2.txps[ntxid].seenBy[priv.id] - ts > 0).should.equal(true);

    var info = w.merge(w2.txps[ntxid], pkr.getCopayerId(0));
    info.events.length.should.equal(2);
    info.events[0].type.should.equal('seen');
    info.events[1].type.should.equal('signed');

    Object.keys(w.txps).length.should.equal(1);

    var tx = w.txps[ntxid].builder.build();
    tx.isComplete().should.equal(false);
    tx.countInputMissingSignatures(0).should.equal(2);
    (w.txps[ntxid].signedBy[priv.id] - ts > 0).should.equal(true);
    (w.txps[ntxid].seenBy[priv.id] - ts > 0).should.equal(true);

  });

  var _dumpChunks = function(scriptSig, label) {
    console.log('## DUMP: ' + label + ' ##');
    for (var i = 0; i < scriptSig.chunks.length; i++) {
      console.log('\tCHUNK ', i, scriptSig.chunks[i]);
    }
  };


  it('#merge, merge signatures case 2', function() {

    var o1 = {
      extendedPrivateKeyString: 'tprv8ZgxMBicQKsPdSF1avR6mXyDj5Uv1XY2UyUHSDpAXQ5TvPN7prGeDppjy4562rBB9gMMAhRfFdJrNDpQ4t69kkqHNEEen3PX1zBJqSehJDH',
      networkName: 'testnet',
      privateKeyCache: {}
    };
    var o2 = {
      extendedPrivateKeyString: 'tprv8ZgxMBicQKsPdVeB5RzuxS9JQcACueZYgUaM5eWzaEBkHjW5Pg6Mqez1APSqoUP1jUdbT8WVG7ZJYTXvUL7XtPzFYBXjmdKuwSor1dcNQ8j',
      networkName: 'testnet',
      privateKeyCache: {}
    };
    var o3 = {
      extendedPrivateKeyString: 'tprv8ZgxMBicQKsPeHWNrPVZtQVgcCtXBr5TACNbDQ56rwqNJce9MEc64US6DJKxpWsrebEomxxWZFDtkvkZGkzA43uLvdF4XHiWqoNaL6Dq2Gd',
      networkName: 'testnet',
      privateKeyCache: {}
    };


    var priv = PrivateKey.fromObj(o1);
    var priv2 = PrivateKey.fromObj(o2);
    var priv3 = PrivateKey.fromObj(o3);
    var pub = priv.publicHex;

    var ts = Date.now();
    var pkr = createPKR([priv, priv2]);
    var opts = {
      remainderOut: {
        address: '2MxK2m7cPtEwjZBB8Ksq7ppjkgJyFPJGemr'
      }
    };
    var addressToSign = pkr.generateAddress(false, pub);
    unspentTest[0].address = addressToSign.toString();
    unspentTest[0].scriptPubKey = pkr.getScriptPubKeyHex(index, isChange, pub);
    var tx, txb;

    var w = new TxProposals({
      networkName: config.networkName,
    });

    w.add(createTx(
      '15q6HKjWHAksHcH91JW23BJEuzZgFwydBt',
      '123456789',
      unspentTest,
      opts,
      priv3,
      pkr
    ));

    var ntxid = Object.keys(w.txps)[0];
    txb = w.txps[ntxid].builder;
    tx = txb.build();

    tx.isComplete().should.equal(false);
    tx.countInputMissingSignatures(0).should.equal(1);

    Object.keys(w.txps[ntxid].signedBy).length.should.equal(0);
    Object.keys(w.txps[ntxid].seenBy).length.should.equal(1);

    var w2 = new TxProposals({
      networkName: config.networkName,
    });



    w2.add(createTx(
      '15q6HKjWHAksHcH91JW23BJEuzZgFwydBt',
      '123456789',
      unspentTest,
      opts,
      priv,
      pkr
    ));
    var ntxid = Object.keys(w2.txps)[0];
    txb = w2.txps[ntxid].builder;
    tx = txb.build();

    tx.isComplete().should.equal(false);
    tx.countInputMissingSignatures(0).should.equal(2);

    (w2.txps[ntxid].signedBy[priv.id] - ts > 0).should.equal(true);
    (w2.txps[ntxid].seenBy[priv.id] - ts > 0).should.equal(true);

    var info = w.merge(w2.txps[ntxid], pkr.getCopayerId(0));
    info.events.length.should.equal(2);
    info.events[0].type.should.equal('seen');
    info.events[1].type.should.equal('signed');

    tx = w.txps[ntxid].builder.build();
    tx.isComplete().should.equal(false);
    tx.countInputMissingSignatures(0).should.equal(2);
    (w.txps[ntxid].signedBy[priv.id] - ts > 0).should.equal(true);
    (w.txps[ntxid].seenBy[priv.id] - ts > 0).should.equal(true);


    var w3 = new TxProposals({
      networkName: config.networkName,
      publicKeyRing: pkr,
    });
    w3.add(createTx(
      '15q6HKjWHAksHcH91JW23BJEuzZgFwydBt',
      '123456789',
      unspentTest,
      opts,
      priv2,
      pkr
    ));
    tx = w3.txps[ntxid].builder.build();
    tx.isComplete().should.equal(false);
    tx.countInputMissingSignatures(0).should.equal(2);

    (w3.txps[ntxid].signedBy[priv2.id] - ts > 0).should.equal(true);
    (w3.txps[ntxid].seenBy[priv2.id] - ts > 0).should.equal(true);

    var info = w.merge(w3.txps[ntxid], pkr.getCopayerId(1));

    Object.keys(w.txps).length.should.equal(1);

    (w.txps[ntxid].signedBy[priv.id] - ts > 0).should.equal(true);
    (w.txps[ntxid].seenBy[priv.id] - ts > 0).should.equal(true);
    (w.txps[ntxid].signedBy[priv2.id] - ts > 0).should.equal(true);
    (w.txps[ntxid].seenBy[priv2.id] - ts > 0).should.equal(true);

    tx = w.txps[ntxid].builder.build();
    tx.isComplete().should.equal(false);
    tx.countInputMissingSignatures(0).should.equal(1);
  });


  it('#merge, merge signatures case 3', function() {

    var priv = new PrivateKey(config);
    var priv2 = new PrivateKey(config);
    var priv3 = new PrivateKey(config);
    var pub = priv.publicHex;


    var ts = Date.now();
    var pkr = createPKR([priv, priv2, priv3]);
    var opts = {
      remainderOut: {
        address: pkr.generateAddress(true, pub).toString()
      }
    };

    var w = new TxProposals({
      networkName: config.networkName,
    });
    unspentTest[0].address = pkr.getAddress(index, isChange, pub).toString();
    unspentTest[0].scriptPubKey = pkr.getScriptPubKeyHex(index, isChange, pub);
    w.add(createTx(
      '15q6HKjWHAksHcH91JW23BJEuzZgFwydBt',
      '123456789',
      unspentTest,
      opts,
      priv,
      pkr
    ));
    var ntxid = Object.keys(w.txps)[0];
    var tx = w.txps[ntxid].builder.build();
    tx.isComplete().should.equal(false);
    tx.countInputMissingSignatures(0).should.equal(2);
    (w.txps[ntxid].signedBy[priv.id] - ts > 0).should.equal(true);
    (w.txps[ntxid].seenBy[priv.id] - ts > 0).should.equal(true);


    var w2 = new TxProposals({
      networkName: config.networkName,
    });
    unspentTest[0].address = pkr.getAddress(index, isChange, pub).toString();
    unspentTest[0].scriptPubKey = pkr.getScriptPubKeyHex(index, isChange, pub);
    w2.add(createTx(
      '15q6HKjWHAksHcH91JW23BJEuzZgFwydBt',
      '123456789',
      unspentTest,
      opts,
      priv2,
      pkr
    ));
    var tx = w2.txps[ntxid].builder.build();
    tx.isComplete().should.equal(false);
    tx.countInputMissingSignatures(0).should.equal(2);
    (w2.txps[ntxid].signedBy[priv2.id] - ts > 0).should.equal(true);
    (w2.txps[ntxid].seenBy[priv2.id] - ts > 0).should.equal(true);

    var w3 = new TxProposals({
      networkName: config.networkName,
    });
    unspentTest[0].address = pkr.getAddress(index, isChange, pub).toString();
    unspentTest[0].scriptPubKey = pkr.getScriptPubKeyHex(index, isChange, pub);
    w3.add(createTx(
      '15q6HKjWHAksHcH91JW23BJEuzZgFwydBt',
      '123456789',
      unspentTest,
      opts,
      priv3,
      pkr
    ));
    var tx = w3.txps[ntxid].builder.build();
    tx.isComplete().should.equal(false);
    tx.countInputMissingSignatures(0).should.equal(2);
    (w3.txps[ntxid].signedBy[priv3.id] - ts > 0).should.equal(true);
    (w3.txps[ntxid].seenBy[priv3.id] - ts > 0).should.equal(true);

    var info = w.merge(w2.txps[ntxid], pkr.getCopayerId(1));

    Object.keys(w.txps).length.should.equal(1);
    var tx = w.txps[ntxid].builder.build();
    tx.isComplete().should.equal(false);
    tx.countInputMissingSignatures(0).should.equal(1);
    (w.txps[ntxid].seenBy[priv.id] - ts > 0).should.equal(true);
    (w.txps[ntxid].seenBy[priv2.id] - ts > 0).should.equal(true);
    (w.txps[ntxid].signedBy[priv.id] - ts > 0).should.equal(true);
    (w.txps[ntxid].signedBy[priv2.id] - ts > 0).should.equal(true);


    var info = w.merge(w3.txps[ntxid], pkr.getCopayerId(2));

    var tx = w.txps[ntxid].builder.build();
    tx.isComplete().should.equal(true);
    tx.countInputMissingSignatures(0).should.equal(0);
    Object.keys(w.txps).length.should.equal(1);
    (w.txps[ntxid].seenBy[priv.id] - ts > 0).should.equal(true);
    (w.txps[ntxid].seenBy[priv2.id] - ts > 0).should.equal(true);
    (w.txps[ntxid].seenBy[priv3.id] - ts > 0).should.equal(true);
    (w.txps[ntxid].signedBy[priv.id] - ts > 0).should.equal(true);
    (w.txps[ntxid].signedBy[priv2.id] - ts > 0).should.equal(true);
    (w.txps[ntxid].signedBy[priv3.id] - ts > 0).should.equal(true);
  });



  it('#toObj #fromObj roundtrip', function() {

    var priv = new PrivateKey(config);
    var pub = priv.publicHex;

    var pkr = createPKR([priv]);
    var w = new TxProposals({
      walletId: 'qwerty',
      networkName: config.networkName,
    });
    var ts = Date.now();

    unspentTest[0].address = pkr.getAddress(index, isChange, pub).toString();
    unspentTest[0].scriptPubKey = pkr.getScriptPubKeyHex(index, isChange, pub);
    w.add(createTx(
      '15q6HKjWHAksHcH91JW23BJEuzZgFwydBt',
      '123456789',
      unspentTest, {},
      priv,
      pkr
    ));
    var ntxid = Object.keys(w.txps)[0];
    var tx = w.txps[ntxid].builder.build();
    tx.isComplete().should.equal(false);
    tx.countInputMissingSignatures(0).should.equal(2);
    (w.txps[ntxid].signedBy[priv.id] - ts > 0).should.equal(true);
    (w.txps[ntxid].seenBy[priv.id] - ts > 0).should.equal(true);

    var o = w.toObj();
    should.exist(o);
    o.txps.length.should.equal(1);

    should.exist(o.txps[0]);
    should.exist(o.txps[0].signedBy);
    should.exist(o.txps[0].seenBy);
    should.exist(o.txps[0].builderObj);
    should.exist(o.txps[0].signedBy[priv.id]);

    var o2 = JSON.parse(JSON.stringify(o));
    var w2 = TxProposals.fromObj(o2);
    w2.walletId.should.equal(w.walletId);

    var tx2 = w2.txps[ntxid].builder.build();
    tx2.isComplete().should.equal(false);
    tx2.countInputMissingSignatures(0).should.equal(2);
    (w2.txps[ntxid].signedBy[priv.id] - ts > 0).should.equal(true);
    (w2.txps[ntxid].seenBy[priv.id] - ts > 0).should.equal(true);
    should.exist(w2.txps[ntxid].builder);
    should.exist(w2.txps[ntxid].builder.valueInSat);

    w2.merge(w.txps[ntxid], pkr.getCopayerId(0));
    Object.keys(w2.txps).length.should.equal(1);
  });


});