Merge pull request #284 from brainspec/ar-type-yaml

Support YAML serialization for the custom AR type

Author: lest

lib/enumerize/activerecord.rb

@@ -90,6 +90,19 @@ def deserialize(value)
       def as_json(options = nil)
         {attr: @attr.name, subtype: @subtype}.as_json(options)
       end
+
+      def encode_with(coder)
+        coder[:class_name] = @attr.klass.name
+        coder[:attr_name] = @attr.name
+        coder[:subtype] = @subtype
+      end
+
+      def init_with(coder)
+        initialize(
+          coder[:class_name].constantize.enumerized_attributes[coder[:attr_name]],
+          coder[:subtype]
+        )
+      end
     end
   end
 end

lib/enumerize/attribute.rb

@@ -1,6 +1,6 @@
 module Enumerize
   class Attribute
-    attr_reader :name, :values, :default_value, :i18n_scope
+    attr_reader :klass, :name, :values, :default_value, :i18n_scope
 
     def initialize(klass, name, options={})
       raise ArgumentError, ':in option is required' unless options[:in]

test/activerecord_test.rb

@@ -517,4 +517,9 @@ class InterestsRequiredUser < User
     type = User.type_for_attribute('status')
     type.as_json['attr'].must_equal 'status'
   end
+
+  it "doesn't break YAML serialization" do
+    user = YAML.load(User.create(status: :blocked).to_yaml)
+    user.status.must_equal 'blocked'
+  end
 end