2012-02-10 21:56:07 +00:00
|
|
|
require "active_support/core_ext/hash/except"
|
2012-01-03 14:31:40 +00:00
|
|
|
require "active_support/core_ext/class/attribute"
|
|
|
|
|
2011-12-02 18:09:35 +00:00
|
|
|
module FactoryGirl
|
|
|
|
class Evaluator
|
2012-04-13 18:20:19 +00:00
|
|
|
class_attribute :attribute_lists
|
2012-01-03 14:31:40 +00:00
|
|
|
|
|
|
|
def self.attribute_list
|
|
|
|
AttributeList.new.tap do |list|
|
|
|
|
attribute_lists.each do |attribute_list|
|
|
|
|
list.apply_attributes attribute_list.to_a
|
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|
2012-01-29 00:00:35 +00:00
|
|
|
|
|
|
|
private_instance_methods.each do |method|
|
|
|
|
undef_method(method) unless method =~ /^__|initialize/
|
|
|
|
end
|
2011-12-19 04:10:03 +00:00
|
|
|
|
2012-01-03 14:31:40 +00:00
|
|
|
def initialize(build_strategy, overrides = {})
|
2011-12-04 06:34:50 +00:00
|
|
|
@build_strategy = build_strategy
|
2011-12-16 19:27:38 +00:00
|
|
|
@overrides = overrides
|
2011-12-02 18:09:35 +00:00
|
|
|
@cached_attributes = overrides
|
2011-12-16 16:37:30 +00:00
|
|
|
|
2012-01-29 00:00:35 +00:00
|
|
|
@overrides.each do |name, value|
|
2012-04-20 20:59:39 +00:00
|
|
|
singleton_class.send :define_method, name, -> { value }
|
2012-01-29 00:00:35 +00:00
|
|
|
end
|
2011-12-02 18:09:35 +00:00
|
|
|
end
|
|
|
|
|
2012-02-08 17:13:27 +00:00
|
|
|
def association(factory_name, overrides = {})
|
2012-04-20 20:36:30 +00:00
|
|
|
strategy_override = overrides.fetch(:strategy) { Strategy::Create }
|
2012-02-17 16:29:11 +00:00
|
|
|
|
2012-04-20 20:36:30 +00:00
|
|
|
build_strategy = StrategyCalculator.new(strategy_override).strategy
|
2012-03-09 20:42:28 +00:00
|
|
|
runner = FactoryRunner.new(factory_name, build_strategy, [overrides.except(:strategy)])
|
2012-02-10 21:56:07 +00:00
|
|
|
@build_strategy.association(runner)
|
2012-02-08 17:13:27 +00:00
|
|
|
end
|
2011-12-02 18:17:57 +00:00
|
|
|
|
2012-01-08 03:13:38 +00:00
|
|
|
def instance=(object_instance)
|
|
|
|
@instance = object_instance
|
|
|
|
end
|
|
|
|
|
2011-12-02 18:09:35 +00:00
|
|
|
def method_missing(method_name, *args, &block)
|
|
|
|
if @cached_attributes.key?(method_name)
|
|
|
|
@cached_attributes[method_name]
|
|
|
|
else
|
2012-04-20 01:34:47 +00:00
|
|
|
if @instance.respond_to?(method_name)
|
|
|
|
@instance.send(method_name, *args, &block)
|
|
|
|
else
|
|
|
|
SyntaxRunner.new.send(method_name, *args, &block)
|
|
|
|
end
|
2011-12-02 18:09:35 +00:00
|
|
|
end
|
|
|
|
end
|
2011-12-06 01:18:34 +00:00
|
|
|
|
2012-04-21 02:33:54 +00:00
|
|
|
def __override_names__
|
|
|
|
@overrides.keys
|
2011-12-06 01:18:34 +00:00
|
|
|
end
|
2011-12-02 18:09:35 +00:00
|
|
|
end
|
|
|
|
end
|