Handle array column limits

lib/annotate/annotate_models.rb

@@ -120,7 +120,11 @@ module AnnotateModels
           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

@@ -91,6 +91,23 @@ EOS
 
 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
 
   it "should get schema info as RDoc" do
     klass = mock_class(:users, :id, [