From df5d7143f7a7ce0e7de490dc215cdc96b92ae316 Mon Sep 17 00:00:00 2001
From: Kamoris <kamil.slowiena@rtbhouse.com>
Date: Fri, 22 May 2020 16:09:00 +0200
Subject: [PATCH] [rtbhouse] Add schain support (#5281)

---
 modules/rtbhouseBidAdapter.js                | 50 ++++++++++++++++++--
 test/spec/modules/rtbhouseBidAdapter_spec.js | 42 ++++++++++++++++
 2 files changed, 89 insertions(+), 3 deletions(-)

diff --git a/modules/rtbhouseBidAdapter.js b/modules/rtbhouseBidAdapter.js
index ca760ca49eb..3337c3f1b59 100644
--- a/modules/rtbhouseBidAdapter.js
+++ b/modules/rtbhouseBidAdapter.js
@@ -46,9 +46,7 @@ export const spec = {
       site: mapSite(validBidRequests, bidderRequest),
       cur: DEFAULT_CURRENCY_ARR,
       test: validBidRequests[0].params.test || 0,
-      source: {
-        tid: validBidRequests[0].transactionId
-      }
+      source: mapSource(validBidRequests[0]),
     };
     if (bidderRequest && bidderRequest.gdprConsent && bidderRequest.gdprConsent.gdprApplies) {
       const consentStr = (bidderRequest.gdprConsent.consentString)
@@ -145,6 +143,52 @@ function mapSite(slot, bidderRequest) {
   }
 }
 
+/**
+ * @param {object} slot Ad Unit Params by Prebid
+ * @returns {object} Source by OpenRTB 2.5 §3.2.2
+ */
+function mapSource(slot) {
+  const source = {
+    tid: slot.transactionId,
+  };
+  const schain = mapSchain(slot.schain);
+  if (schain) {
+    source.ext = {
+      schain: schain
+    }
+  }
+  return source;
+}
+
+/**
+ * @param {object} schain object set by Publisher
+ * @returns {object} OpenRTB SupplyChain object
+ */
+function mapSchain(schain) {
+  if (!schain) {
+    return null;
+  }
+  if (!validateSchain(schain)) {
+    utils.logError('RTB House: required schain params missing');
+    return null;
+  }
+  return schain;
+}
+
+/**
+ * @param {object} schain object set by Publisher
+ * @returns {object} bool
+ */
+function validateSchain(schain) {
+  if (!schain.nodes) {
+    return false;
+  }
+  const requiredFields = ['asi', 'sid', 'hp'];
+  return schain.nodes.every(node => {
+    return requiredFields.every(field => node[field]);
+  });
+}
+
 /**
  * @param {object} slot Ad Unit Params by Prebid
  * @returns {object} Request by OpenRTB Native Ads 1.1 §4
diff --git a/test/spec/modules/rtbhouseBidAdapter_spec.js b/test/spec/modules/rtbhouseBidAdapter_spec.js
index 93b9692d2e2..efaed87dd15 100644
--- a/test/spec/modules/rtbhouseBidAdapter_spec.js
+++ b/test/spec/modules/rtbhouseBidAdapter_spec.js
@@ -69,6 +69,18 @@ describe('RTBHouseAdapter', () => {
         'bidderRequestId': '22edbae2733bf6',
         'auctionId': '1d1a030790a475',
         'transactionId': 'example-transaction-id',
+        'schain': {
+          'ver': '1.0',
+          'complete': 1,
+          'nodes': [
+            {
+              'asi': 'directseller.com',
+              'sid': '00001',
+              'rid': 'BidRequest1',
+              'hp': 1
+            }
+          ]
+        }
       }
     ];
     const bidderRequest = {
@@ -173,6 +185,36 @@ describe('RTBHouseAdapter', () => {
       expect(data.imp[0].bidfloor).to.equal(0.01)
     });
 
+    it('should include source.ext.schain in request', () => {
+      const bidRequest = Object.assign([], bidRequests);
+      const request = spec.buildRequests(bidRequest, bidderRequest);
+      const data = JSON.parse(request.data);
+      expect(data.source.ext.schain).to.deep.equal({
+        'ver': '1.0',
+        'complete': 1,
+        'nodes': [
+          {
+            'asi': 'directseller.com',
+            'sid': '00001',
+            'rid': 'BidRequest1',
+            'hp': 1
+          }
+        ]
+      });
+    });
+
+    it('should not include invalid schain', () => {
+      const bidRequest = Object.assign([], bidRequests);
+      bidRequest[0].schain = {
+        'nodes': [{
+          'unknown_key': 1
+        }]
+      };
+      const request = spec.buildRequests(bidRequest, bidderRequest);
+      const data = JSON.parse(request.data);
+      expect(data.source).to.not.have.property('ext');
+    });
+
     describe('native imp', () => {
       function basicRequest(extension) {
         return Object.assign({