feat: add fn _isEnterpriseUser (#35)

Author: pritamstyz4ever

index.js

@@ -478,6 +478,17 @@ class ScmRouter extends Scm {
     _openPr(config) {
         return this.chooseScm(config).then(scm => scm.openPr(config));
     }
+
+    /**
+     * Check if user belongs to an enterprise
+     * @method _isEnterpriseUser
+     * @param  {Object}     config              Configuration
+     * @param  {String}     config.scmContext   Name of scm context
+     * @return {Boolean}                        True if user belongs to an enterprise
+     */
+    _isEnterpriseUser(config) {
+        return this.chooseScm(config).then(scm => scm.isEnterpriseUser(config));
+    }
 }
 
 module.exports = ScmRouter;

test/index.test.js

@@ -75,6 +75,7 @@ describe('index test', () => {
         mock.getReadOnlyInfo = sinon.stub().returns(plugin);
         mock.autoDeployKeyGenerationEnabled = sinon.stub().returns(plugin);
         mock.getWebhookEventsMapping = sinon.stub().returns({ pr: 'pull_request' });
+        mock.isEnterpriseUser = sinon.stub().returns(true);
 
         return mock;
     };
@@ -964,4 +965,17 @@ describe('index test', () => {
                 assert.calledWith(exampleScm.openPr, config);
             }));
     });
+
+    describe('_isEnterpriseUser', () => {
+        const config = { scmContext: exampleScmContext };
+
+        it('call origin _isEnterpriseUser', () =>
+            scm._isEnterpriseUser(config).then(result => {
+                assert.strictEqual(result, true);
+                assert.notCalled(scmGithub.isEnterpriseUser);
+                assert.notCalled(scmGitlab.isEnterpriseUser);
+                assert.calledOnce(exampleScm.isEnterpriseUser);
+                assert.calledWith(exampleScm.isEnterpriseUser, config);
+            }));
+    });
 });