activerecord-import/test/support/generate.rb
2010-08-10 21:39:26 -04:00

29 lines
825 B
Ruby

class ActiveSupport::TestCase
def Build(*args)
n = args.shift if args.first.is_a?(Numeric)
factory = args.shift
factory_girl_args = args.shift || {}
if n
Array.new.tap do |collection|
n.times.each { collection << Factory.build(factory.to_s.singularize.to_sym, factory_girl_args) }
end
else
Factory.build(factory.to_s.singularize.to_sym, factory_girl_args)
end
end
def Generate(*args)
n = args.shift if args.first.is_a?(Numeric)
factory = args.shift
factory_girl_args = args.shift || {}
if n
Array.new.tap do |collection|
n.times.each { collection << Factory.create(factory.to_s.singularize.to_sym, factory_girl_args) }
end
else
Factory.create(factory.to_s.singularize.to_sym, factory_girl_args)
end
end
end