Fix loopbackio#123: Set default value during autoupdate.

This is a mergeable update to loopbackio#153 that also respects dbDefault.
STRML · Sep 12, 2016 · 9dacaab · 9dacaab
1 parent 7273f41
commit 9dacaab
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 6 deletions.
diff --git a/lib/migration.js b/lib/migration.js
@@ -236,7 +236,8 @@ function mixinMigration(PostgreSQL) {
     var prop = this.getModelDefinition(model).properties[propName];
     var sqlCommand = self.columnEscaped(model, propName)
       + ' ' + self.columnDataType(model, propName) +
-      (self.isNullable(prop) ? '' : ' NOT NULL');
+      (self.isNullable(prop) ? '' : ' NOT NULL') +
+      self.columnDbDefault(model, propName);
     return sqlCommand;
   };
 
@@ -496,7 +497,8 @@ function mixinMigration(PostgreSQL) {
   };
 
   /*!
-   * Get the database-default value for column from given model property
+   * Get the database-default value for column from given model property.
+   * Falls back to LDL's prop.default.
    *
    * @param {String} model The model name
    * @param {String} property The property name
@@ -505,8 +507,14 @@ function mixinMigration(PostgreSQL) {
   PostgreSQL.prototype.columnDbDefault = function(model, property) {
     var columnMetadata = this.columnMetadata(model, property);
     var colDefault = columnMetadata && columnMetadata.dbDefault;
+    if (!colDefault) {
+      var prop = this.getModelDefinition(model).properties[property];
+      if (prop.hasOwnProperty('default')) {
+        colDefault = String(this.escapeValue(prop.default));
+      }
+    }
 
-    return colDefault ? (' DEFAULT ' + columnMetadata.dbDefault) : '';
+    return colDefault ? (' DEFAULT ' + colDefault) : '';
   };
 
   /*!

diff --git a/test/postgresql.migration.test.js b/test/postgresql.migration.test.js
@@ -14,9 +14,7 @@ describe('migrations', function() {
   before(setup);
 
   it('should run migration', function(done) {
-    db.automigrate('UserDataWithIndexes', function() {
-      done();
-    });
+    db.automigrate('UserDataWithIndexes', done);
   });
 
   it('UserDataWithIndexes should have correct indexes', function(done) {
@@ -89,6 +87,7 @@ function setup(done) {
     birthDate: Date,
     pendingPeriod: Number,
     createdByAdmin: Boolean,
+    deleted: {type: Boolean, required: true, default: false},
   }, {
     indexes: {
       udwi_index1: {