From 4f7999d82d376635d844bdf8f8bceaec0f283764 Mon Sep 17 00:00:00 2001
From: Henrique Dias <hacdias@gmail.com>
Date: Tue, 30 Jan 2018 05:02:14 +0000
Subject: [PATCH] fix: update stats API (#684)

* implement stats

* update tests
---
 src/bitswap/stat.js  | 18 ++++++++++++++++--
 test/bitswap.spec.js | 15 +++++++++------
 2 files changed, 25 insertions(+), 8 deletions(-)

diff --git a/src/bitswap/stat.js b/src/bitswap/stat.js
index e29d9c7ad..733b99617 100644
--- a/src/bitswap/stat.js
+++ b/src/bitswap/stat.js
@@ -2,10 +2,24 @@
 
 const promisify = require('promisify-es6')
 
+const transform = function (res, callback) {
+  callback(null, {
+    provideBufLen: res.ProvideBufLen,
+    wantlist: res.Wantlist,
+    peers: res.Peers,
+    blocksReceived: res.BlocksReceived,
+    dataReceived: res.DataReceived,
+    blocksSent: res.BlocksSent,
+    dataSent: res.DataSent,
+    dupBlksReceived: res.DupBlksReceived,
+    dupDataReceived: res.DupDataReceived
+  })
+}
+
 module.exports = (send) => {
   return promisify((callback) => {
-    send({
+    send.andTransform({
       path: 'bitswap/stat'
-    }, callback)
+    }, transform, callback)
   })
 }
diff --git a/test/bitswap.spec.js b/test/bitswap.spec.js
index 46051e258..d8364f464 100644
--- a/test/bitswap.spec.js
+++ b/test/bitswap.spec.js
@@ -41,12 +41,15 @@ describe('.bitswap', function () {
   it('.stat', (done) => {
     ipfs.bitswap.stat((err, res) => {
       expect(err).to.not.exist()
-      expect(res).to.have.property('BlocksReceived')
-      expect(res).to.have.property('DupBlksReceived')
-      expect(res).to.have.property('DupDataReceived')
-      expect(res).to.have.property('Peers')
-      expect(res).to.have.property('ProvideBufLen')
-      expect(res).to.have.property('Wantlist')
+      expect(res).to.have.a.property('provideBufLen')
+      expect(res).to.have.a.property('wantlist')
+      expect(res).to.have.a.property('peers')
+      expect(res).to.have.a.property('blocksReceived')
+      expect(res).to.have.a.property('dataReceived')
+      expect(res).to.have.a.property('blocksSent')
+      expect(res).to.have.a.property('dataSent')
+      expect(res).to.have.a.property('dupBlksReceived')
+      expect(res).to.have.a.property('dupDataReceived')
 
       done()
     })