fix: update dht tests

Author: vasco-santos

js/src/dht/get.js

@@ -1,6 +1,7 @@
 /* eslint-env mocha */
 'use strict'
 
+const hat = require('hat')
 const waterfall = require('async/waterfall')
 const { spawnNodesWithId } = require('../utils/spawn')
 const { getDescribe, getIt, expect } = require('../utils/mocha')
@@ -30,7 +31,6 @@ module.exports = (createCommon, options) => {
 
           nodeA = nodes[0]
           nodeB = nodes[1]
-
           connect(nodeA, nodeB.peerId.addresses[0], done)
         })
       })
@@ -39,29 +39,54 @@ module.exports = (createCommon, options) => {
     after((done) => common.teardown(done))
 
     it('should error when getting a non-existent key from the DHT', (done) => {
-      nodeA.dht.get('non-existing', { timeout: '100ms' }, (err, value) => {
+      nodeA.dht.get('non-existing', { timeout: '100ms' }, (err) => {
         expect(err).to.be.an.instanceof(Error)
         done()
       })
     })
 
-    it('should get a value after it was put on another node', function (done) {
+    it('should get a value after it was added on another node', function (done) {
       this.timeout(80 * 1000)
 
-      // TODO - this test needs to keep tryingl instead of the setTimeout
       waterfall([
-        (cb) => nodeB.object.new('unixfs-dir', cb),
+        (cb) => nodeB.object.put(Buffer.from(hat()), cb),
         (dagNode, cb) => setTimeout(() => cb(null, dagNode), 20000),
         (dagNode, cb) => {
           const multihash = dagNode.toJSON().multihash
 
-          nodeA.dht.get(multihash, cb)
+          nodeA.object.get(multihash, cb)
         },
         (result, cb) => {
-          expect(result).to.eql('')
+          expect(result).to.exist()
           cb()
         }
       ], done)
     })
+
+    it('should get a value after it was put on another node', function (done) {
+      this.timeout(80 * 1000)
+      const multihash = Buffer.from('/v/hello')
+      const data = Buffer.from('data')
+
+      // Rewrite validators to simply validate the record
+      nodeA._libp2pNode._dht.validators.v = nodeB._libp2pNode._dht.validators.v = {
+        func (key, publicKey, callback) {
+          setImmediate(callback)
+        },
+        sign: false
+      }
+
+      // Rewrite selectors to select first received record
+      nodeA._libp2pNode._dht.selectors.v = () => 0
+
+      waterfall([
+        (cb) => nodeB.dht.put(multihash, data, cb),
+        (cb) => nodeA.dht.get(multihash, (err, res) => {
+          expect(err).to.not.exist()
+          expect(res).to.eql(data)
+          cb()
+        })
+      ], done)
+    })
   })
 }

js/src/dht/put.js

@@ -1,29 +1,50 @@
 /* eslint-env mocha */
 'use strict'
 
+const { spawnNodesWithId } = require('../utils/spawn')
 const { getDescribe, getIt, expect } = require('../utils/mocha')
+const { connect } = require('../utils/swarm')
 
 module.exports = (createCommon, options) => {
   const describe = getDescribe(options)
   const it = getIt(options)
   const common = createCommon()
 
   describe('.dht.put', function () {
+    this.timeout(80 * 1000)
+
+    let nodeA
+    let nodeB
+
     before(function (done) {
       // CI takes longer to instantiate the daemon, so we need to increase the
       // timeout for the before step
       this.timeout(60 * 1000)
 
       common.setup((err, factory) => {
         expect(err).to.not.exist()
-        done()
+
+        spawnNodesWithId(2, factory, (err, nodes) => {
+          expect(err).to.not.exist()
+
+          nodeA = nodes[0]
+          nodeB = nodes[1]
+          connect(nodeA, nodeB.peerId.addresses[0], done)
+        })
       })
     })
 
     after((done) => common.teardown(done))
 
-    it.skip('should put a value on the DHT', (done) => {
-      // TODO: implement me
+    it('should put a value on the DHT and it become provided by the peer', (done) => {
+      this.timeout(80 * 1000)
+      const key = Buffer.from('QmUNLLsPACCz1vLxQVkXqqLX5R1X345qqfHbsf67hvA3Nn')
+      const data = Buffer.from('data')
+
+      nodeA.dht.put(key, data, (err) => {
+        expect(err).to.not.exist()
+        done()
+      })
     })
   })
 }