added base test for getContexts

Signed-off-by: Jeromy Cannon <[email protected]>

Author: jeromy-cannon

src/commands/base.ts

@@ -271,4 +271,18 @@ export abstract class BaseCommand extends ShellRunner {
     // return the consensus nodes
     return consensusNodes;
   }
+
+  /**
+   * Gets a list of distinct contexts from the consensus nodes
+   * @returns an array of context strings
+   */
+  public getContexts(): string[] {
+    const contexts: string[] = [];
+    this.getConsensusNodes().forEach(node => {
+      if (!contexts.includes(node.context)) {
+        contexts.push(node.context);
+      }
+    });
+    return contexts;
+  }
 }

test/unit/commands/base.test.ts

@@ -127,7 +127,7 @@ describe('BaseCommand', () => {
   });
 
   describe('get consensus nodes', () => {
-    it('should return consensus nodes', () => {
+    before(() => {
       const testLogger = sinon.stub();
       const helm = sinon.stub();
       const chartManager = sinon.stub();
@@ -157,7 +157,9 @@ describe('BaseCommand', () => {
         localConfig,
         remoteConfigManager,
       });
+    });
 
+    it('should return consensus nodes', () => {
       const consensusNodes = baseCmd.getConsensusNodes();
       expect(consensusNodes).to.be.an('array');
       expect(consensusNodes[0].context).to.equal('context1');
@@ -171,5 +173,12 @@ describe('BaseCommand', () => {
       expect(consensusNodes[0].cluster).to.equal('cluster');
       expect(consensusNodes[1].cluster).to.equal('cluster2');
     });
+
+    it('should return contexts', () => {
+      const contexts = baseCmd.getContexts();
+      expect(contexts).to.be.an('array');
+      expect(contexts[0]).to.equal('context1');
+      expect(contexts[1]).to.equal('context2');
+    });
   });
 });