63 lines
1.2 KiB
Ruby
63 lines
1.2 KiB
Ruby
module FactoryGirl
|
|
class Definition
|
|
attr_reader :callbacks, :defined_traits, :declarations
|
|
|
|
def initialize(name = nil)
|
|
@declarations = DeclarationList.new(name)
|
|
@callbacks = []
|
|
@defined_traits = []
|
|
@to_create = lambda {|instance| instance.save! }
|
|
@traits = []
|
|
end
|
|
|
|
delegate :declare_attribute, :to => :declarations
|
|
|
|
def attributes
|
|
@attributes ||= declarations.attribute_list
|
|
end
|
|
|
|
def compile
|
|
attributes
|
|
end
|
|
|
|
def overridable
|
|
declarations.overridable
|
|
self
|
|
end
|
|
|
|
def traits
|
|
@traits.reverse.map { |name| trait_by_name(name) }
|
|
end
|
|
|
|
def inherit_traits(new_traits)
|
|
@traits += new_traits
|
|
end
|
|
|
|
def add_callback(callback)
|
|
@callbacks << callback
|
|
end
|
|
|
|
def to_create(&block)
|
|
if block_given?
|
|
@to_create = block
|
|
else
|
|
@to_create
|
|
end
|
|
end
|
|
|
|
def define_trait(trait)
|
|
@defined_traits << trait
|
|
end
|
|
|
|
private
|
|
|
|
def trait_by_name(name)
|
|
trait_for(name) || FactoryGirl.trait_by_name(name)
|
|
end
|
|
|
|
def trait_for(name)
|
|
defined_traits.detect {|trait| trait.name == name }
|
|
end
|
|
end
|
|
end
|