@@ -80,6 +80,37 @@ Notice how the keys correspond to the user names. FixtureScenariosBuilder will to an extent, to guess the name of your key. If it can't figure it out, keys will be the standard user_001, user_002, etc format. +Thanks to Paul Cantrell's handywork, custom key names are supported. Simply use the ++name+ method: + + scenario :foo do + name "small_red_widget", Widget.create(:size => 'small', :color => 'red') + name "big_blue_widget", Widget.create(:size => 'big', :color => 'blue') + end + +Another option is to assign your records to instance variables, then call +names_from_ivars+ +at the conclusion of your +scenario+ block. + + scenario :foo do + @small_red_widget = Widget.create(:size => 'small', :color => 'red') + @big_blue_widget = Widget.create(:size => 'big', :color => 'blue') + + names_from_ivars! + end + +The above produces the following YAML: + + small_red_widget: + size: small + color: red + updated_at: 2007-12-27 10:09:05 + created_at: 2007-12-27 10:09:05 + big_blue_widget: + size: big + color: blue + updated_at: 2007-12-27 10:19:23 + created_at: 2007-12-27 10:19:23 + On subsequent test runs this YAML file will not be needlessly re-created. YAML files will only be re-generated when the <tt>scenarios.rb</tt> file is modified. README @@ -33,6 +33,7 @@ class ScenarioBuilder @block = block @children = [] + @custom_names = {} end def validate_parent(scenario) @@ -88,13 +89,30 @@ class ScenarioBuilder def skip_tables %w( schema_info ) end + + def name(custom_name, model_object) + key = [model_object.class.name, model_object.id] + @custom_names[key] = custom_name + model_object + end + + def names_from_ivars! + instance_values.each do |var, value| + name(var, value) if value.is_a? ActiveRecord::Base + end + end + def record_name(record_hash) + key = [@table_name.classify, record_hash['id'].to_i] + @record_names << (@custom_names[key] || inferred_record_name(record_hash) ) + name + end + def inferred_record_name(record_hash) @@record_name_fields.each do |try| if name = record_hash[try] inferred_name = name.underscore.gsub(/\W/, ' ').squeeze(' ').tr(' ', '_') - count = @inferred_names.select { |name| name == inferred_name }.size - @inferred_names << inferred_name + count = @record_names.select { |name| name == inferred_name }.size return count.zero? ? inferred_name : "#{inferred_name}_#{count}" end end @@ -110,9 +128,9 @@ class ScenarioBuilder next files if rows.empty? @row_index = '000' - @inferred_names = [] + @record_names = [] fixture_data = rows.inject({}) do |hash, record| - hash.merge(inferred_record_name(record) => record) + hash.merge(record_name(record) => record) end write_fixture_file fixture_data lib/scenario_builder.rb fc1ef6efa12c1b3cbd38af70a550511ac2dcdb2f Chris Wanstrath chris@ozmm.org http://github.com/defunkt/fixture_scenarios_builder/commit/212be98be4a96b242e40afa5b5139e581a528e99 212be98be4a96b242e40afa5b5139e581a528e99 2007-12-27T14:40:32-08:00 2007-12-27T14:40:32-08:00 fixture_scenarios_builder: custom names [Paul Cantrell] dc45022552572d10db2ccea9a1ac43d806596d71 Chris Wanstrath chris@ozmm.org