thoughtbot--factory_bot/lib/factory_girl/attribute_assigner.rb

71 lines
1.8 KiB
Ruby
Raw Normal View History

module FactoryGirl
class AttributeAssigner
2011-12-05 20:08:47 -05:00
def initialize(build_class, evaluator, attribute_list)
@build_class = build_class
2011-12-05 20:08:47 -05:00
@evaluator = evaluator
@attribute_list = attribute_list
@attribute_names_assigned = []
end
def object
build_class_instance.tap do |instance|
attributes_to_set_on_instance.each do |attribute|
instance.send("#{attribute}=", get(attribute))
@attribute_names_assigned << attribute
end
end
end
def hash
attributes_to_set_on_hash.inject({}) do |result, attribute|
result[attribute] = get(attribute)
result
end
end
private
def build_class_instance
@build_class_instance ||= @build_class.new
end
def get(attribute_name)
2011-12-05 20:08:47 -05:00
@evaluator.send(attribute_name)
end
def attributes_to_set_on_instance
attribute_names_to_assign - @attribute_names_assigned
end
def attributes_to_set_on_hash
attribute_names_to_assign - association_names
end
def attribute_names_to_assign
2011-12-16 12:13:59 -05:00
(non_ignored_attribute_names + override_names - ignored_attribute_names - aliases_to_ignore)
end
def non_ignored_attribute_names
@attribute_list.reject(&:ignored).map(&:name)
end
def ignored_attribute_names
@attribute_list.select(&:ignored).map(&:name)
end
def association_names
@attribute_list.select(&:association?).map(&:name)
end
2011-12-16 12:13:59 -05:00
def override_names
@evaluator.__overrides.keys
end
def aliases_to_ignore
@attribute_list.reject(&:ignored).map do |attribute|
override_names.map {|o| attribute.name if attribute.alias_for?(o) && attribute.name != o }
end.flatten.compact
end
end
end