Merge pull request #287 from iamcharliegoddard/Issue-266

Issue-266: on create added UNIQUE column constraint

Author: brianc

lib/dialect/postgres.js

@@ -785,6 +785,9 @@ Postgres.prototype.visitColumn = function(columnNode) {
       } else if (columnNode.notNull) {
         txt.push(' NOT NULL');
       }
+      if (!columnNode.primaryKey && columnNode.unique) {
+        txt.push(' UNIQUE')
+      }
     }
 
     if (!!columnNode.references) {

lib/node/column.js

@@ -24,6 +24,7 @@ module.exports = Node.define({
     this.subfieldContainer = config.subfieldContainer;
     this.subfields = config.subfields;
     this.autoGenerated = !!config.autoGenerated;
+    this.unique = !!config.unique;
   },
   as: function(alias) {
     this.alias = alias;

test/dialects/create-table-tests.js

@@ -461,3 +461,105 @@ Harness.test({
   params: []
 });
 
+// UNIQUE COLUMN TESTS
+Harness.test({
+  query: Table.define({
+    name: 'post',
+    columns: [{
+      name: 'id',
+      dataType: 'int',
+      //primaryKey: true,
+      //notNull: true,
+      unique: true
+    }]
+  }).create(),
+  pg: {
+    text  : 'CREATE TABLE "post" ("id" int UNIQUE)',
+    string: 'CREATE TABLE "post" ("id" int UNIQUE)'
+  },
+  sqlite: {
+    text  : 'CREATE TABLE "post" ("id" int UNIQUE)',
+    string: 'CREATE TABLE "post" ("id" int UNIQUE)'
+  },
+  mysql: {
+    text  : 'CREATE TABLE `post` (`id` int UNIQUE)',
+    string: 'CREATE TABLE `post` (`id` int UNIQUE)'
+  },
+  mssql: {
+    text  : 'CREATE TABLE [post] ([id] int UNIQUE)',
+    string: 'CREATE TABLE [post] ([id] int UNIQUE)'
+  },
+  oracle: {
+    text  : 'CREATE TABLE "post" ("id" int UNIQUE)',
+    string: 'CREATE TABLE "post" ("id" int UNIQUE)'
+  },
+  params: []
+});
+
+Harness.test({
+  query: Table.define({
+    name: 'post',
+    columns: [{
+      name: 'id',
+      dataType: 'int',
+      //primaryKey: true,
+      notNull: true,
+      unique: true
+    }]
+  }).create(),
+  pg: {
+    text  : 'CREATE TABLE "post" ("id" int NOT NULL UNIQUE)',
+    string: 'CREATE TABLE "post" ("id" int NOT NULL UNIQUE)'
+  },
+  sqlite: {
+    text  : 'CREATE TABLE "post" ("id" int NOT NULL UNIQUE)',
+    string: 'CREATE TABLE "post" ("id" int NOT NULL UNIQUE)'
+  },
+  mysql: {
+    text  : 'CREATE TABLE `post` (`id` int NOT NULL UNIQUE)',
+    string: 'CREATE TABLE `post` (`id` int NOT NULL UNIQUE)'
+  },
+  mssql: {
+    text  : 'CREATE TABLE [post] ([id] int NOT NULL UNIQUE)',
+    string: 'CREATE TABLE [post] ([id] int NOT NULL UNIQUE)'
+  },
+  oracle: {
+    text  : 'CREATE TABLE "post" ("id" int NOT NULL UNIQUE)',
+    string: 'CREATE TABLE "post" ("id" int NOT NULL UNIQUE)'
+  },
+  params: []
+});
+
+Harness.test({
+  query: Table.define({
+    name: 'post',
+    columns: [{
+      name: 'id',
+      dataType: 'int',
+      primaryKey: true,
+      //notNull: true,
+      unique: true
+    }]
+  }).create(),
+  pg: {
+    text  : 'CREATE TABLE "post" ("id" int PRIMARY KEY)',
+    string: 'CREATE TABLE "post" ("id" int PRIMARY KEY)'
+  },
+  sqlite: {
+    text  : 'CREATE TABLE "post" ("id" int PRIMARY KEY)',
+    string: 'CREATE TABLE "post" ("id" int PRIMARY KEY)'
+  },
+  mysql: {
+    text  : 'CREATE TABLE `post` (`id` int PRIMARY KEY)',
+    string: 'CREATE TABLE `post` (`id` int PRIMARY KEY)'
+  },
+  mssql: {
+    text  : 'CREATE TABLE [post] ([id] int PRIMARY KEY)',
+    string: 'CREATE TABLE [post] ([id] int PRIMARY KEY)'
+  },
+  oracle: {
+    text  : 'CREATE TABLE "post" ("id" int PRIMARY KEY)',
+    string: 'CREATE TABLE "post" ("id" int PRIMARY KEY)'
+  },
+  params: []
+});