Merge 6d9e360 into f477660

andela · Jul 27, 2019 · 244d8f6 · 244d8f6
2 parents f477660 + 6d9e360
commit 244d8f6
Show file tree

Hide file tree

Showing 5 changed files with 169 additions and 5 deletions.
diff --git a/src/controllers/userController.js b/src/controllers/userController.js
@@ -1,5 +1,6 @@
 import { User, Follow, UserArchive } from '../db/models';
-import findItem  from '../helpers/findItem';
+import findItem from '../helpers/findItem';
+
 /**
  * @description This class handles user requests
  * @class UserController
@@ -197,7 +198,7 @@ class UserController {
     * @returns {object} - response object
     * @memberof UserController
     */
-  static async changeAccessLevel(req, res) {
+   static async changeAccessLevel(req, res) {
     try {
       const { userRole } = req.body;
       const { userId } = req.params;
@@ -247,6 +248,126 @@ class UserController {
       });
     }
   }
-}  
+
+  /**
+  *
+  *@description Get user followers
+  * @static
+  * @param {object} req
+  * @param {object} res
+  * @returns {object} res
+  * @memberof UserController
+  */
+  static async getUserFollowers(req, res) {
+    try {
+      let offset = 0;
+
+      const { userId } = req.params;
+      const { currentPage, limit } = req.query; // page number
+      const defaultLimit = limit || 10; // number of records per page
+
+      offset = currentPage ? defaultLimit * (currentPage - 1) : 0;
+
+
+      const { count, rows: followers } = await Follow.findAndCountAll({
+        where: {
+          followee: userId
+        },
+        raw: true,
+        attributes: ['follower'],
+        include: [
+          {
+            model: User,
+            as: 'followers',
+            attributes: ['id', 'firstName', 'lastName', 'userName', 'imageUrl']
+          }
+        ],
+        limit: defaultLimit,
+        offset
+      });
+
+      const pages = Math.ceil(count / limit) || 1;
+
+      return res.status(200).json({
+        status: 200,
+        message: 'All followers fetched successfully',
+        data: [ 
+          {
+            followers,
+            totalRating: count,
+            currentPage,
+            limit,
+            totalPages: pages
+          }
+        ]
+      });
+    } catch (error) {
+      return res.status(500).json({
+        status: 500,
+        message: error.message,
+      });
+    }
+  }
+
+  /**
+  *
+  *@description Get user followers
+  * @static
+  * @param {object} req
+  * @param {object} res
+  * @returns {object} res
+  * @memberof UserController
+  */
+  static async getUserFollowee(req, res) {
+    try {
+      let offset = 0;
+
+      const { userId } = req.params;
+      const { currentPage, limit } = req.query; // page number
+      const defaultLimit = limit || 10; // number of records per page
+
+      offset = currentPage ? defaultLimit * (currentPage - 1) : 0;
+
+
+      const { count, rows: followees } = await Follow.findAndCountAll({
+        where: {
+          follower: userId
+        },
+        raw: true,
+        attributes: ['followee'],
+        include: [
+          {
+            model: User,
+            as: 'following',
+            attributes: ['id', 'firstName', 'lastName', 'userName', 'imageUrl']
+          }
+        ],
+        limit: defaultLimit,
+        offset
+      });
+
+      const pages = Math.ceil(count / limit) || 1;
+
+      return res.status(200).json({
+        status: 200,
+        message: 'All followees fetched successfully',
+        data: [ 
+          {
+            followees,
+            totalRating: count,
+            currentPage,
+            limit,
+            totalPages: pages
+          }
+        ]
+      });
+    } catch (error) {
+      return res.status(500).json({
+        status: 500,
+        message: error.message,
+      });
+    }
+  }
+}
 
 export default UserController;
diff --git a/src/db/models/follow.js b/src/db/models/follow.js
@@ -8,7 +8,14 @@ module.exports = (sequelize, DataTypes) => {
   Follow.associate = (models) => {
     // associations can be defined here
     Follow.belongsTo(models.User, {
-      foreignKey: 'id',
+      foreignKey: 'followee',
+      as: 'following',
+      onDelete: 'CASCADE',
+      onUpdate: 'CASCADE'
+    });
+    Follow.belongsTo(models.User, {
+      foreignKey: 'follower',
+      as: 'followers',
       onDelete: 'CASCADE',
       onUpdate: 'CASCADE'
     });

diff --git a/src/db/seeders/20190701155734-user.js b/src/db/seeders/20190701155734-user.js
@@ -23,7 +23,7 @@ module.exports = {
     bio: '#ghost',
     isVerified: true,
     role: 'admin',
-    password: 'password',
+    password: hash,
     verificationToken: '',
     imageUrl: 'image.png'
   }, {

diff --git a/src/routes/userRoute.js b/src/routes/userRoute.js
@@ -49,5 +49,15 @@ router.delete('/:userId',
   findItem.findUser,
   UserController.deleteUser
 );
+router.get(
+  '/followers/:userId',
+  verify,
+  UserController.getUserFollowers
+);
+router.get(
+  '/followees/:userId',
+  verify,
+  UserController.getUserFollowee
+);
 
 export default router;
diff --git a/test/follow.test.js b/test/follow.test.js
@@ -45,6 +45,32 @@ describe('Testing Follow System Controller', () => {
       },
     );
 
+    it(
+      'user should get followers',
+      async () => {
+        const response = await chai.request(app)
+          .get(`${followRoute}followers/1`)
+          .set('token', token)
+          .send();
+        expect(response).to.be.an('object');
+        expect(response).to.have.status(200);
+        expect(response.body).to.have.property('message');
+      },
+    );
+
+    it(
+      'user should get followees',
+      async () => {
+        const response = await chai.request(app)
+          .get(`${followRoute}followees/1`)
+          .set('token', token)
+          .send();
+        expect(response).to.be.an('object');
+        expect(response).to.have.status(200);
+        expect(response.body).to.have.property('message');
+      },
+    );
+
     it(
       'user should not follow themselves',
       async () => {