Exit with error code when create-test-tables fails

charmander · charmander · commit 755b3929484c · 2019-11-20T11:05:06.000-08:00
diff --git a/script/create-test-tables.js b/script/create-test-tables.js
@@ -38,15 +38,25 @@ var con = new pg.Client({
   password: args.password,
   database: args.database
 })
-con.connect()
-var query = con.query('drop table if exists person')
-con.query('create table person(id serial, name varchar(10), age integer)', (err, res) => {
-  console.log('Created table person')
-  console.log('Filling it with people')
-})
-people.map(function (person) {
-  return con.query(new pg.Query("insert into person(name, age) values('" + person.name + "', '" + person.age + "')"))
-}).pop().on('end', function () {
-  console.log('Inserted 26 people')
-  con.end()
+
+con.connect((err) => {
+  if (err) {
+    throw err
+  }
+
+  con.query(
+    'DROP TABLE IF EXISTS person;'
+    + ' CREATE TABLE person (id serial, name varchar(10), age integer);'
+    + people
+      .map((person) =>
+        ` INSERT INTO person (name, age) VALUES ('${person.name}', ${person.age});`)
+      .join(''),
+    (err, result) => {
+      if (err) {
+        throw err
+      }
+
+      console.log(`Inserted ${result.rowCount} people`)
+      con.end()
+    })
 })
