check if cohort id not found

relates #24
GSG-G8 · Mar 30, 2020 · 3a0fd47 · 3a0fd47
1 parent 82105fb
commit 3a0fd47
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');
 
 const cohortDelete = async (req, res, next) => {
   try {
-    await deleteCohort(req.params.cohortId);
-    res.json({
-      StatusCode: 200,
-      data: { message: 'Cohort deleted successfully' },
-    });
+    const test = await deleteCohort(req.params.cohortId);
+    if (test.rowCount !== 0) {
+      res.json({
+        StatusCode: 200,
+        data: { message: 'Cohort deleted successfully' },
+      });
+    } else {
+      const err = new Error();
+      err.msg = 'chohrt id not found';
+      next(err);
+    }
   } catch (err) {
     next(err);
   }