29 lines
837 B
Ruby
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 |