github-tools · j-rewerts · Jun 17, 2019 · Apr 11, 2019 · May 2, 2019 · May 2, 2019
diff --git a/lib/User.js b/lib/User.js
@@ -187,7 +187,7 @@ class User extends Requestable {
     * @return {Promise} - the promise for the http request
     */
    follow(username, cb) {
-      return this._request('PUT', `/user/following/${this.__user}`, null, cb);
+      return this._request('PUT', `/user/following/${username}`, null, cb);
    }
 
    /**
@@ -198,7 +198,7 @@ class User extends Requestable {
     * @return {Promise} - the promise for the http request
     */
    unfollow(username, cb) {
-      return this._request('DELETE', `/user/following/${this.__user}`, null, cb);
+      return this._request('DELETE', `/user/following/${username}`, null, cb);
    }
 
    /**

diff --git a/test/user.spec.js b/test/user.spec.js
@@ -1,3 +1,5 @@
+import expect from 'must';
+
 import Github from '../lib/GitHub';
 import testUser from './fixtures/user.json';
 import {assertSuccessful, assertArray} from './helpers/callbacks';
@@ -76,14 +78,57 @@ describe('User', function() {
       user.listStarredGists(option, assertArray(done));
    });
 
-   it('should follow user', function(done) {
-      user.follow('ingalls', assertSuccessful(done));
-   });
+   describe('following a user', function() {
+      const userToFollow = 'ingalls';
+
+      before(function() {
+         return user.unfollow(userToFollow);
+      })
 
-   it('should unfollow user', function(done) {
-      user.unfollow('ingalls', assertSuccessful(done));
+      it('should follow user', function(done) {
+         user.follow(userToFollow, assertSuccessful(done, function(err, resp) {
+            user._request('GET', `/user/following`, null, assertSuccessful(done, function(err, following) {
+               expect((following.some(user => user['login'] === userToFollow))).to.be.true();
+               done();
+            }));
+         }));
+      });
    });
 
+   describe('following yourself', function() {
+      const userToFollow = testUser.USERNAME;
+
+      before(function() {
+         return user.unfollow(userToFollow);
+      })
+
+      it('should not list yourself as one of your followers', function(done) {
+         user.follow(userToFollow, assertSuccessful(done, function(err, resp) {
+            user._request('GET', `/user/following`, null, assertSuccessful(done, function(err, following) {
+               expect((following.some(user => user['login'] === userToFollow))).to.be.false();
+               done();
+            }));
+         }));
+      });
+   })
+
+   describe('unfollowing a user', function(done) {
+      const userToUnfollow = 'ingalls';
+
+      before(function() {
+         return user.follow(userToUnfollow);
+      })
+
+      it('should unfollow a user', function(done) {
+         user.unfollow(userToUnfollow, assertSuccessful(done, function(err, resp) {
+            user._request('GET', `/user/following`, null, assertSuccessful(done, function(err, following) {
+               expect((following.some(user => user['login'] === userToUnfollow))).to.be.false();
+               done();
+            }));
+         }));
+      });
+   })
+
    it('should list the email addresses of the user', function(done) {
       user.getEmails(assertSuccessful(done));
    });