From 567f8bf94eba5f02db08c83a0e269f4e5d9b0b5f Mon Sep 17 00:00:00 2001
From: Wyatt Johnson <wyattjoh@gmail.com>
Date: Mon, 7 Aug 2017 14:54:11 +1000
Subject: [PATCH] adjusted wrapResponse function

---
 graph/helpers/response.js | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

diff --git a/graph/helpers/response.js b/graph/helpers/response.js
index f5e7a54f4..1db1e9b89 100644
--- a/graph/helpers/response.js
+++ b/graph/helpers/response.js
@@ -6,15 +6,17 @@ const {Error: {ValidationError}} = require('mongoose');
  * keyed at `key` or an error caught at `errors`.
  */
 
-const wrapResponse = (key) => (promiseOrValue) => {
-  return Promise.resolve(promiseOrValue).then((value) => {
+const wrapResponse = (key) => async (promise) => {
+  try {
+    let value = await promise;
+    
     let res = {};
     if (key) {
       res[key] = value;
     }
+
     return res;
-  })
-  .catch((err) => {
+  } catch (err) {
     if (err instanceof errors.APIError) {
       return {
         errors: [err]
@@ -26,7 +28,7 @@ const wrapResponse = (key) => (promiseOrValue) => {
     }
 
     throw err;
-  });
+  }
 };
 
 module.exports = wrapResponse;