Add enum type.

coldbox-modules · Sep 22, 2017 · e2f17ab · e2f17ab
1 parent aa13c72
commit e2f17ab
Show file tree

Hide file tree

Showing 4 changed files with 20 additions and 2 deletions.
diff --git a/models/Grammars/Grammar.cfc b/models/Grammars/Grammar.cfc
@@ -803,6 +803,13 @@ component displayname="Grammar" accessors="true" {
         return "DECIMAL(#column.getLength()#,#column.getPrecision()#)";
     }
 
+    function typeEnum( column ) {
+        var values = column.getValues().map( function ( value ) {
+            return wrapValue( value );
+        } ).toList( "," );
+        return "ENUM(#values#)";
+    }
+
     function typeInteger( column ) {
         return "INTEGER(#column.getPrecision()#)";
     }

diff --git a/models/Schema/Blueprint.cfc b/models/Schema/Blueprint.cfc
@@ -97,6 +97,11 @@ component accessors="true" {
         return addColumn( argumentCollection = arguments );
     }
 
+    function enum( name, values ) {
+        arguments.type = "enum";
+        return addColumn( argumentCollection = arguments );
+    }
+
     function increments( name ) {
         arguments.autoIncrement = true;
         addIndex( type = "primary", column = name );

diff --git a/models/Schema/Column.cfc b/models/Schema/Column.cfc
@@ -10,9 +10,11 @@ component accessors="true" {
     property name="unsigned" default="false";
     property name="autoIncrement" default="false";
     property name="default" default="";
+    property name="values";
 
     function init( blueprint ) {
         setBlueprint( blueprint );
+        variables.values = [];
         return this;
     }
 

diff --git a/tests/specs/Schema/SchemaBuilder+GrammarSpec.cfc b/tests/specs/Schema/SchemaBuilder+GrammarSpec.cfc
@@ -146,8 +146,12 @@ component extends="testbox.system.BaseSpec" {
                         expect( blueprint.toSql() ).toBeWithCase( "CREATE TABLE ""employees"" (""salary"" DECIMAL(3,2) NOT NULL)" );
                     } );
 
-                    xit( "enum", function() {
-                        fail( "test not implemented yet" );
+                    it( "enum", function() {
+                        var schema = getBuilder();
+                        var blueprint = schema.create( "employees", function( table ) {
+                            table.enum( "tshirt_size", [ "S", "M", "L", "XL", "XXL" ] );
+                        }, false );
+                        expect( blueprint.toSql() ).toBeWithCase( "CREATE TABLE ""employees"" (""tshirt_size"" ENUM(""S"",""M"",""L"",""XL"",""XXL"") NOT NULL)" );
                     } );
 
                     xit( "float", function() {