From 4b7eb453e81ac47a87087587a3e94f125c184b0e Mon Sep 17 00:00:00 2001
From: Deepti Singh <136986628+deepti-50@users.noreply.github.com>
Date: Tue, 19 Dec 2023 18:13:23 +0530
Subject: [PATCH] Update attendance.js

---
 controller/attendance.js | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/controller/attendance.js b/controller/attendance.js
index ce86ab9..b98d726 100644
--- a/controller/attendance.js
+++ b/controller/attendance.js
@@ -5,6 +5,9 @@ import {
   getAttendances,
 } from "#services/attendance";
 import { logger } from "#util";
+import { isEntityIdValid } from "#middleware/entityIdValidation";
+import Student from "#models/attendance";
+import Course from "#models/course";
 
 async function addAttendance(req, res) {
   const {
@@ -15,8 +18,15 @@ async function addAttendance(req, res) {
     cumulativeAttended,
     cumulativeOccured,
   } = req.body;
+  const isStudentValid = await isEntityIdValid(student, Student);
+  const isCourseValid = await isEntityIdValid(course, Course);
   try {
     // eslint-disable-next-line max-len
+    if (!isStudentValid || !isCourseValid) {
+      res.status(400).json({
+        error: "Invalid Id",
+      });
+    }
     const attendance = await addNewAttendance(
       student,
       course,