view_mapper 0.3.3 → 0.3.4

data/test/views/belongs_to/expected_templates/_form.html.erb

@@ -4,10 +4,10 @@
   <%= f.text_field :name %>
 </p>
 <p>
-  Parent:<br />
-  <%= f.collection_select(:parent_id, Parent.all, :id, :name, { :prompt => true }) %>
+  Parent Model:<br />
+  <%= f.collection_select(:parent_model_id, ParentModel.all, :id, :name, { :prompt => true }) %>
 </p>
 <p>
-  Second Parent:<br />
-  <%= f.collection_select(:second_parent_id, SecondParent.all, :id, :other_field, { :prompt => true }) %>
+  Second Parent Model:<br />
+  <%= f.collection_select(:second_parent_model_id, SecondParentModel.all, :id, :other_field, { :prompt => true }) %>
 </p>

data/test/views/belongs_to/expected_templates/child_model.rb

@@ -0,0 +1,10 @@
+class ChildModel < ActiveRecord::Base
+  belongs_to :parent_model
+  belongs_to :second_parent_model
+  def parent_model_name
+    parent_model.name if parent_model
+  end
+  def second_parent_model_other_field
+    second_parent_model.other_field if second_parent_model
+  end
+end

data/test/views/belongs_to/expected_templates/create_child_models.rb

@@ -0,0 +1,18 @@
+class CreateChildModels < ActiveRecord::Migration
+  def self.up
+    create_table :child_models do |t|
+      t.string :first_name
+      t.string :last_name
+      t.string :address
+      t.boolean :some_flag
+      t.integer :parent_model_id
+      t.integer :second_parent_model_id
+
+      t.timestamps
+    end
+  end
+
+  def self.down
+    drop_table :child_models
+  end
+end

data/test/views/belongs_to/expected_templates/edit.html.erb

@@ -1,11 +1,11 @@
-<h1>Editing some_other_model</h1>
+<h1>Editing child_model</h1>
 
-<% form_for(@some_other_model) do |f| %>
+<% form_for(@child_model) do |f| %>
   <%= render :partial => 'form', :locals => { :f => f } %>
   <p>
     <%= f.submit 'Update' %>
   </p>
 <% end %>
 
-<%= link_to 'Show', @some_other_model %> |
-<%= link_to 'Back', some_other_models_path %>
+<%= link_to 'Show', @child_model %> |
+<%= link_to 'Back', child_models_path %>

data/test/views/belongs_to/expected_templates/index.html.erb

@@ -1,24 +1,24 @@
-<h1>Listing some_other_models</h1>
+<h1>Listing child_models</h1>
 
 <table>
   <tr>
     <th>Name</th>
-    <th>Parent</th>
-    <th>Second Parent</th>
+    <th>Parent Model</th>
+    <th>Second Parent Model</th>
   </tr>
 
-<% @some_other_models.each do |some_other_model| %>
+<% @child_models.each do |child_model| %>
   <tr>
-    <td><%=h some_other_model.name %></td>
-    <td><%=h some_other_model.parent_name %></td>
-    <td><%=h some_other_model.second_parent_other_field %></td>
-    <td><%= link_to 'Show', some_other_model %></td>
-    <td><%= link_to 'Edit', edit_some_other_model_path(some_other_model) %></td>
-    <td><%= link_to 'Destroy', some_other_model, :confirm => 'Are you sure?', :method => :delete %></td>
+    <td><%=h child_model.name %></td>
+    <td><%=h child_model.parent_model_name %></td>
+    <td><%=h child_model.second_parent_model_other_field %></td>
+    <td><%= link_to 'Show', child_model %></td>
+    <td><%= link_to 'Edit', edit_child_model_path(child_model) %></td>
+    <td><%= link_to 'Destroy', child_model, :confirm => 'Are you sure?', :method => :delete %></td>
   </tr>
 <% end %>
 </table>
 
 <br />
 
-<%= link_to 'New some_other_model', new_some_other_model_path %>
+<%= link_to 'New child_model', new_child_model_path %>

data/test/views/belongs_to/expected_templates/new.html.erb

@@ -1,10 +1,10 @@
-<h1>New some_other_model</h1>
+<h1>New child_model</h1>
 
-<% form_for(@some_other_model) do |f| %>
+<% form_for(@child_model) do |f| %>
   <%= render :partial => 'form', :locals => { :f => f } %>
   <p>
     <%= f.submit 'Create' %>
   </p>
 <% end %>
 
-<%= link_to 'Back', some_other_models_path %>
+<%= link_to 'Back', child_models_path %>

data/test/views/belongs_to/expected_templates/show.html.erb

@@ -1,17 +1,17 @@
 <p>
   <b>Name:</b>
-  <%=h @some_other_model.name %>
+  <%=h @child_model.name %>
 </p>
 
 <p>
-  <b>Parent:</b>
-  <%=h @some_other_model.parent_name %>
+  <b>Parent Model:</b>
+  <%=h @child_model.parent_model_name %>
 </p>
 
 <p>
-  <b>Second Parent:</b>
-  <%=h @some_other_model.second_parent_other_field %>
+  <b>Second Parent Model:</b>
+  <%=h @child_model.second_parent_model_other_field %>
 </p>
 
-<%= link_to 'Edit', edit_some_other_model_path(@some_other_model) %> |
-<%= link_to 'Back', some_other_models_path %>
+<%= link_to 'Edit', edit_child_model_path(@child_model) %> |
+<%= link_to 'Back', child_models_path %>

data/test/views/belongs_to_auto_complete/belongs_to_auto_complete_test.rb

@@ -13,460 +13,494 @@ class BelongsToAutoCompleteViewTest < Test::Unit::TestCase
   attr_reader :controller_class_name
   attr_reader :file_name
 
-  context "A view_for generator instantiated for a test model" do
+  context "When the auto_complete plugin is installed" do
     setup do
-      setup_test_model
-      setup_parent_test_model
+      ActionController::Base.send(:include, MockAutoComplete)
     end
 
-    should "detect the existing parents when no parent is specified" do
-      gen = new_generator_for_test_model('view_for', ['--view', 'belongs_to_auto_complete'], 'some_other_model')
-      parent_models = gen.parent_models
-      assert_equal 1, parent_models.size
-      assert_equal 'Parent', parent_models[0].name
-    end
-
-    should "use the specified parent if provided" do
-      gen = new_generator_for_test_model('view_for', ['--view', 'belongs_to_auto_complete:parent'], 'some_other_model')
-      parent_models = gen.parent_models
-      assert_equal 1, parent_models.size
-      assert_equal 'Parent', parent_models[0].name
-    end
-
-    should "return an error message with a bad parent param" do
-      Rails::Generator::Base.logger.expects('error').with('Class \'blah\' does not exist or contains a syntax error and could not be loaded.')
-      new_generator_for_test_model('view_for', ['--view', 'belongs_to_auto_complete:blah'])
-    end
-
-    should "use 'name' as the default parent field" do
-      gen = new_generator_for_test_model('view_for', ['--view', 'belongs_to_auto_complete:parent'], 'some_other_model')
-      assert_equal 'name', gen.field_for(ViewMapper::ModelInfo.new('parent'))
-    end
-
-    should "parse the parent model field" do
-      Rails::Generator::Base.logger.expects('warning').with('Model SomeOtherModel does not have a method parent_first_name.')
-      gen = new_generator_for_test_model('view_for', ['--view', 'belongs_to_auto_complete:parent[first_name]'], 'some_other_model')
-      assert_equal 'first_name', gen.field_for(ViewMapper::ModelInfo.new('parent'))
-    end
-  end
-
-  context "A scaffold_for_view generator instantiated for a test model" do
-    setup do
-      setup_test_model
-      setup_parent_test_model
-    end
-
-    should "return a warning when run with scaffold_for_view when no belongs_to_auto_complete is specified and not run any actions" do
-      expect_no_actions
-      Rails::Generator::Base.logger.expects('error').with('No belongs_to association specified.')
-      @generator_script = Rails::Generator::Scripts::Generate.new
-      @generator_script.run(generator_script_cmd_line('scaffold_for_view', ['--view', 'belongs_to_auto_complete']))
-    end
-  end
-
-  context "A test model with no belongs_to associations" do
-    setup do
-      setup_test_model
-      setup_parent_test_model(true, false)
-    end
-
-    should "return a error when run with view_for and not run any actions" do
-      expect_no_actions
-      Rails::Generator::Base.logger.expects('error').with('No belongs_to associations exist in class Testy.')
-      @generator_script = Rails::Generator::Scripts::Generate.new
-      @generator_script.run(generator_script_cmd_line('view_for', ['--view', 'belongs_to_auto_complete']))
-    end
+    context "A view_for generator instantiated for a test model" do
+      setup do
+        ClassFactory :parent_model
+        ClassFactory :second_parent_model
+        ClassFactory :child_model
+      end
 
-    should "return a error when run with scaffold_for_view and not run any actions" do
-      expect_no_actions
-      Rails::Generator::Base.logger.expects('error').with('No belongs_to association specified.')
-      @generator_script = Rails::Generator::Scripts::Generate.new
-      @generator_script.run(generator_script_cmd_line('scaffold_for_view', ['--view', 'belongs_to_auto_complete']))
-    end
-  end
+      should "detect the existing parents when no parent is specified" do
+        gen = new_generator_for_test_model('view_for', ['--view', 'belongs_to_auto_complete'], 'child_model')
+        parent_models = gen.parent_models
+        assert_equal 2, parent_models.size
+        assert_equal 'ParentModel', parent_models[0].name
+        assert_equal 'SecondParentModel', parent_models[1].name
+      end
 
-  context "A test model with a belongs_to association for a model for which it does not have a name virtual attribute" do
-    setup do
-      setup_test_model
-      setup_parent_test_model
-      setup_second_parent_test_model(true, false)
-    end
+      should "use the specified parent if provided" do
+        gen = new_generator_for_test_model('view_for', ['--view', 'belongs_to_auto_complete:parent_model'], 'child_model')
+        parent_models = gen.parent_models
+        assert_equal 1, parent_models.size
+        assert_equal 'ParentModel', parent_models[0].name
+      end
 
-    should "return a warning and stop when the problem model is specified" do
-      expect_no_actions
-      Rails::Generator::Base.logger.expects('warning').with('Model SomeOtherModel does not have a method second_parent_name.')
-      @generator_script = Rails::Generator::Scripts::Generate.new
-      @generator_script.run(generator_script_cmd_line('view_for', ['--view', 'belongs_to_auto_complete:second_parent'], 'some_other_model'))
-    end
+      should "return an error message with a bad parent param" do
+        Rails::Generator::Base.logger.expects('error').with('Class \'blah\' does not exist or contains a syntax error and could not be loaded.')
+        new_generator_for_test_model('view_for', ['--view', 'belongs_to_auto_complete:blah'])
+      end
 
-    should "return a warning and not include the problem model when run with view_for but continue to run for other models" do
-      stub_actions
-      Rails::Generator::Base.logger.expects('warning').with('Model SomeOtherModel does not have a method second_parent_name.')
-      Rails::Generator::Commands::Create.any_instance.expects(:directory).with('app/controllers/')
-      @generator_script = Rails::Generator::Scripts::Generate.new
-      @generator_script.run(generator_script_cmd_line('view_for', ['--view', 'belongs_to_auto_complete'], 'some_other_model'))
-    end
-  end
+      should "use 'name' as the default parent field" do
+        gen = new_generator_for_test_model('view_for', ['--view', 'belongs_to_auto_complete:parent_model'], 'child_model')
+        assert_equal 'name', gen.field_for(ViewMapper::ModelInfo.new('parent_model'))
+      end
 
-  context "A test model with a belongs_to association for a model for which it does not have a name virtual attribute setter" do
-    setup do
-      setup_test_model
-      setup_parent_test_model
-      setup_second_parent_test_model(false, true)
+      should "parse the parent model field" do
+        Rails::Generator::Base.logger.expects('warning').with('Model ChildModel does not have a method parent_model_first_name.')
+        gen = new_generator_for_test_model('view_for', ['--view', 'belongs_to_auto_complete:parent_model[first_name]'], 'child_model')
+        assert_equal 'first_name', gen.field_for(ViewMapper::ModelInfo.new('parent_model'))
+      end
     end
 
-    should "return a warning and stop when the problem model is specified" do
-      expect_no_actions
-      Rails::Generator::Base.logger.expects('warning').with('Model SomeOtherModel does not have a method second_parent_name=.')
-      @generator_script = Rails::Generator::Scripts::Generate.new
-      @generator_script.run(generator_script_cmd_line('view_for', ['--view', 'belongs_to_auto_complete:second_parent'], 'some_other_model'))
-    end
+    context "A test model with no belongs_to associations" do
+      setup do
+        ClassFactory :test_model
+      end
 
-    should "return a warning and not include the problem model when run with view_for but continue to run for other models" do
-      stub_actions
-      Rails::Generator::Base.logger.expects('warning').with('Model SomeOtherModel does not have a method second_parent_name=.')
-      Rails::Generator::Commands::Create.any_instance.expects(:directory).with('app/controllers/')
-      @generator_script = Rails::Generator::Scripts::Generate.new
-      @generator_script.run(generator_script_cmd_line('view_for', ['--view', 'belongs_to_auto_complete'], 'some_other_model'))
-    end
-  end
+      should "return a error when run with view_for and not run any actions" do
+        expect_no_actions
+        Rails::Generator::Base.logger.expects('error').with('No belongs_to associations exist in class TestModel.')
+        @generator_script = Rails::Generator::Scripts::Generate.new
+        @generator_script.run(generator_script_cmd_line('view_for', ['--view', 'belongs_to_auto_complete']))
+      end
 
-  context "A test model with a belongs_to association for a model that has some other virtual attribute missing" do
-    setup do
-      setup_test_model
-      setup_parent_test_model
+      should "return a error when run with scaffold_for_view and not run any actions" do
+        expect_no_actions
+        Rails::Generator::Base.logger.expects('error').with('No belongs_to association specified.')
+        @generator_script = Rails::Generator::Scripts::Generate.new
+        @generator_script.run(generator_script_cmd_line('scaffold_for_view', ['--view', 'belongs_to_auto_complete']))
+      end
     end
 
-    should "return a warning and stop when the problem model is specified" do
-      expect_no_actions
-      Rails::Generator::Base.logger.expects('warning').with('Model SomeOtherModel does not have a method parent_first_name.')
-      @generator_script = Rails::Generator::Scripts::Generate.new
-      @generator_script.run(generator_script_cmd_line('view_for', ['--view', 'belongs_to_auto_complete:parent[first_name]'], 'some_other_model'))
-    end
-  end
+    context "A test model with a belongs_to association for a model for which it does not have a name virtual attribute" do
+      setup do
+        ClassFactory :parent_model
+        ClassFactory :second_parent_model
 
-  context "A test model with a belongs_to association for a model which does not have a name method or column" do
-    setup do
-      setup_test_model
-      setup_parent_test_model
-      setup_second_parent_test_model(true, true, true, false, false)
-    end
+        child_model_code_missing_virtual_attribute = <<END
+          belongs_to :parent_model
 
-    should "return a warning and stop when the problem model is specified" do
-      expect_no_actions
-      Rails::Generator::Base.logger.expects('warning').with('Model SecondParent does not have a name attribute.')
-      @generator_script = Rails::Generator::Scripts::Generate.new
-      @generator_script.run(generator_script_cmd_line('view_for', ['--view', 'belongs_to_auto_complete:second_parent'], 'some_other_model'))
-    end
+          belongs_to :second_parent_model
+          def second_parent_model_name
+            'something else'
+          end
+          def second_parent_model_name=
+          end
+END
+        ClassFactory :child_model, :class_eval => child_model_code_missing_virtual_attribute
+      end
 
-    should "return a warning and not include the problem model when run with view_for but continue to run for other models" do
-      stub_actions
-      Rails::Generator::Base.logger.expects('warning').with('Model SecondParent does not have a name attribute.')
-      Rails::Generator::Commands::Create.any_instance.expects(:directory).with('app/controllers/')
-      @generator_script = Rails::Generator::Scripts::Generate.new
-      @generator_script.run(generator_script_cmd_line('view_for', ['--view', 'belongs_to_auto_complete'], 'some_other_model'))
-    end
-  end
+      should "return a warning and stop when the problem model is specified" do
+        expect_no_actions
+        Rails::Generator::Base.logger.expects('warning').with('Model ChildModel does not have a method parent_model_name.')
+        @generator_script = Rails::Generator::Scripts::Generate.new
+        @generator_script.run(generator_script_cmd_line('view_for', ['--view', 'belongs_to_auto_complete:parent_model'], 'child_model'))
+      end
 
-  context "A test model with a belongs_to association for a model which does not have some other attribute or column" do
-    setup do
-      setup_test_model
-      setup_parent_test_model
-      ViewMapper::ModelInfo.any_instance.stubs(:has_method?).returns(:true)
+      should "return a warning and not include the problem model when run with view_for but continue to run for other models" do
+        stub_actions
+        Rails::Generator::Base.logger.expects('warning').with('Model ChildModel does not have a method parent_model_name.')
+        Rails::Generator::Commands::Create.any_instance.expects(:directory).with('app/controllers/')
+        @generator_script = Rails::Generator::Scripts::Generate.new
+        @generator_script.run(generator_script_cmd_line('view_for', ['--view', 'belongs_to_auto_complete'], 'child_model'))
+      end
     end
 
-    should "return a warning and stop when the problem model is specified" do
-      expect_no_actions
-      Rails::Generator::Base.logger.expects('warning').with('Model Parent does not have a missing_field column.')
-      @generator_script = Rails::Generator::Scripts::Generate.new
-      @generator_script.run(generator_script_cmd_line('view_for', ['--view', 'belongs_to_auto_complete:parent[missing_field]'], 'some_other_model'))
-    end
-  end
+    context "A test model with a belongs_to association for a model for which it does not have a name virtual attribute setter" do
+      setup do
+        ClassFactory :parent_model
+        ClassFactory :second_parent_model
+        child_model_code_missing_virtual_attribute_setter = <<END
+          belongs_to :parent_model
+          def parent_model_name
+            'something else'
+          end
+          def parent_model_name=
+          end
+
+          belongs_to :second_parent_model
+          def second_parent_model_name
+            'something else'
+          end
+END
+        ClassFactory :child_model, :class_eval => child_model_code_missing_virtual_attribute_setter
 
-  context "A test model with a belongs_to association for a model for which it does not have a foreign key" do
-    setup do
-      setup_test_model
-      setup_parent_test_model
-      setup_second_parent_test_model(true, true, false)
-    end
+      end
 
-    should "return a warning and stop when the problem model is specified" do
-      expect_no_actions
-      Rails::Generator::Base.logger.expects('warning').with('Model SomeOtherModel does not contain a foreign key for SecondParent.')
-      @generator_script = Rails::Generator::Scripts::Generate.new
-      @generator_script.run(generator_script_cmd_line('view_for', ['--view', 'belongs_to_auto_complete:second_parent'], 'some_other_model'))
-    end
+      should "return a warning and stop when the problem model is specified" do
+        expect_no_actions
+        Rails::Generator::Base.logger.expects('warning').with('Model ChildModel does not have a method second_parent_model_name=.')
+        @generator_script = Rails::Generator::Scripts::Generate.new
+        @generator_script.run(generator_script_cmd_line('view_for', ['--view', 'belongs_to_auto_complete:second_parent_model'], 'child_model'))
+      end
 
-    should "return a warning and not include the problem model when run with view_for but continue to run for other models" do
-      stub_actions
-      Rails::Generator::Base.logger.expects('warning').with('Model SomeOtherModel does not contain a foreign key for SecondParent.')
-      Rails::Generator::Commands::Create.any_instance.expects(:directory).with('app/controllers/')
-      @generator_script = Rails::Generator::Scripts::Generate.new
-      @generator_script.run(generator_script_cmd_line('view_for', ['--view', 'belongs_to_auto_complete'], 'some_other_model'))
+      should "return a warning and not include the problem model when run with view_for but continue to run for other models" do
+        stub_actions
+        Rails::Generator::Base.logger.expects('warning').with('Model ChildModel does not have a method second_parent_model_name=.')
+        Rails::Generator::Commands::Create.any_instance.expects(:directory).with('app/controllers/')
+        @generator_script = Rails::Generator::Scripts::Generate.new
+        @generator_script.run(generator_script_cmd_line('view_for', ['--view', 'belongs_to_auto_complete'], 'child_model'))
+      end
     end
-  end
 
-  context "A view_for generator instantiated for a test model with two belongs_to associations" do
-    setup do
-      setup_test_model
-      setup_parent_test_model
-      setup_second_parent_test_model
-      @gen = new_generator_for_test_model('view_for', ['--view', 'belongs_to_auto_complete:parent,second_parent[other_field]'], 'some_other_model')
-    end
+    context "A test model with a belongs_to association for a model that has some other virtual attribute missing" do
+      setup do
+        ClassFactory :parent_model
+        ClassFactory :child_model
+      end
 
-    should "return the proper source root" do
-      assert_equal File.expand_path(File.dirname(__FILE__) + '/../../..//lib/view_mapper/views/belongs_to_auto_complete/templates'), ViewMapper::BelongsToAutoCompleteView.source_root
+      should "return a warning and stop when the problem model is specified" do
+        expect_no_actions
+        Rails::Generator::Base.logger.expects('warning').with('Model ChildModel does not have a method parent_model_first_name.')
+        @generator_script = Rails::Generator::Scripts::Generate.new
+        @generator_script.run(generator_script_cmd_line('view_for', ['--view', 'belongs_to_auto_complete:parent_model[first_name]'], 'child_model'))
+      end
     end
 
-    view_for_templates = %w{ new edit index show }
-    view_for_templates.each do | template |
-      should "render the #{template} template as expected" do
-        @attributes = @gen.attributes
-        @singular_name = @gen.singular_name
-        @plural_name = @gen.plural_name
-        @parent_models = @gen.parent_models
-        template_file = File.open(@gen.source_path("view_#{template}.html.erb"))
-        result = ERB.new(template_file.read, nil, '-').result(binding)
-        expected_file = File.open(File.join(File.dirname(__FILE__), "expected_templates/#{template}.html.erb"))
-        assert_equal expected_file.read, result
+    context "A test model with a belongs_to association for a model which does not have a name method or column" do
+      setup do
+        ClassFactory :parent_model
+        ClassFactory :second_parent_model do |p|
+          p.string :not_the_name_column
+        end
+        ClassFactory :child_model
       end
-    end
 
-    should "render the form template as expected" do
-      @attributes = @gen.attributes
-      @singular_name = @gen.singular_name
-      @plural_name = @gen.plural_name
-      @parent_models = @gen.parent_models
-      template_file = File.open(@gen.source_path("view_form.html.erb"))
-      result = ERB.new(template_file.read, nil, '-').result(binding)
-      expected_file = File.open(File.join(File.dirname(__FILE__), "expected_templates/_form.html.erb"))
-      assert_equal expected_file.read, result
-    end
+      should "return a warning and stop when the problem model is specified" do
+        expect_no_actions
+        Rails::Generator::Base.logger.expects('warning').with('Model SecondParentModel does not have a name attribute.')
+        @generator_script = Rails::Generator::Scripts::Generate.new
+        @generator_script.run(generator_script_cmd_line('view_for', ['--view', 'belongs_to_auto_complete:second_parent_model'], 'child_model'))
+      end
 
-    should "render the layout template as expected" do
-      @controller_class_name = @gen.controller_class_name
-      template_file = File.open(@gen.source_path("layout.html.erb"))
-      result = ERB.new(template_file.read, nil, '-').result(binding)
-      expected_file = File.open(File.join(File.dirname(__FILE__), "expected_templates/some_other_models.html.erb"))
-      assert_equal expected_file.read, result
+      should "return a warning and not include the problem model when run with view_for but continue to run for other models" do
+        stub_actions
+        Rails::Generator::Base.logger.expects('warning').with('Model SecondParentModel does not have a name attribute.')
+        Rails::Generator::Commands::Create.any_instance.expects(:directory).with('app/controllers/')
+        @generator_script = Rails::Generator::Scripts::Generate.new
+        @generator_script.run(generator_script_cmd_line('view_for', ['--view', 'belongs_to_auto_complete'], 'child_model'))
+      end
     end
 
-    should "render the controller template as expected" do
-      @controller_class_name = @gen.controller_class_name
-      @table_name = @gen.table_name
-      @class_name = @gen.class_name
-      @file_name = @gen.file_name
-      @controller_class_name = @gen.controller_class_name
-      @parent_models = @gen.parent_models
-      template_file = File.open(@gen.source_path("controller.rb"))
-      result = ERB.new(template_file.read, nil, '-').result(binding)
-      expected_file = File.open(File.join(File.dirname(__FILE__), "expected_templates/some_other_models_controller.rb"))
-      assert_equal expected_file.read, result
-    end
-  end
+    context "A test model with a belongs_to association for a model which does not have some other attribute or column" do
+      setup do
+        ClassFactory :parent_model
+        ClassFactory :second_parent_model
+        ClassFactory :child_model
+        ViewMapper::ModelInfo.any_instance.stubs(:has_method?).returns(:true)
+      end
 
-  context "A scaffold_for_view generator instantiated for a test model with two belongs_to associations" do
-    setup do
-      setup_test_model
-      setup_parent_test_model
-      setup_second_parent_test_model
-      @gen = new_generator_for_test_model('scaffold_for_view', ['--view', 'belongs_to_auto_complete:parent,second_parent[other_field]'], 'some_other_model')
+      should "return a warning and stop when the problem model is specified" do
+        expect_no_actions
+        Rails::Generator::Base.logger.expects('warning').with('Model ParentModel does not have a missing_field column.')
+        @generator_script = Rails::Generator::Scripts::Generate.new
+        @generator_script.run(generator_script_cmd_line('view_for', ['--view', 'belongs_to_auto_complete:parent_model[missing_field]'], 'child_model'))
+      end
     end
 
-    should "render the model template as expected" do
-      @parent_models = @gen.parent_models
-      @class_name = @gen.class_name
-      @attributes = @gen.attributes
-      template_file = File.open(@gen.source_path("model.erb"))
-      result = ERB.new(template_file.read, nil, '-').result(binding)
-      expected_file = File.open(File.join(File.dirname(__FILE__), "expected_templates/some_other_model.rb"))
-      assert_equal expected_file.read, result
-    end
+    context "A test model with a belongs_to association for a model for which it does not have a foreign key" do
+      setup do
+        ClassFactory :parent_model
+        ClassFactory :second_parent_model
+        ClassFactory :child_model do |child|
+          child.string  :name
+          child.integer :parent_model_id
+        end
+      end
 
-    should "render the migration template as expected" do
-      @class_name = @gen.class_name
-      @attributes = @gen.attributes
-      @migration_name = 'CreateSomeOtherModels'
-      @table_name = @gen.table_name
-      @parent_models = @gen.parent_models
-      @options = {}
-      template_file = File.open(@gen.source_path("migration.rb"))
-      result = ERB.new(template_file.read, nil, '-').result(binding)
-      expected_file = File.open(File.join(File.dirname(__FILE__), "expected_templates/create_some_other_models.rb"))
-      assert_equal expected_file.read, result
-    end
-  end
+      should "return a warning and stop when the problem model is specified" do
+        expect_no_actions
+        Rails::Generator::Base.logger.expects('warning').with('Model ChildModel does not contain a foreign key for SecondParentModel.')
+        @generator_script = Rails::Generator::Scripts::Generate.new
+        @generator_script.run(generator_script_cmd_line('view_for', ['--view', 'belongs_to_auto_complete:second_parent_model'], 'child_model'))
+      end
 
-  context "A Rails generator script" do
-    setup do
-      setup_test_model
-      setup_parent_test_model
-      setup_second_parent_test_model
-      @generator_script = Rails::Generator::Scripts::Generate.new
+      should "return a warning and not include the problem model when run with view_for but continue to run for other models" do
+        stub_actions
+        Rails::Generator::Base.logger.expects('warning').with('Model ChildModel does not contain a foreign key for SecondParentModel.')
+        Rails::Generator::Commands::Create.any_instance.expects(:directory).with('app/controllers/')
+        @generator_script = Rails::Generator::Scripts::Generate.new
+        @generator_script.run(generator_script_cmd_line('view_for', ['--view', 'belongs_to_auto_complete'], 'child_model'))
+      end
     end
 
-    should "add the proper auto_complete route to routes.rb" do
-      Rails::Generator::Commands::Create.any_instance.stubs(:directory)
-      Rails::Generator::Commands::Create.any_instance.stubs(:template)
-      Rails::Generator::Commands::Create.any_instance.stubs(:route_resources)
-      Rails::Generator::Commands::Create.any_instance.stubs(:file)
-      Rails::Generator::Commands::Create.any_instance.stubs(:dependency)
-      Rails::Generator::Base.logger.stubs(:route)
-
-      expected_path = File.dirname(__FILE__) + '/expected_templates'
-      standard_routes_file = expected_path + '/standard_routes.rb'
-      expected_routes_file = expected_path + '/expected_routes.rb'
-      test_routes_file = expected_path + '/routes.rb'
-      ViewForGenerator.any_instance.stubs(:destination_path).returns test_routes_file
-      FileUtils.copy(standard_routes_file, test_routes_file)
-      Rails::Generator::Commands::Create.any_instance.stubs(:route_file).returns(test_routes_file)
-      @generator_script.run(generator_script_cmd_line('view_for', ['--view', 'belongs_to_auto_complete:parent,second_parent[other_field]'], 'some_other_model'))
-      assert_equal File.open(expected_routes_file).read, File.open(test_routes_file).read
-      File.delete(test_routes_file)
-    end
-  end
+    context "When two parent models that both contain a child model exist" do
+      setup do
+        ClassFactory :parent_model
+        ClassFactory :second_parent_model
+        ClassFactory :child_model, :class_eval => <<END
+          belongs_to :parent_model
+          def parent_model_name
+            'something'
+          end
+          def parent_model_name=
+          end
+
+          belongs_to :second_parent_model
+          def second_parent_model_other_field
+            'something else'
+          end
+          def second_parent_model_other_field=
+          end
+END
+      end
 
-  context "A Rails generator script" do
-    setup do
-      setup_test_model
-      setup_parent_test_model
-      setup_second_parent_test_model
-      @generator_script = Rails::Generator::Scripts::Generate.new
-    end
+      context "A view_for generator instantiated for a test model with two belongs_to associations" do
+        setup do
+          @gen = new_generator_for_test_model('view_for', ['--view', 'belongs_to_auto_complete:parent_model,second_parent_model[other_field]'], 'child_model')
+        end
+
+        should "return the proper source root" do
+          assert_equal File.expand_path(File.dirname(__FILE__) + '/../../..//lib/view_mapper/views/belongs_to_auto_complete/templates'), ViewMapper::BelongsToAutoCompleteView.source_root
+        end
+
+        view_for_templates = %w{ new edit index show }
+        view_for_templates.each do | template |
+          should "render the #{template} template as expected" do
+            @attributes = @gen.attributes
+            @singular_name = @gen.singular_name
+            @plural_name = @gen.plural_name
+            @parent_models = @gen.parent_models
+            template_file = File.open(@gen.source_path("view_#{template}.html.erb"))
+            result = ERB.new(template_file.read, nil, '-').result(binding)
+            expected_file = File.open(File.join(File.dirname(__FILE__), "expected_templates/#{template}.html.erb"))
+            assert_equal expected_file.read, result
+          end
+        end
+
+        should "render the form template as expected" do
+          @attributes = @gen.attributes
+          @singular_name = @gen.singular_name
+          @plural_name = @gen.plural_name
+          @parent_models = @gen.parent_models
+          template_file = File.open(@gen.source_path("view_form.html.erb"))
+          result = ERB.new(template_file.read, nil, '-').result(binding)
+          expected_file = File.open(File.join(File.dirname(__FILE__), "expected_templates/_form.html.erb"))
+          assert_equal expected_file.read, result
+        end
+
+        should "render the layout template as expected" do
+          @controller_class_name = @gen.controller_class_name
+          template_file = File.open(@gen.source_path("layout.html.erb"))
+          result = ERB.new(template_file.read, nil, '-').result(binding)
+          expected_file = File.open(File.join(File.dirname(__FILE__), "expected_templates/child_models.html.erb"))
+          assert_equal expected_file.read, result
+        end
+
+        should "render the controller template as expected" do
+          @controller_class_name = @gen.controller_class_name
+          @table_name = @gen.table_name
+          @class_name = @gen.class_name
+          @file_name = @gen.file_name
+          @controller_class_name = @gen.controller_class_name
+          @parent_models = @gen.parent_models
+          template_file = File.open(@gen.source_path("controller.rb"))
+          result = ERB.new(template_file.read, nil, '-').result(binding)
+          expected_file = File.open(File.join(File.dirname(__FILE__), "expected_templates/child_models_controller.rb"))
+          assert_equal expected_file.read, result
+        end
+      end
 
-    should "return a warning when run with view_for on an invalid parent model and not run any actions" do
-      expect_no_actions
-      Rails::Generator::Base.logger.expects('error').with('Class \'blah\' does not exist or contains a syntax error and could not be loaded.')
-      @generator_script = Rails::Generator::Scripts::Generate.new
-      @generator_script.run(generator_script_cmd_line('view_for', ['--view', 'belongs_to_auto_complete:blah'], 'some_other_model'))
-    end
+      context "A scaffold_for_view generator instantiated for a test model with two belongs_to associations" do
+        setup do
+          @gen = new_generator_for_test_model('scaffold_for_view', ['--view', 'belongs_to_auto_complete:parent_model,second_parent_model[other_field]'], 'child_model')
+        end
+
+        should "render the model template as expected" do
+          @parent_models = @gen.parent_models
+          @class_name = @gen.class_name
+          @attributes = @gen.attributes
+          template_file = File.open(@gen.source_path("model.erb"))
+          result = ERB.new(template_file.read, nil, '-').result(binding)
+          expected_file = File.open(File.join(File.dirname(__FILE__), "expected_templates/child_model.rb"))
+          assert_equal expected_file.read, result
+        end
+
+        should "render the migration template as expected" do
+          @class_name = @gen.class_name
+          @attributes = @gen.attributes
+          @migration_name = 'CreateChildModels'
+          @table_name = @gen.table_name
+          @parent_models = @gen.parent_models
+          @options = {}
+          template_file = File.open(@gen.source_path("migration.rb"))
+          result = ERB.new(template_file.read, nil, '-').result(binding)
+          expected_file = File.open(File.join(File.dirname(__FILE__), "expected_templates/create_child_models.rb"))
+          assert_equal expected_file.read, result
+        end
+      end
 
-    should "create the correct manifest when the view_for generator is run with a valid parent model" do
-
-      expect_no_warnings
-
-      directories = [
-        'app/controllers/',
-        'app/helpers/',
-        'app/views/some_other_models',
-        'app/views/layouts/',
-        'test/functional/',
-        'test/unit/',
-        'test/unit/helpers/',
-        'public/stylesheets/'
-      ].each { |path| Rails::Generator::Commands::Create.any_instance.expects(:directory).with(path) }
-
-      templates = {
-        'view_index.html.erb'  => 'app/views/some_other_models/index.html.erb',
-        'view_new.html.erb'    => 'app/views/some_other_models/new.html.erb',
-        'view_edit.html.erb'   => 'app/views/some_other_models/edit.html.erb',
-        'view_show.html.erb'   => 'app/views/some_other_models/show.html.erb',
-        'view_form.html.erb'   => 'app/views/some_other_models/_form.html.erb',
-        'layout.html.erb'      => 'app/views/layouts/some_other_models.html.erb',
-        'style.css'            => 'public/stylesheets/scaffold.css',
-        'controller.rb'        => 'app/controllers/some_other_models_controller.rb',
-        'functional_test.rb'   => 'test/functional/some_other_models_controller_test.rb',
-        'helper.rb'            => 'app/helpers/some_other_models_helper.rb',
-        'helper_test.rb'       => 'test/unit/helpers/some_other_models_helper_test.rb'
-      }.each { |template, target| Rails::Generator::Commands::Create.any_instance.expects(:template).with(template, target) }
-
-      Rails::Generator::Commands::Create.any_instance.expects(:route_resources).with('some_other_models')
-      Rails::Generator::Commands::Create.any_instance.expects(:file).never
-      Rails::Generator::Commands::Create.any_instance.expects(:dependency).never
-
-      Rails::Generator::Commands::Create.any_instance.expects(:route).with(
-        :path => 'auto_complete_for_parent_name',
-        :name => 'connect',
-        :action => 'auto_complete_for_parent_name',
-        :controller => 'some_other_models')
-
-      Rails::Generator::Commands::Create.any_instance.expects(:route).with(
-        :path => 'auto_complete_for_second_parent_other_field',
-        :name => 'connect',
-        :action => 'auto_complete_for_second_parent_other_field',
-        :controller => 'some_other_models')
-
-      @generator_script.run(generator_script_cmd_line('view_for', ['--view', 'belongs_to_auto_complete:parent,second_parent[other_field]'], 'some_other_model'))
-    end
+      context "A Rails generator script" do
+        setup do
+          @generator_script = Rails::Generator::Scripts::Generate.new
+        end
+
+        should "add the proper auto_complete route to routes.rb" do
+          Rails::Generator::Commands::Create.any_instance.stubs(:directory)
+          Rails::Generator::Commands::Create.any_instance.stubs(:template)
+          Rails::Generator::Commands::Create.any_instance.stubs(:route_resources)
+          Rails::Generator::Commands::Create.any_instance.stubs(:file)
+          Rails::Generator::Commands::Create.any_instance.stubs(:dependency)
+          Rails::Generator::Base.logger.stubs(:route)
+
+          expected_path = File.dirname(__FILE__) + '/expected_templates'
+          standard_routes_file = expected_path + '/standard_routes.rb'
+          expected_routes_file = expected_path + '/expected_routes.rb'
+          test_routes_file = expected_path + '/routes.rb'
+          ViewForGenerator.any_instance.stubs(:destination_path).returns test_routes_file
+          FileUtils.copy(standard_routes_file, test_routes_file)
+          Rails::Generator::Commands::Create.any_instance.stubs(:route_file).returns(test_routes_file)
+          @generator_script.run(generator_script_cmd_line('view_for', ['--view', 'belongs_to_auto_complete:parent_model,second_parent_model[other_field]'], 'child_model'))
+          assert_equal File.open(expected_routes_file).read, File.open(test_routes_file).read
+          File.delete(test_routes_file)
+        end
+      end
 
-    should "create the correct manifest when the scaffold_for_view generator is run with a valid parent model" do
-
-      expect_no_warnings
-
-      directories = [
-        'app/models/',
-        'app/controllers/',
-        'app/helpers/',
-        'app/views/some_other_models',
-        'app/views/layouts/',
-        'test/functional/',
-        'test/unit/',
-        'test/unit/helpers/',
-        'test/fixtures/',
-        'public/stylesheets/'
-      ].each { |path| Rails::Generator::Commands::Create.any_instance.expects(:directory).with(path) }
-
-      templates = {
-        'view_index.html.erb'  => 'app/views/some_other_models/index.html.erb',
-        'view_new.html.erb'    => 'app/views/some_other_models/new.html.erb',
-        'view_edit.html.erb'   => 'app/views/some_other_models/edit.html.erb',
-        'view_show.html.erb'   => 'app/views/some_other_models/show.html.erb',
-        'view_form.html.erb'   => 'app/views/some_other_models/_form.html.erb',
-        'layout.html.erb'      => 'app/views/layouts/some_other_models.html.erb',
-        'style.css'            => 'public/stylesheets/scaffold.css',
-        'controller.rb'        => 'app/controllers/some_other_models_controller.rb',
-        'functional_test.rb'   => 'test/functional/some_other_models_controller_test.rb',
-        'helper.rb'            => 'app/helpers/some_other_models_helper.rb',
-        'helper_test.rb'       => 'test/unit/helpers/some_other_models_helper_test.rb',
-        'model.erb'            => 'app/models/some_other_model.rb',
-        'unit_test.rb'         => 'test/unit/some_other_model_test.rb',
-        'fixtures.yml'         => 'test/fixtures/some_other_models.yml'
-      }.each { |template, target| Rails::Generator::Commands::Create.any_instance.expects(:template).with(template, target) }
-
-      Rails::Generator::Commands::Create.any_instance.expects(:route_resources).with('some_other_models')
-      Rails::Generator::Commands::Create.any_instance.expects(:file).never
-      Rails::Generator::Commands::Create.any_instance.expects(:dependency).never
-
-      Rails::Generator::Commands::Create.any_instance.expects(:migration_template).with(
-        'migration.rb',
-        'db/migrate',
-        :assigns => { :migration_name => "CreateSomeOtherModels" },
-        :migration_file_name => "create_some_other_models"
-      )
-
-      Rails::Generator::Commands::Create.any_instance.expects(:route).with(
-        :path => 'auto_complete_for_parent_name',
-        :name => 'connect',
-        :action => 'auto_complete_for_parent_name',
-        :controller => 'some_other_models')
-
-      Rails::Generator::Commands::Create.any_instance.expects(:route).with(
-        :path => 'auto_complete_for_second_parent_other_field',
-        :name => 'connect',
-        :action => 'auto_complete_for_second_parent_other_field',
-        :controller => 'some_other_models')
-
-      @generator_script.run(generator_script_cmd_line('scaffold_for_view', ['--view', 'belongs_to_auto_complete:parent,second_parent[other_field]'], 'some_other_model'))
+      context "A Rails generator script" do
+        setup do
+          @generator_script = Rails::Generator::Scripts::Generate.new
+        end
+
+        should "return a warning when run with view_for on an invalid parent model and not run any actions" do
+          expect_no_actions
+          Rails::Generator::Base.logger.expects('error').with('Class \'blah\' does not exist or contains a syntax error and could not be loaded.')
+          @generator_script = Rails::Generator::Scripts::Generate.new
+          @generator_script.run(generator_script_cmd_line('view_for', ['--view', 'belongs_to_auto_complete:blah'], 'child_model'))
+        end
+
+        should "create the correct manifest when the view_for generator is run with a valid parent model" do
+
+          expect_no_warnings
+
+          directories = [
+            'app/controllers/',
+            'app/helpers/',
+            'app/views/child_models',
+            'app/views/layouts/',
+            'test/functional/',
+            'test/unit/',
+            'test/unit/helpers/',
+            'public/stylesheets/'
+          ].each { |path| Rails::Generator::Commands::Create.any_instance.expects(:directory).with(path) }
+
+          templates = {
+            'view_index.html.erb'  => 'app/views/child_models/index.html.erb',
+            'view_new.html.erb'    => 'app/views/child_models/new.html.erb',
+            'view_edit.html.erb'   => 'app/views/child_models/edit.html.erb',
+            'view_show.html.erb'   => 'app/views/child_models/show.html.erb',
+            'view_form.html.erb'   => 'app/views/child_models/_form.html.erb',
+            'layout.html.erb'      => 'app/views/layouts/child_models.html.erb',
+            'style.css'            => 'public/stylesheets/scaffold.css',
+            'controller.rb'        => 'app/controllers/child_models_controller.rb',
+            'functional_test.rb'   => 'test/functional/child_models_controller_test.rb',
+            'helper.rb'            => 'app/helpers/child_models_helper.rb',
+            'helper_test.rb'       => 'test/unit/helpers/child_models_helper_test.rb'
+          }.each { |template, target| Rails::Generator::Commands::Create.any_instance.expects(:template).with(template, target) }
+
+          Rails::Generator::Commands::Create.any_instance.expects(:route_resources).with('child_models')
+          Rails::Generator::Commands::Create.any_instance.expects(:file).never
+          Rails::Generator::Commands::Create.any_instance.expects(:dependency).never
+
+          Rails::Generator::Commands::Create.any_instance.expects(:route).with(
+            :path => 'auto_complete_for_parent_model_name',
+            :name => 'connect',
+            :action => 'auto_complete_for_parent_model_name',
+            :controller => 'child_models')
+
+          Rails::Generator::Commands::Create.any_instance.expects(:route).with(
+            :path => 'auto_complete_for_second_parent_model_other_field',
+            :name => 'connect',
+            :action => 'auto_complete_for_second_parent_model_other_field',
+            :controller => 'child_models')
+
+          @generator_script.run(generator_script_cmd_line('view_for', ['--view', 'belongs_to_auto_complete:parent_model,second_parent_model[other_field]'], 'child_model'))
+        end
+
+        should "create the correct manifest when the scaffold_for_view generator is run with a valid parent model" do
+
+          expect_no_warnings
+
+          directories = [
+            'app/models/',
+            'app/controllers/',
+            'app/helpers/',
+            'app/views/child_models',
+            'app/views/layouts/',
+            'test/functional/',
+            'test/unit/',
+            'test/unit/helpers/',
+            'test/fixtures/',
+            'public/stylesheets/'
+          ].each { |path| Rails::Generator::Commands::Create.any_instance.expects(:directory).with(path) }
+
+          templates = {
+            'view_index.html.erb'  => 'app/views/child_models/index.html.erb',
+            'view_new.html.erb'    => 'app/views/child_models/new.html.erb',
+            'view_edit.html.erb'   => 'app/views/child_models/edit.html.erb',
+            'view_show.html.erb'   => 'app/views/child_models/show.html.erb',
+            'view_form.html.erb'   => 'app/views/child_models/_form.html.erb',
+            'layout.html.erb'      => 'app/views/layouts/child_models.html.erb',
+            'style.css'            => 'public/stylesheets/scaffold.css',
+            'controller.rb'        => 'app/controllers/child_models_controller.rb',
+            'functional_test.rb'   => 'test/functional/child_models_controller_test.rb',
+            'helper.rb'            => 'app/helpers/child_models_helper.rb',
+            'helper_test.rb'       => 'test/unit/helpers/child_models_helper_test.rb',
+            'model.erb'            => 'app/models/child_model.rb',
+            'unit_test.rb'         => 'test/unit/child_model_test.rb',
+            'fixtures.yml'         => 'test/fixtures/child_models.yml'
+          }.each { |template, target| Rails::Generator::Commands::Create.any_instance.expects(:template).with(template, target) }
+
+          Rails::Generator::Commands::Create.any_instance.expects(:route_resources).with('child_models')
+          Rails::Generator::Commands::Create.any_instance.expects(:file).never
+          Rails::Generator::Commands::Create.any_instance.expects(:dependency).never
+
+          Rails::Generator::Commands::Create.any_instance.expects(:migration_template).with(
+            'migration.rb',
+            'db/migrate',
+            :assigns => { :migration_name => "CreateChildModels" },
+            :migration_file_name => "create_child_models"
+          )
+
+          Rails::Generator::Commands::Create.any_instance.expects(:route).with(
+            :path => 'auto_complete_for_parent_model_name',
+            :name => 'connect',
+            :action => 'auto_complete_for_parent_model_name',
+            :controller => 'child_models')
+
+          Rails::Generator::Commands::Create.any_instance.expects(:route).with(
+            :path => 'auto_complete_for_second_parent_model_other_field',
+            :name => 'connect',
+            :action => 'auto_complete_for_second_parent_model_other_field',
+            :controller => 'child_models')
+
+          @generator_script.run(generator_script_cmd_line('scaffold_for_view', ['--view', 'belongs_to_auto_complete:parent_model,second_parent_model[other_field]'], 'child_model'))
+        end
+      end
     end
-  end
 
-  context "A Rails generator script with a parent model without a has_many association" do
-    setup do
-      setup_test_model
-      setup_parent_test_model(true, true, false)
-      @generator_script = Rails::Generator::Scripts::Generate.new
-    end
+    context "A Rails generator script with a parent model without a has_many association" do
+      setup do
+        ClassFactory :parent_model, :class_eval => ''
+        ClassFactory :child_model
+        @generator_script = Rails::Generator::Scripts::Generate.new
+      end
 
-    should "return a warning when run with view_for and not run any actions" do
-      expect_no_actions
-      Rails::Generator::Base.logger.expects('warning').with('Model Parent does not contain a has_many association for SomeOtherModel.')
-      @generator_script.run(generator_script_cmd_line('view_for', ['--view', 'belongs_to_auto_complete:parent'], 'some_other_model'))
-    end
+      should "return a warning when run with view_for and not run any actions" do
+        expect_no_actions
+        Rails::Generator::Base.logger.expects('warning').with('Model ParentModel does not contain a has_many association for ChildModel.')
+        @generator_script.run(generator_script_cmd_line('view_for', ['--view', 'belongs_to_auto_complete:parent_model'], 'child_model'))
+      end
 
-    should "return a warning when run with scaffold_for_view and not run any actions" do
-      expect_no_actions
-      Rails::Generator::Base.logger.expects('warning').with('Model Parent does not contain a has_many association for SomeOtherModel.')
-      @generator_script.run(generator_script_cmd_line('scaffold_for_view', ['--view', 'belongs_to_auto_complete:parent'], 'some_other_model'))
+      should "return a warning when run with scaffold_for_view and not run any actions" do
+        expect_no_actions
+        Rails::Generator::Base.logger.expects('warning').with('Model ParentModel does not contain a has_many association for ChildModel.')
+        @generator_script.run(generator_script_cmd_line('scaffold_for_view', ['--view', 'belongs_to_auto_complete:parent_model'], 'child_model'))
+      end
     end
   end