ModuleManager unit tests 🔧

GLIBX · Jul 9, 2021 · d3f747f · d3f747f
1 parent 58b58e8
commit d3f747f
Showing 1 changed file with 71 additions and 0 deletions.
diff --git a/packages/bitcore-node/test/unit/modules.spec.ts b/packages/bitcore-node/test/unit/modules.spec.ts
@@ -1,4 +1,6 @@
 import { expect } from 'chai';
+import sinon from 'sinon';
+import { Modules } from '../../src/modules';
 import { ChainStateProvider } from '../../src/providers/chain-state';
 import { Libs } from '../../src/providers/libs';
 import { Config } from '../../src/services/config';
@@ -10,6 +12,27 @@ describe('Modules', function() {
   before(unitBeforeHelper);
   after(unitAfterHelper);
 
+  it('should load configured modules correctly', () => {
+    const sandbox = sinon.createSandbox();
+    sandbox.stub(Config, 'get').returns(mockConfig);
+
+    validateModules();
+    sandbox.restore();
+  });
+
+  it('should prevent double module registration', () => {
+    const sandbox = sinon.createSandbox();
+    sandbox.stub(Config, 'get').returns({
+      // Double module registration test case. ModuleManager will attempt to register BTC Module from
+      // "config.modules" as well as "config.chains". Should only register once.
+      modules: ['./bitcoin'],
+      ...mockConfig
+    });
+
+    validateModules();
+    sandbox.restore();
+  });
+
   it('should have a test which runs', function() {
     expect(true).to.equal(true);
   });
@@ -46,3 +69,51 @@ describe('Modules', function() {
     }
   });
 });
+
+const mockConfig = {
+  chains: {
+    BTC: {
+      testnet: {
+        chainSource: 'p2p',
+        trustedPeers: [
+          {
+            host: '127.0.0.1',
+            port: 18333
+          }
+        ],
+        rpc: {
+          host: '127.0.0.1',
+          port: 18332,
+          username: 'bitpaytest',
+          password: 'local321'
+        }
+      },
+    },
+    ETH: {
+      dev: {
+        trustedPeers: [
+          {
+            host: '127.0.0.1',
+            port: 8545
+          }
+        ],
+        chainSource: 'p2p',
+        provider: {
+          protocol: 'http',
+          host: '127.0.0.1',
+          port: 8545,
+          chain: 'ETH'
+        }
+      }
+    }
+  }
+};
+
+const validateModules = () => {
+  Modules.internalServices = []; // Remove all loaded modules from internalServices array for a fresh load
+  Modules.loadConfigured(); // Re-load modules with stubbed Config.get()
+
+  expect(Modules.internalServices.length).to.equal(2);
+  expect(Modules.internalServices[0].constructor.name).to.equal('BitcoinModule');
+  expect(Modules.internalServices[1].constructor.name).to.equal('ETHModule');
+};