check if cohort id exist or not

relates #29
GSG-G8 · Mar 30, 2020 · c613d5b · c613d5b
1 parent fa8c6f0
commit c613d5b
Showing 1 changed file with 11 additions and 5 deletions.
diff --git a/server/controllers/routes/admin/cohort/deleteCohort.js b/server/controllers/routes/admin/cohort/deleteCohort.js
@@ -2,11 +2,17 @@ const deleteCohort = require('../../../../database/queries/cohort/deleteCohort')
 
 const cohortDelete = async (req, res, next) => {
   try {
-    await deleteCohort(req.params.cohortId);
-    res.json({
-      StatusCode: 200,
-      data: { message: 'Cohort deleted successfully' },
-    });
+    const check = await deleteCohort(req.params.cohortId);
+    if (check.rowCount !== 0) {
+      res.json({
+        StatusCode: 200,
+        data: { message: 'Cohort deleted successfully' },
+      });
+    } else {
+      const err = new Error();
+      err.message = 'cohort id does not exist';
+      next(err);
+    }
   } catch (err) {
     next(err);
   }