Updated the logic to return 404 instead of 200 when wrong taskRequest Id is sent

models/taskRequests.js

@@ -171,16 +171,21 @@ const fetchTaskRequestById = async (taskRequestId) => {
     const taskRequestSnapshot = await taskRequestsCollection.doc(taskRequestId).get();
     const taskRequestData = taskRequestSnapshot.data();
 
-    if (taskRequestData) {
-      taskRequestData.id = taskRequestSnapshot.id;
-      taskRequestData.url = new URL(`/taskRequests/${taskRequestData.id}`, config.get("services.rdsUi.baseUrl"));
+    if (!taskRequestData) {
+      return {
+        taskRequestExists: false,
+      };
     }
+
+    taskRequestData.id = taskRequestSnapshot.id;
+    taskRequestData.url = new URL(`/taskRequests/${taskRequestData.id}`, config.get("services.rdsUi.baseUrl"));
+
     return {
       taskRequestData,
       taskRequestExists: true,
     };
   } catch (err) {
-    logger.error("Error in updating task", err);
+    logger.error("Error in fetching taskRequest by Id", err);
   }
 
   return {

test/integration/taskRequests.test.js

@@ -176,6 +176,22 @@ describe("Task Requests", function () {
           });
       });
 
+      it("should return 404 if the task request is not found for the given Id", function (done) {
+        chai
+          .request(app)
+          .get(`/taskRequests/1234`)
+          .set("cookie", `${cookieName}=${jwt}`)
+          .end((err, res) => {
+            if (err) {
+              return done(err);
+            }
+
+            expect(res).to.have.status(404);
+            expect(res.body.message).to.be.equal("Task request not found");
+            return done();
+          });
+      });
+
       it("should return 404 if the resource is not found", function (done) {
         sinon.stub(taskRequestsModel, "fetchTaskRequestById").callsFake(() => []);
 
@@ -196,21 +212,26 @@ describe("Task Requests", function () {
     });
 
     describe("When the user is not a super user", function () {
+      let taskRequestId;
+
       before(async function () {
         userId = await addUser(member);
-        sinon.stub(authService, "verifyAuthToken").callsFake(() => ({ userId }));
-        jwt = authService.generateAuthToken({ userId });
+        sinon.stub(authService, "verifyAuthToken").callsFake(() => ({
+          userId,
+        }));
+        jwt = authService.generateAuthToken({
+          userId,
+        });
 
         taskId = (await tasksModel.updateTask(taskData[4])).taskId;
-
         await userStatusModel.updateUserStatus(userId, idleUserStatus);
-        await taskRequestsModel.addOrUpdate(taskId, userId);
+        taskRequestId = (await taskRequestsModel.addOrUpdate(taskId, userId)).id;
       });
 
       it("should be successful when the user is not a super user", function (done) {
         chai
           .request(app)
-          .get(`/taskRequests/taskrequstid`)
+          .get(`/taskRequests/${taskRequestId}`)
           .set("cookie", `${cookieName}=${jwt}`)
           .end((err, res) => {
             if (err) {