activerecord-import/test/support/generate.rb

29 lines
837 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
returning Array.new 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
returning Array.new 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