lib/torque/postgresql/adapter/schema_dumper.rb in torque-postgresql-0.2.16 vs lib/torque/postgresql/adapter/schema_dumper.rb in torque-postgresql-1.0.0

@@ -8,10 +8,19 @@
           def migration_keys
             super + [:subtype]
           end
         end
 
+        # Translate +:enum_set+ into +:enum+
+        def schema_type(column)
+          if column.type == :enum_set
+            :enum
+          else
+            super
+          end
+        end
+
         # Adds +:subtype+ option to the default set
         def prepare_column_options(column)
           spec = super
 
           if subtype = schema_subtype(column)
@@ -22,10 +31,10 @@
         end
 
         private
 
           def schema_subtype(column)
-            column.sql_type.to_sym.inspect if column.type == :enum
+            column.sql_type.to_sym.inspect if column.type == :enum || column.type == :enum_set
           end
 
       end
     end
   end