added error checking

StephenGrider · StephenGrider · commit b2cdb1a408c8 · 2019-07-17T17:17:47.000-07:00
diff --git a/server/routes/surveyRoutes.js b/server/routes/surveyRoutes.js
@@ -7,7 +7,7 @@ const surveyTemplate = require('../services/emailTemplates/surveyTemplate');
 const Survey = mongoose.model('surveys');
 
 module.exports = app => {
-  app.post('/api/surveys', requireLogin, requireCredits, (req, res) => {
+  app.post('/api/surveys', requireLogin, requireCredits, async (req, res) => {
     const { title, subject, body, recipients } = req.body;
 
     const survey = new Survey({
@@ -21,6 +21,16 @@ module.exports = app => {
 
     // Great place to send an email!
     const mailer = new Mailer(survey, surveyTemplate(survey));
-    mailer.send();
+
+    try {
+      await mailer.send();
+      await survey.save();
+      req.user.credits -= 1;
+      const user = await req.user.save();
+
+      res.send(user);
+    } catch (err) {
+      res.status(422).send(err);
+    }
   });
 };
diff --git a/server/services/Mailer.js b/server/services/Mailer.js
@@ -47,7 +47,7 @@ class Mailer extends helper.Mail {
       body: this.toJSON()
     });
 
-    const response = this.sgApi.API(request);
+    const response = await this.sgApi.API(request);
     return response;
   }
 }

Original file line number	Diff line number	Diff line change
`@@ -47,7 +47,7 @@ class Mailer extends helper.Mail {`
`47`	`47`	`body: this.toJSON()`
`48`	`48`	`});`
`49`	`49`
`50`		`- const response = this.sgApi.API(request);`
	`50`	`+ const response = await this.sgApi.API(request);`
`51`	`51`	`return response;`
`52`	`52`	`}`
`53`	`53`	`}`