1
0
Fork 0
mirror of https://github.com/thoughtbot/factory_bot.git synced 2022-11-09 11:43:51 -05:00
thoughtbot--factory_bot/lib/factory_girl/step_definitions.rb
2010-11-05 10:27:49 -04:00

60 lines
2.1 KiB
Ruby

module FactoryGirlStepHelpers
def convert_association_string_to_instance(factory_name, assignment)
attribute, value = assignment.split(':', 2)
return if value.blank?
factory = FactoryGirl.factory_by_name(factory_name)
attributes = convert_human_hash_to_attribute_hash({attribute => value.strip}, factory.associations)
attributes_find = {}
attributes.each do |k, v|
k = "#{k}_id" if v.is_a? ActiveRecord::Base
attributes_find[k] = v
end
model_class = factory.build_class
model_class.find(:first, :conditions => attributes_find) or
Factory(factory_name, attributes)
end
def convert_human_hash_to_attribute_hash(human_hash, associations = [])
human_hash.inject({}) do |attribute_hash, (human_key, value)|
key = human_key.downcase.gsub(' ', '_').to_sym
if association = associations.detect {|association| association.name == key }
value = convert_association_string_to_instance(association.factory, value)
end
attribute_hash.merge(key => value)
end
end
end
World(FactoryGirlStepHelpers)
FactoryGirl.factories.values.each do |factory|
Given /^the following (?:#{factory.human_name}|#{factory.human_name.pluralize}) exists?:$/ do |table|
table.hashes.each do |human_hash|
attributes = convert_human_hash_to_attribute_hash(human_hash, factory.associations)
factory.run(FactoryGirl::Proxy::Create, attributes)
end
end
Given /^an? #{factory.human_name} exists$/ do
Factory(factory.name)
end
Given /^(\d+) #{factory.human_name.pluralize} exist$/ do |count|
count.to_i.times { Factory(factory.name) }
end
if factory.build_class.respond_to?(:columns)
factory.build_class.columns.each do |column|
human_column_name = column.name.downcase.gsub('_', ' ')
Given /^an? #{factory.human_name} exists with an? #{human_column_name} of "([^"]*)"$/i do |value|
Factory(factory.name, column.name => value)
end
Given /^(\d+) #{factory.human_name.pluralize} exist with an? #{human_column_name} of "([^"]*)"$/i do |count, value|
count.to_i.times { Factory(factory.name, column.name => value) }
end
end
end
end