Merge pull request #115 from tableonline/handle_column_limit_with_arrays

Handle array column limits

Author: ctran

lib/annotate/annotate_models.rb

@@ -120,7 +120,11 @@ def get_schema_info(klass, header, options = {})
           col_type << "(#{col.precision}, #{col.scale})"
         else
           if (col.limit)
-            col_type << "(#{col.limit})" unless NO_LIMIT_COL_TYPES.include?(col_type)
+            if col.limit.is_a? Array
+              attrs << "(#{col.limit.join(', ')})" 
+            else
+              col_type << "(#{col.limit})" unless NO_LIMIT_COL_TYPES.include?(col_type)
+            end
           end
         end
 

spec/annotate/annotate_models_spec.rb

@@ -90,6 +90,23 @@ def mock_column(name, type, options={})
 #  name :string(50)       not null
 #
 
+EOS
+  end
+  it "should get schema info with enum type " do
+    klass = mock_class(:users, nil, [
+                                     mock_column(:id, :integer),
+                                     mock_column(:name, :enum, :limit => [:enum1, :enum2])
+                                    ])
+
+    AnnotateModels.get_schema_info(klass, "Schema Info").should eql(<<-EOS)
+# Schema Info
+#
+# Table name: users
+#
+#  id   :integer          not null
+#  name :enum             not null, (enum1, enum2)
+#
+
 EOS
   end