add defaultValue parameter when creating tables (brianc#311)

kanthoney · brianc · commit 2808409da82c · 2016-12-20T12:47:57.000-06:00
diff --git a/lib/column.js b/lib/column.js
@@ -20,6 +20,7 @@ var Column = function(config) {
     direction : new TextNode('DESC')
   });
   this.dataType = config.dataType;
+  this.defaultValue = config.defaultValue;
 };
 
 // mix in value expression
diff --git a/lib/dialect/postgres.js b/lib/dialect/postgres.js
@@ -826,6 +826,9 @@ Postgres.prototype.visitColumn = function(columnNode) {
       if (!columnNode.primaryKey && columnNode.unique) {
         txt.push(' UNIQUE');
       }
+      if (columnNode.defaultValue !== undefined) {
+        txt.push(' DEFAULT ' + this._getParameterValue(columnNode.defaultValue));
+      }
     }
 
     if (!!columnNode.references) {
diff --git a/lib/node/column.js b/lib/node/column.js
@@ -20,6 +20,7 @@ module.exports = Node.define({
     this.distinct = config.distinct;
     this.primaryKey = config.primaryKey;
     this.notNull = config.notNull;
+    this.defaultValue = config.defaultValue;
     this.references = config.references;
     // If subfieldContainer is present, this is a subfield and subfieldContainer
     // is the parent Column
diff --git a/test/dialects/create-table-tests.js b/test/dialects/create-table-tests.js
@@ -248,6 +248,39 @@ Harness.test({
   }
 });
 
+Harness.test({
+  query: Table.define({
+    name: 'user',
+    columns: [{
+      name: 'id',
+      dataType: 'int',
+      primaryKey: true,
+      notNull: true
+    }, {
+      name: 'posts',
+      dataType: 'int',
+      notNull: true,
+      defaultValue: 0
+    }]
+  }).create(),
+  pg: {
+    text  : 'CREATE TABLE "user" ("id" int PRIMARY KEY, "posts" int NOT NULL DEFAULT 0)',
+    string: 'CREATE TABLE "user" ("id" int PRIMARY KEY, "posts" int NOT NULL DEFAULT 0)'
+  },
+  sqlite: {
+    text  : 'CREATE TABLE "user" ("id" int PRIMARY KEY, "posts" int NOT NULL DEFAULT 0)',
+    string: 'CREATE TABLE "user" ("id" int PRIMARY KEY, "posts" int NOT NULL DEFAULT 0)'
+  },
+  mysql: {
+    text  : 'CREATE TABLE `user` (`id` int PRIMARY KEY, `posts` int NOT NULL DEFAULT 0)',
+    string: 'CREATE TABLE `user` (`id` int PRIMARY KEY, `posts` int NOT NULL DEFAULT 0)'
+  },
+  oracle: {
+    text  : 'CREATE TABLE "user" ("id" int PRIMARY KEY, "posts" int NOT NULL DEFAULT 0)',
+    string: 'CREATE TABLE "user" ("id" int PRIMARY KEY, "posts" int NOT NULL DEFAULT 0)'
+  }
+});
+
 Harness.test({
   query: Table.define({
     name: 'post',

Original file line number	Diff line number	Diff line change
`@@ -826,6 +826,9 @@ Postgres.prototype.visitColumn = function(columnNode) {`
`826`	`826`	`if (!columnNode.primaryKey && columnNode.unique) {`
`827`	`827`	`txt.push(' UNIQUE');`
`828`	`828`	`}`
	`829`	`+ if (columnNode.defaultValue !== undefined) {`
	`830`	`+ txt.push(' DEFAULT ' + this._getParameterValue(columnNode.defaultValue));`
	`831`	`+ }`
`829`	`832`	`}`
`830`	`833`
`831`	`834`	`if (!!columnNode.references) {`