Alternate endpoint accessor that falls back to dotcom

Author: smashwilson

lib/models/remote.js

@@ -1,4 +1,4 @@
-import {getEndpoint} from './endpoint';
+import {getEndpoint, DOTCOM} from './endpoint';
 
 export default class Remote {
   constructor(name, url) {
@@ -57,6 +57,10 @@ export default class Remote {
     return this.domain === null ? null : getEndpoint(this.domain);
   }
 
+  getEndpointOrDotcom() {
+    return this.getEndpoint() || DOTCOM;
+  }
+
   isPresent() {
     return true;
   }
@@ -135,6 +139,10 @@ export const nullRemote = {
     return null;
   },
 
+  getEndpointOrDotcom() {
+    return DOTCOM;
+  },
+
   isPresent() {
     return false;
   },

test/models/remote.test.js

@@ -76,6 +76,7 @@ describe('Remote', function() {
     assert.isNull(nullRemote.getSlug());
     assert.strictEqual(nullRemote.getNameOr('else'), 'else');
     assert.isNull(nullRemote.getEndpoint());
+    assert.strictEqual(nullRemote.getEndpointOrDotcom().getGraphQLRoot(), 'https://api.github.com/graphql');
   });
 
   describe('getEndpoint', function() {
@@ -89,4 +90,16 @@ describe('Remote', function() {
       assert.isNull(elsewhere.getEndpoint());
     });
   });
+
+  describe('getEndpointOrDotcom', function() {
+    it('accesses the same Endpoint for the corresponding GitHub host', function() {
+      const remote = new Remote('origin', '[email protected]:atom/github.git');
+      assert.strictEqual(remote.getEndpointOrDotcom().getGraphQLRoot(), 'https://api.github.com/graphql');
+    });
+
+    it('returns dotcom for non-GitHub URLs', function() {
+      const elsewhere = new Remote('mirror', 'https://[email protected]/team/repo.git');
+      assert.strictEqual(elsewhere.getEndpointOrDotcom().getGraphQLRoot(), 'https://api.github.com/graphql');
+    });
+  });
 });