Feature/one time user colors

constants/cardColorIdArray.js

@@ -0,0 +1,12 @@
+module.exports = [
+  { COLOR_ID: 1 },
+  { COLOR_ID: 2 },
+  { COLOR_ID: 3 },
+  { COLOR_ID: 4 },
+  { COLOR_ID: 5 },
+  { COLOR_ID: 6 },
+  { COLOR_ID: 7 },
+  { COLOR_ID: 8 },
+  { COLOR_ID: 9 },
+  { COLOR_ID: 10 },
+];

controllers/migrations.js

@@ -0,0 +1,27 @@
+const userQuery = require("../models/migrations");
+const logger = require("../utils/logger");
+
+/**
+ * Returns the lists of usernames where default colors were added
+ *
+ * @param req {Object} - Express request object
+ * @param res {Object} - Express response object
+ */
+
+const addDefaultColors = async (req, res) => {
+  try {
+    const addedDefaultColorsData = await userQuery.addDefaultColors();
+
+    return res.json({
+      message: "User colors updated successfully!",
+      ...addedDefaultColorsData,
+    });
+  } catch (error) {
+    logger.error(`Error adding default colors to users: ${error}`);
+    return res.boom.badImplementation("Something went wrong. Please contact admin");
+  }
+};
+
+module.exports = {
+  addDefaultColors,
+};

models/migrations.js

@@ -0,0 +1,40 @@
+const firestore = require("../utils/firestore");
+const userModel = firestore.collection("users");
+const cardColorIdArray = require("../constants/cardColorIdArray");
+const { getRandomIndex } = require("../utils/helpers");
+/**
+ * Adds default colors property
+ * @return {Promise<usersMigrated|Object>}
+ */
+
+const addDefaultColors = async () => {
+  try {
+    const usersSnapshot = await userModel.get();
+    const migratedUsers = [];
+    const updateUserPromises = [];
+    const usersArr = [];
+
+    usersSnapshot.forEach((doc) => usersArr.push({ id: doc.id, ...doc.data() }));
+
+    for (const user of usersArr) {
+      const colors = user.colors ? user.colors : {};
+      if (user.colors === undefined) {
+        const userColorIndex = getRandomIndex(cardColorIdArray);
+
+        colors.color_id = cardColorIdArray[parseInt(userColorIndex)].COLOR_ID;
+        updateUserPromises.push(userModel.doc(user.id).set({ ...user, colors }));
+        migratedUsers.push(user.username);
+      }
+    }
+
+    await Promise.all(updateUserPromises);
+    return { count: migratedUsers.length, users: migratedUsers };
+  } catch (err) {
+    logger.error("Error adding default colors to users", err);
+    throw err;
+  }
+};
+
+module.exports = {
+  addDefaultColors,
+};

routes/index.js

@@ -17,5 +17,6 @@ app.use("/trade", require("./trading.js"));
 app.use("/users", require("./users.js"));
 app.use("/profileDiffs", require("./profileDiffs.js"));
 app.use("/wallet", require("./wallets.js"));
+app.use("/migrations", require("./migrations.js"));
 
 module.exports = app;

routes/migrations.js

@@ -0,0 +1,10 @@
+const express = require("express");
+const router = express.Router();
+const authenticate = require("../middlewares/authenticate");
+const authorizeRoles = require("../middlewares/authorizeRoles");
+const { SUPERUSER } = require("../constants/roles");
+const migrations = require("../controllers/migrations");
+
+router.patch("/addDefaultColorProperty", authenticate, authorizeRoles([SUPERUSER]), migrations.addDefaultColors);
+
+module.exports = router;

test/fixtures/user/user.js

@@ -115,6 +115,9 @@ module.exports = () => {
         app_owner: true,
         archived: true,
       },
+      colors: {
+        color_id: 2,
+      },
     },
     {
       username: "mehul",

test/integration/migrations.test.js

@@ -0,0 +1,81 @@
+const chai = require("chai");
+const { expect } = chai;
+const chaiHttp = require("chai-http");
+
+const app = require("../../server");
+const authService = require("../../services/authService");
+const addUser = require("../utils/addUser");
+const cleanDb = require("../utils/cleanDb");
+// Import fixtures
+const userData = require("../fixtures/user/user")();
+const superUser = userData[4];
+const nonSuperUser = userData[0];
+const userWithColorProperty = [userData[5].username];
+const colorBearingUsernames = [superUser.username, nonSuperUser.username];
+
+const config = require("config");
+const cookieName = config.get("userToken.cookieName");
+
+chai.use(chaiHttp);
+
+describe("Migrations", function () {
+  let superUserId;
+  let superUserAuthToken;
+  let userId = "";
+  let nonSuperUserId = "";
+  beforeEach(async function () {
+    userId = await addUser();
+    superUserId = await addUser(superUser);
+    nonSuperUserId = userId;
+    superUserAuthToken = authService.generateAuthToken({ userId: superUserId });
+  });
+
+  afterEach(async function () {
+    await cleanDb();
+  });
+
+  describe("PATCH /migrations/addDefaultColorProperty", function () {
+    it("Should return 401 if user is not a super user", function (done) {
+      const nonSuperUserJwt = authService.generateAuthToken({ userId: nonSuperUserId });
+      chai
+        .request(app)
+        .patch(`/migrations/addDefaultColorProperty`)
+        .set("cookie", `${cookieName}=${nonSuperUserJwt}`)
+        .end((err, res) => {
+          if (err) {
+            return done(err);
+          }
+          expect(res).to.have.status(401);
+          expect(res.body).to.be.a("object");
+          expect(res.body.message).to.equal("You are not authorized for this action.");
+          return done();
+        });
+    });
+    it("Should add default color property to all users,using authorized user (super_user)", function (done) {
+      chai
+        .request(app)
+        .patch(`/migrations/addDefaultColorProperty`)
+        .set("cookie", `${cookieName}=${superUserAuthToken}`)
+        .end((err, res) => {
+          if (err) {
+            return done(err);
+          }
+
+          expect(res).to.have.status(200);
+          expect(res.body.count).to.be.equal(colorBearingUsernames.length);
+          const migratedUsernames = res.body.users;
+
+          expect(migratedUsernames).to.include(
+            colorBearingUsernames[0],
+            "Should add default color property to user without color property"
+          );
+
+          expect(migratedUsernames).to.not.include.any.members(
+            userWithColorProperty,
+            "Should not modify color property of user with color object"
+          );
+          return done();
+        });
+    });
+  });
+});

test/integration/users.test.js

@@ -27,6 +27,7 @@ describe("Users", function () {
     userId = await addUser();
     jwt = authService.generateAuthToken({ userId });
     superUserId = await addUser(superUser);
+
     superUserAuthToken = authService.generateAuthToken({ userId: superUserId });
   });
 

utils/helpers.js

@@ -0,0 +1,12 @@
+/**
+ * Returns a random object from the array of colors to user
+ * @param array {array} : array containing objects
+ * @returns random Index number : index between the range 0 to array.length
+ */
+const getRandomIndex = (array = []) => {
+  return Math.floor(Math.random() * array.length);
+};
+
+module.exports = {
+  getRandomIndex,
+};