drop tables afterEach for PG, clean up relation tables too

Author: flovilmart

spec/ParseQuery.spec.js

@@ -848,25 +848,28 @@ describe('Parse.Query testing', () => {
     var makeBoxedNumber = function(num, i) {
       return new BoxedNumber({ number: num, string: strings[i] });
     };
-    Parse.Object.saveAll([3, 1, 3, 2].map(makeBoxedNumber)).then(
-                         function(list) {
-                           var query = new Parse.Query(BoxedNumber);
-                           query.descending("number").addAscending("string");
-                           query.find(expectSuccess({
-                             success: function(results) {
-                               equal(results.length, 4);
-                               equal(results[0].get("number"), 3);
-                               equal(results[0].get("string"), "a");
-                               equal(results[1].get("number"), 3);
-                               equal(results[1].get("string"), "c");
-                               equal(results[2].get("number"), 2);
-                               equal(results[2].get("string"), "d");
-                               equal(results[3].get("number"), 1);
-                               equal(results[3].get("string"), "b");
-                               done();
-                             }
-                           }));
-                         });
+
+    let objects = [3, 1, 3, 2].map(makeBoxedNumber);
+    Parse.Object.saveAll(objects)
+    .then((list) => {
+      var query = new Parse.Query(BoxedNumber);
+      query.descending("number").addAscending("string");
+      return query.find();
+    }).then((results) => {
+      equal(results.length, 4);
+      equal(results[0].get("number"), 3);
+      equal(results[0].get("string"), "a");
+      equal(results[1].get("number"), 3);
+      equal(results[1].get("string"), "c");
+      equal(results[2].get("number"), 2);
+      equal(results[2].get("string"), "d");
+      equal(results[3].get("number"), 1);
+      equal(results[3].get("string"), "b");
+      done();
+    }, (err) => {
+      jfail(err);
+      done();
+    });
   });
 
   it("order by descending number and string", function(done) {

spec/helper.js

@@ -170,7 +170,9 @@ afterEach(function(done) {
     if (Object.keys(openConnections).length > 0) {
       fail('There were open connections to the server left after the test finished');
     }
-    done();
+    on_db('postgres', () => {
+      TestUtils.destroyAllDataPermanently().then(done, done);
+    }, done);
   };
   Parse.Cloud._removeAllHooks();
   databaseAdapter.getAllClasses()

src/Adapters/Storage/Postgres/PostgresStorageAdapter.js

@@ -417,9 +417,17 @@ export class PostgresStorageAdapter {
 
   // Delete all data known to this adapter. Used for testing.
   deleteAllClasses() {
-    return this._client.any('SELECT "className" FROM "_SCHEMA"')
+    return this._client.any('SELECT * FROM "_SCHEMA"')
     .then(results => {
-      const classes = ['_SCHEMA','_PushStatus','_Hooks','_GlobalConfig', ...results.map(result => result.className)];
+      let joins = results.reduce((list, schema) => {
+        Object.keys(schema.schema.fields).forEach((field) => {
+          if (schema.schema.fields[field].type === 'Relation') {
+            list.push(`_Join:${field}:${schema.className}`);
+          }
+        })
+        return list;
+      }, []);
+      const classes = ['_SCHEMA','_PushStatus','_Hooks','_GlobalConfig', ...results.map(result => result.className), ...joins];
       return this._client.tx(t=>t.batch(classes.map(className=>t.none('DROP TABLE IF EXISTS $<className:name>', { className }))));
     }, error => {
       if (error.code === PostgresRelationDoesNotExistError) {