ditkrg/active_model_serializers
5
0
Fork 0
mirror of https://github.com/ditkrg/active_model_serializers.git synced 2026-01-25 07:16:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

Encapsulate serialized_associations; test inline associations

Browse Source
This commit is contained in:
Benjamin Fleischer
2015-11-30 17:47:12 -06:00
parent 7cbef1b3b5
commit e2903643c5
2 changed files with 42 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
lib/active_model/serializer/associations.rb
View File

@@ -13,7 +13,9 @@ module ActiveModel
DEFAULT_INCLUDE_TREE = ActiveModel::Serializer::IncludeTree.from_string('*')
included do |base|
base.class_attribute :_reflections
base.class_attribute :serialized_associations, instance_writer: false # @api public: maps association name to 'Reflection' instance
base.serialized_associations ||= {}
base.class_attribute :_reflections, instance_writer: false
base._reflections ||= []
extend ActiveSupport::Autoload
@@ -77,13 +79,16 @@ module ActiveModel
def associate(reflection, block)
self._reflections = _reflections.dup
define_method reflection.name do
if block_given?
instance_eval(&block)
else
object.send reflection.name
end
end unless method_defined?(reflection.name)
reflection_name = reflection.name
if block
serialized_associations[reflection_name] = ->(instance) { instance.instance_eval(&block) }
else
serialized_associations[reflection_name] = ->(instance) { instance.object.send(reflection_name) }
end
define_method reflection_name do
serialized_associations[reflection_name].call(self)
end unless method_defined?(reflection_name)
self._reflections << reflection
end