test: add test for default registration of function module

License: MIT
Signed-off-by: Alan Shaw <alan@tableflip.io>

test/peer-discovery.node.js

@@ -58,8 +58,8 @@ describe('peer discovery', () => {
     })
   }
 
-  describe('module registration', () => {
-    it('should enable module by default', (done) => {
+  describe.only('module registration', () => {
+    it('should enable by default a module passed as an object', (done) => {
       const mockDiscovery = {
         on: sinon.stub(),
         start: sinon.stub().callsArg(0),
@@ -79,6 +79,28 @@ describe('peer discovery', () => {
       })
     })
 
+    it('should enable by default a module passed as a function', (done) => {
+      const mockDiscovery = {
+        on: sinon.stub(),
+        start: sinon.stub().callsArg(0),
+        stop: sinon.stub().callsArg(0)
+      }
+
+      const MockDiscovery = sinon.stub().returns(mockDiscovery)
+
+      const options = { modules: { peerDiscovery: [ MockDiscovery ] } }
+
+      createNode(['/ip4/0.0.0.0/tcp/0'], options, (err, node) => {
+        expect(err).to.not.exist()
+
+        node.start((err) => {
+          expect(err).to.not.exist()
+          expect(mockDiscovery.start.called).to.be.true()
+          node.stop(done)
+        })
+      })
+    })
+
     it('should enable module by configutation', (done) => {
       const mockDiscovery = {
         on: sinon.stub(),