fix(switchDatabase): scope configuration not working with schema

Signed-off-by: Jéssica Schissato <jessicaschissato@gmail.com>
db-migrate · Jul 10, 2023 · d52cdcf · d52cdcf
1 parent b64a02a
commit d52cdcf
Show file tree

Hide file tree

Showing 2 changed files with 34 additions and 2 deletions.
diff --git a/index.js b/index.js
@@ -132,12 +132,18 @@ var PgDriver = Base.extend({
           'Ignore database option, not available with postgres. Use schema instead!'
         );
         this.runSql(
-          util.format('SET search_path TO `%s`', options.database),
+          util.format('SET search_path TO %s', options.database),
+          callback
+        );
+      } else if (typeof options.schema === 'string') {
+        this.schema = options.schema
+        this.runSql(
+          util.format('SET search_path TO %s', options.schema),
           callback
         );
       }
     } else if (typeof options === 'string') {
-      this.runSql(util.format('SET search_path TO `%s`', options), callback);
+      this.runSql(util.format('SET search_path TO %s', options), callback);
     } else callback(null);
   },
 

diff --git a/test/pg_test.js b/test/pg_test.js
@@ -1157,6 +1157,32 @@ vows
       }
     }
   })
+  .addBatch({
+    switchDatabase: {
+      topic: function () {
+        db.switchDatabase({ schema: 'test_schema2' }, this.callback);
+      },
+
+      'has search path': {
+        topic: function () {
+          db.runSql('SHOW search_path', this.callback);
+        },
+
+        'containing the new schema': function (err, result) {
+          assert.isNull(err);
+          var rows = result.rows;
+          assert.isNotNull(rows);
+          assert.strictEqual(rows.length, 1);
+          var row = rows[0];
+          assert.strictEqual(row.search_path, 'test_schema2');
+        }
+      },
+
+      teardown: function () {
+        db.switchDatabase({ schema: config.schema }, this.callback);
+      }
+    }
+  })
   .export(module);
 
 function findByName (columns, name) {