From a16b652146b5dc70c67130e5fa379516ab13af4b Mon Sep 17 00:00:00 2001
From: Pierre Theo Klein <pierre.klein@mail.mcgill.ca>
Date: Tue, 26 Feb 2019 15:53:59 -0500
Subject: [PATCH] Clean up code in seed files

---
 seed/index.js      | 26 +++++++++++++-------------
 seed/roles.seed.js |  2 +-
 2 files changed, 14 insertions(+), 14 deletions(-)

diff --git a/seed/index.js b/seed/index.js
index 09858612..01ab7b8f 100644
--- a/seed/index.js
+++ b/seed/index.js
@@ -9,21 +9,21 @@ const Seed = {
 
 const db = require("../services/database.service");
 //connect to db
-db.connect(undefined, onConnected);
-
-//called when the db is connected
-function onConnected() {
-    dropAll().then(() => {
-        console.log("Finished dropping");
-        storeAll().then(() => {
-            console.log("Finished seeding");
-        }).catch((error) => {
-            console.log(error);
-        });
-    }, (err) => {
-        console.error(err);
+db.connect(undefined, () => {
+    onConnected().catch((reason) => {
+        console.error(reason);
+        process.exit(1);
+    }).then(() => {
+        process.exit(0);
     });
+});
 
+//called when the db is connected
+async function onConnected() {
+    await dropAll();
+    console.log("Finished dropping");
+    await storeAll();
+    console.log("Finished seeding");
 }
 
 async function dropAll() {
diff --git a/seed/roles.seed.js b/seed/roles.seed.js
index 867e6c31..eeddf709 100644
--- a/seed/roles.seed.js
+++ b/seed/roles.seed.js
@@ -14,7 +14,7 @@ if (envLoadResult.error) {
  * Drops all elements in Role
  */
 function dropAll() {
-    return Role.remove({});
+    return Role.deleteMany({});
 }
 
 /**