Merge pull request bitpay#2341 from matiu/feat/paypro-headers

add bp_partner

Author: matiu

packages/bitcore-wallet-client/src/lib/api.ts

@@ -47,6 +47,8 @@ export class API extends EventEmitter {
   keyDerivationOk: boolean;
   noSign: any;
   password: any;
+  bp_partner: string;
+  bp_partner_version: string;
 
   static PayPro = PayPro;
   static Key = Key;
@@ -68,6 +70,9 @@ export class API extends EventEmitter {
     this.logLevel = opts.logLevel || 'silent';
     this.supportStaffWalletId = opts.supportStaffWalletId;
 
+    this.bp_partner = opts.bp_partner;
+    this.bp_partner_version = opts.bp_partner_version;
+
     this.request = new Request(opts.baseUrl || BASE_URL, { r: opts.request });
     log.setLevel(this.logLevel);
   }
@@ -1728,6 +1733,9 @@ export class API extends EventEmitter {
             coin: txp.coin || 'btc',
             network: txp.network || 'livenet',
 
+            bp_partner: this.bp_partner,
+            bp_partner_version: this.bp_partner_version,
+
             // for testing
             request: this.request
           },

packages/bitcore-wallet-client/src/lib/paypro.ts

@@ -280,9 +280,16 @@ export class PayPro {
 
       opts.headers = {
         'Content-Type': JSON_PAYMENT_CONTENT_TYPE,
-        'Accept': JSON_PAYMENT_ACK_CONTENT_TYPE,
+        'Accept': JSON_PAYMENT_ACK_CONTENT_TYPE
       };
 
+      if (opts.bp_partner){
+        opts.headers['BP_PARTNER'] =  opts.bp_partner;
+        if (opts.bp_partner_version){
+          opts.headers['BP_PARTNER_VERSION'] =  opts.bp_partner_version;
+        }
+      }
+
       opts.body = JSON.stringify({
         currency: COIN,
         transactions: [

packages/bitcore-wallet-client/test/api.test.js

@@ -48,6 +48,8 @@ helpers.newClient = (app) => {
   return new Client({
     baseUrl: '/bws/api',
     request: request(app),
+    bp_partner: 'xxx',
+    bp_partner_version: 'yyy',
     //    logLevel: 'debug',
   });
 };
@@ -3322,6 +3324,32 @@ describe('client API', () => {
           });
         });
       });
+
+      it('Should set bp_partner', (done) => {
+        clients[0].getTxProposals({}, (err, txps) => {
+          should.not.exist(err);
+          var changeAddress = txps[0].changeAddress.address;
+          let signatures = keys[0].sign(clients[0].getRootPath(), txps[0]);
+          clients[0].pushSignatures(txps[0], signatures, (err, xx, paypro) => {
+            should.not.exist(err);
+            let signatures = keys[1].sign(clients[1].getRootPath(), txps[0]);
+            clients[1].pushSignatures(xx, signatures, (err, yy, paypro) => {
+              should.not.exist(err);
+
+              yy.status.should.equal('accepted');
+              let spy = sinon.spy(Client.PayPro, 'request');
+              clients[1].broadcastTxProposal(yy, (err, zz, memo) => {
+                should.not.exist(err);
+                var args = spy.lastCall.args[0];
+                var data = args.headers;
+                data.BP_PARTNER.should.equal('xxx');
+                data.BP_PARTNER_VERSION.should.equal('yyy');
+                done();
+              });
+            });
+          });
+        });
+      });
     });