view_mapper 0.3.1 → 0.3.2

data/README.rdoc

@@ -41,6 +41,10 @@ This is the same as view_for, but will also generate a new model at the same tim
 
 {has_many}[http://patshaughnessy.net/2009/11/9/scaffolding-for-complex-forms-using-nested-attributes]: Generates scaffolding for a complex form to edit two or more associated models.
 
+{has_many_auto_complete}[http://patshaughnessy.net/2009/11/25/scaffolding-for-auto-complete-on-a-complex-nested-form]: This is the same as "has_many" but also uses the auto_complete plugin to implement type ahead behavior for each text field. This view requires you to install the {repeated_auto_complete}[http://patshaughnessy.net/repeated_auto_complete] plugin.
+
+{belongs_to}: Generates scaffolding that allows you to select an existing, associated model.
+
 == Examples
 
 For detailed examples and more information see: http://patshaughnessy.net/view_mapper

data/VERSION

@@ -1 +1 @@
-0.3.1
+0.3.2

data/lib/view_mapper/model_info.rb

@@ -19,6 +19,14 @@ module ViewMapper
       @columns ||= active_record_columns.collect { |col| col.name }
     end
 
+    def has_column?(col_name)
+      model.column_names.include? col_name
+    end
+
+    def has_method?(method_name)
+      model.new.methods.include?(method_name)
+    end
+
     def attributes
       @attributes ||= active_record_columns.collect { |col| Rails::Generator::GeneratedAttribute.new col.name, col.type }
     end
@@ -44,7 +52,15 @@ module ViewMapper
     end
 
     def child_models
-      model.reflections.select { |key, value| value.macro == :has_many }.collect do |kvpair|
+      model_info_array_for_association(:has_many)
+    end
+
+    def parent_models
+      model_info_array_for_association(:belongs_to)
+    end
+
+    def model_info_array_for_association(association_type)
+      model.reflections.select { |key, value| value.macro == association_type }.collect do |kvpair|
         kvpair[0].to_s.singularize
       end.sort.collect do |model_name|
         ModelInfo.new model_name
@@ -76,6 +92,10 @@ module ViewMapper
       model.columns.detect { |col| is_foreign_key_for?(col, parent_model_name) }
     end
 
+    def has_virtual_name_method?(parent_model_name)
+      model.new.methods.include? "#{parent_model_name.underscore}_name"
+    end
+
     private
 
     def find_model(model_name)

data/lib/view_mapper/views/belongs_to/belongs_to_view.rb

@@ -0,0 +1,105 @@
+module ViewMapper
+  module BelongsToView
+
+    attr_reader :parent_models
+
+    def self.source_root
+      File.expand_path(File.dirname(__FILE__) + "/templates")
+    end
+
+    def source_roots_for_view
+      [ BelongsToView.source_root, File.expand_path(source_root), File.join(self.class.lookup('model').path, 'templates') ]
+    end
+
+    def manifest
+      m = super.edit do |action|
+        action unless is_model_dependency_action(action) || !valid
+      end
+      if valid
+        m.template(
+          "view_form.html.erb",
+          File.join('app/views', controller_class_path, controller_file_name, "_form.html.erb")
+        )
+        add_model_actions(m) unless view_only?
+      end
+      m
+    end
+
+    def add_model_actions(m)
+      m.directory(File.join('test/fixtures', class_path))
+      m.template   'model.rb',     File.join('app/models', class_path, "#{file_name}.rb")
+      m.template   'unit_test.rb', File.join('test/unit', class_path, "#{file_name}_test.rb")
+      unless options[:skip_fixture]
+        m.template 'fixtures.yml', File.join('test/fixtures', "#{table_name}.yml")
+      end
+      unless options[:skip_migration]
+        m.migration_template 'migration.rb',
+                             'db/migrate',
+                             :assigns => { :migration_name => "Create#{class_name.pluralize.gsub(/::/, '')}" },
+                             :migration_file_name => "create_#{file_path.gsub(/\//, '_').pluralize}"
+      end
+    end
+
+    def is_model_dependency_action(action)
+      action[0] == :dependency && action[1].include?('model')
+    end
+
+    def parent_models
+      @parent_models ||= find_parent_models
+    end
+
+    def find_parent_models
+      if view_param
+        view_param.split(',').collect { |param| ModelInfo.new(param.singularize) }
+      elsif view_only?
+        model.parent_models
+      else
+        []
+      end
+    end
+
+    def validate
+      super
+      @valid &&= validate_parent_models
+    end
+
+    def validate_parent_models
+      parents = parent_models
+      if parents.empty?
+        if view_only?
+          logger.error "No belongs_to associations exist in class #{model.name}."
+        else
+          logger.error "No belongs_to association specified."
+        end
+        return false
+      end
+      parents.reject! { |parent_model| !validate_parent_model(parent_model) }
+      @parent_models = parents
+      !parents.empty?
+    end
+
+    def validate_parent_model(parent_model)
+      if !parent_model.valid?
+        logger.error parent_model.error
+        return false
+      elsif view_only? && !model.belongs_to?(parent_model.name)
+        logger.warning "Model #{model.name} does not belong to model #{parent_model.name}."
+        return false
+      elsif view_only? && !model.has_virtual_name_method?(parent_model.name)
+        logger.warning "Model #{model.name} does not have a method #{parent_model.name.underscore}_name."
+        return false
+      elsif view_only? && !model.has_foreign_key_for?(parent_model.name)
+        logger.warning "Model #{class_name} does not contain a foreign key for #{parent_model.name}."
+        return false
+      elsif !parent_model.has_many?(class_name.pluralize)
+        logger.warning "Model #{parent_model.name} does not contain a has_many association for #{class_name}."
+        return false
+      elsif !parent_model.has_column?('name') && !parent_model.has_method?('name')
+        logger.warning "Model #{parent_model.name} does not have a name attribute."
+        return false
+      end
+      true
+    end
+
+  end
+end

data/lib/view_mapper/views/belongs_to/templates/migration.rb

@@ -0,0 +1,19 @@
+class <%= migration_name %> < ActiveRecord::Migration
+  def self.up
+    create_table :<%= table_name %> do |t|
+<% for attribute in attributes -%>
+      t.<%= attribute.type %> :<%= attribute.name %>
+<% end -%>
+<% for parent_model in parent_models -%>
+      t.integer :<%= parent_model.name.underscore %>_id
+<% end -%>
+<% unless options[:skip_timestamps] %>
+      t.timestamps
+<% end -%>
+    end
+  end
+
+  def self.down
+    drop_table :<%= table_name %>
+  end
+end

data/lib/view_mapper/views/belongs_to/templates/model.rb

@@ -0,0 +1,13 @@
+class <%= class_name %> < ActiveRecord::Base
+<% attributes.select(&:reference?).each do |attribute| -%>
+  belongs_to :<%= attribute.name %>
+<% end -%>
+<% for parent_model in parent_models -%>
+  belongs_to :<%= parent_model.name.underscore %>
+<% end -%>
+<% for parent_model in parent_models -%>
+  def <%= parent_model.name.underscore %>_name
+    <%= parent_model.name.underscore %>.name if <%= parent_model.name.underscore %>
+  end
+<% end -%>
+end

data/lib/view_mapper/views/belongs_to/templates/view_edit.html.erb

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

data/lib/view_mapper/views/belongs_to/templates/view_form.html.erb

@@ -0,0 +1,13 @@
+<%%= f.error_messages %>
+<% for attribute in attributes -%>
+<p>
+  <%%= f.label :<%= attribute.name %> %><br />
+  <%%= f.<%= attribute.field_type %> :<%= attribute.name %> %>
+</p>
+<%- end -%>
+<% for parent_model in parent_models -%>
+<p>
+  <%= parent_model.name.titleize %>:<br />
+  <%%= f.collection_select(:<%= parent_model.name.underscore %>_id, <%= parent_model.name %>.all, :id, :name, { :prompt => true }) %>
+</p>
+<% end -%>

data/lib/view_mapper/views/belongs_to/templates/view_index.html.erb

@@ -0,0 +1,30 @@
+<h1>Listing <%= plural_name %></h1>
+
+<table>
+  <tr>
+<% for attribute in attributes -%>
+    <th><%= attribute.column.human_name %></th>
+<% end -%>
+<% for parent_model in parent_models -%>
+    <th><%= parent_model.name.titleize %></th>
+<% end -%>
+  </tr>
+
+<%% @<%= plural_name %>.each do |<%= singular_name %>| %>
+  <tr>
+<% for attribute in attributes -%>
+    <td><%%=h <%= singular_name %>.<%= attribute.name %> %></td>
+<% end -%>
+<% for parent_model in parent_models -%>
+    <td><%%=h <%= singular_name %>.<%= parent_model.name.underscore %>_name %></td>
+<% end -%>
+    <td><%%= link_to 'Show', <%= singular_name %> %></td>
+    <td><%%= link_to 'Edit', edit_<%= singular_name %>_path(<%= singular_name %>) %></td>
+    <td><%%= link_to 'Destroy', <%= singular_name %>, :confirm => 'Are you sure?', :method => :delete %></td>
+  </tr>
+<%% end %>
+</table>
+
+<br />
+
+<%%= link_to 'New <%= singular_name %>', new_<%= singular_name %>_path %>

data/lib/view_mapper/views/belongs_to/templates/view_new.html.erb

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

data/lib/view_mapper/views/belongs_to/templates/view_show.html.erb

@@ -0,0 +1,16 @@
+<% for attribute in attributes -%>
+<p>
+  <b><%= attribute.column.human_name %>:</b>
+  <%%=h @<%= singular_name %>.<%= attribute.name %> %>
+</p>
+
+<% end -%>
+<% for parent_model in parent_models -%>
+<p>
+  <b><%= parent_model.name.titleize %>:</b>
+  <%%=h @<%= singular_name %>.<%= parent_model.name.underscore %>_name %>
+</p>
+
+<% end -%>
+<%%= link_to 'Edit', edit_<%= singular_name %>_path(@<%= singular_name %>) %> |
+<%%= link_to 'Back', <%= plural_name %>_path %>

data/lib/view_mapper.rb

@@ -6,3 +6,4 @@ require 'view_mapper/view_mapper'
 require 'view_mapper/views/auto_complete/auto_complete_view'
 require 'view_mapper/views/paperclip/paperclip_view'
 require 'view_mapper/views/has_many/has_many_view'
+require 'view_mapper/views/belongs_to/belongs_to_view'

data/test/test_helper.rb

@@ -3,7 +3,7 @@ require 'rubygems'
 require 'test/unit'
 require 'shoulda'
 require 'mocha'
-require 'activerecord'
+require 'active_record'
 
 ActiveRecord::Base.establish_connection({ :adapter => 'sqlite3', :database => ':memory:' })
 
@@ -54,7 +54,7 @@ def setup_test_model(paperclip_columns = false)
   Object.const_get("Testy")
 end
 
-def setup_parent_test_model(create_foreign_key = true, child_belongs_to_parent = true)
+def setup_parent_test_model(create_foreign_key = true, child_belongs_to_parent = true, parent_has_many_children = true)
   ActiveRecord::Base.connection.create_table :parents, :force => true do |table|
     table.column :name, :string
   end
@@ -69,7 +69,7 @@ def setup_parent_test_model(create_foreign_key = true, child_belongs_to_parent =
   Object.const_set("SomeOtherModel", Class.new(ActiveRecord::Base))
   Parent.class_eval do
     has_many :testies
-    has_many :some_other_model
+    has_many :some_other_models unless !parent_has_many_children
     def testies_attributes=
       'fake'
     end
@@ -82,10 +82,38 @@ def setup_parent_test_model(create_foreign_key = true, child_belongs_to_parent =
   end
   SomeOtherModel.class_eval do
     belongs_to :parent
+    def parent_name
+      'something'
+    end
   end
   Object.const_get("Parent")
 end
 
+def setup_second_parent_test_model(has_name_virtual_attribute = true, has_foreign_key = true, parent_has_name_method = false, parent_has_name_column = true)
+  ActiveRecord::Base.connection.create_table :second_parents, :force => true do |table|
+    table.column :name, :string unless !parent_has_name_column
+    table.column :other_field, :string
+  end
+  ActiveRecord::Base.connection.add_column :some_other_models, :second_parent_id, :integer unless !has_foreign_key
+  Object.send(:remove_const, "SecondParent") rescue nil
+  Object.const_set("SecondParent", Class.new(ActiveRecord::Base))
+  SecondParent.class_eval do
+    has_many :some_other_models
+    def some_other_models_attributes=
+      'fake'
+    end
+    def name
+      'fake'
+    end unless !parent_has_name_method
+  end
+  SomeOtherModel.class_eval do
+    belongs_to :second_parent
+    def second_parent_name
+      'something'
+    end unless !has_name_virtual_attribute
+  end
+end
+
 def setup_test_model_without_nested_attributes
   ActiveRecord::Base.connection.create_table :third_models, :force => true do |table|
     table.column :name, :string

data/test/view_mapper_test.rb

@@ -13,17 +13,17 @@ class ViewMapperTest < Test::Unit::TestCase
 
     should "have the proper default value for source_roots_for_view" do
       assert_equal [
-        '/Users/pat/rails-apps/view_mapper/test/views/fake/templates',
-        '/Users/pat/rails-apps/view_mapper/test/generators/fake/templates'
+        File.expand_path(File.dirname(__FILE__) + '/views/fake/templates'),
+        File.expand_path(File.dirname(__FILE__) + '/generators/fake/templates')
       ], @gen.source_roots_for_view
     end
 
     should "use the correct source path for a template file that is overridden in the view" do
-      assert_equal '/Users/pat/rails-apps/view_mapper/test/views/fake/templates/fake_template1.html.erb', @gen.source_path('fake_template1.html.erb')
+      assert_equal File.expand_path(File.dirname(__FILE__) + '/views/fake/templates/fake_template1.html.erb'), @gen.source_path('fake_template1.html.erb')
     end
 
     should "use the correct source path for a template file that is not overridden in the view" do
-      assert_equal '/Users/pat/rails-apps/view_mapper/test/generators/fake/templates/fake_template2.html.erb', @gen.source_path('fake_template2.html.erb')
+      assert_equal File.expand_path(File.dirname(__FILE__) + '/generators/fake/templates/fake_template2.html.erb'), @gen.source_path('fake_template2.html.erb')
     end
   end
 

data/test/views/belongs_to/belongs_to_test.rb

@@ -0,0 +1,348 @@
+require File.dirname(__FILE__) + '/../../test_helper'
+
+class BelongsToViewTest < Test::Unit::TestCase
+
+  attr_reader :singular_name
+  attr_reader :attributes
+  attr_reader :plural_name
+  attr_reader :parent_models
+  attr_reader :class_name
+  attr_reader :migration_name
+  attr_reader :table_name
+  attr_reader :options
+
+  context "A view_for generator instantiated for a test model" do
+    setup do
+      setup_test_model
+      setup_parent_test_model
+    end
+
+    should "detect the existing parents when no parent is specified" do
+      gen = new_generator_for_test_model('view_for', ['--view', 'belongs_to'], '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: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:blah'])
+    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 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']))
+    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']))
+    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']))
+    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
+      setup_test_model
+      setup_parent_test_model
+      setup_second_parent_test_model(false)
+    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:second_parent'], 'some_other_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'], 'some_other_model'))
+    end
+  end
+
+  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, false, false)
+    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 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:second_parent'], 'some_other_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 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'], 'some_other_model'))
+    end
+  end
+
+  context "A test model with a belongs_to association for a model which has a name method but not a name column" do
+    setup do
+      setup_test_model
+      setup_parent_test_model
+      setup_second_parent_test_model(true, true, true, false)
+    end
+
+    should "continue to generate as usual" do
+      stub_actions
+      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:second_parent'], 'some_other_model'))
+    end
+  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
+      setup_test_model
+      setup_parent_test_model
+      setup_second_parent_test_model(true, false)
+    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:second_parent'], 'some_other_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'], 'some_other_model'))
+    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'], 'some_other_model')
+    end
+
+    should "return the proper source root" do
+      assert_equal File.expand_path(File.dirname(__FILE__) + '/../../..//lib/view_mapper/views/belongs_to/templates'), ViewMapper::BelongsToView.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
+  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:parent,second_parent'], 'some_other_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.rb"))
+      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
+
+    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
+
+  context "A Rails generator script" do
+    setup do
+      setup_test_model
+      setup_parent_test_model
+      @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:blah'], 'some_other_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/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
+
+      @generator_script.run(generator_script_cmd_line('view_for', ['--view', 'belongs_to:parent'], 'some_other_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/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.rb'            => '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"
+      )
+
+      @generator_script.run(generator_script_cmd_line('scaffold_for_view', ['--view', 'belongs_to:parent'], 'some_other_model'))
+    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
+
+    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:parent'], 'some_other_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:parent'], 'some_other_model'))
+    end
+  end
+
+end

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

@@ -0,0 +1,13 @@
+<%= f.error_messages %>
+<p>
+  <%= f.label :name %><br />
+  <%= f.text_field :name %>
+</p>
+<p>
+  Parent:<br />
+  <%= f.collection_select(:parent_id, Parent.all, :id, :name, { :prompt => true }) %>
+</p>
+<p>
+  Second Parent:<br />
+  <%= f.collection_select(:second_parent_id, SecondParent.all, :id, :name, { :prompt => true }) %>
+</p>

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

@@ -0,0 +1,18 @@
+class CreateSomeOtherModels < ActiveRecord::Migration
+  def self.up
+    create_table :some_other_models do |t|
+      t.string :first_name
+      t.string :last_name
+      t.string :address
+      t.boolean :some_flag
+      t.integer :parent_id
+      t.integer :second_parent_id
+
+      t.timestamps
+    end
+  end
+
+  def self.down
+    drop_table :some_other_models
+  end
+end

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

@@ -0,0 +1,11 @@
+<h1>Editing some_other_model</h1>
+
+<% form_for(@some_other_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 %>

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

@@ -0,0 +1,24 @@
+<h1>Listing some_other_models</h1>
+
+<table>
+  <tr>
+    <th>Name</th>
+    <th>Parent</th>
+    <th>Second Parent</th>
+  </tr>
+
+<% @some_other_models.each do |some_other_model| %>
+  <tr>
+    <td><%=h some_other_model.name %></td>
+    <td><%=h some_other_model.parent_name %></td>
+    <td><%=h some_other_model.second_parent_name %></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>
+  </tr>
+<% end %>
+</table>
+
+<br />
+
+<%= link_to 'New some_other_model', new_some_other_model_path %>

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

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

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

@@ -0,0 +1,17 @@
+<p>
+  <b>Name:</b>
+  <%=h @some_other_model.name %>
+</p>
+
+<p>
+  <b>Parent:</b>
+  <%=h @some_other_model.parent_name %>
+</p>
+
+<p>
+  <b>Second Parent:</b>
+  <%=h @some_other_model.second_parent_name %>
+</p>
+
+<%= link_to 'Edit', edit_some_other_model_path(@some_other_model) %> |
+<%= link_to 'Back', some_other_models_path %>

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

@@ -0,0 +1,10 @@
+class SomeOtherModel < ActiveRecord::Base
+  belongs_to :parent
+  belongs_to :second_parent
+  def parent_name
+    parent.name if parent
+  end
+  def second_parent_name
+    second_parent.name if second_parent
+  end
+end

data/view_mapper.gemspec

@@ -5,11 +5,11 @@
 
 Gem::Specification.new do |s|
   s.name = %q{view_mapper}
-  s.version = "0.3.1"
+  s.version = "0.3.2"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["Pat Shaughnessy"]
-  s.date = %q{2009-11-25}
+  s.date = %q{2010-01-22}
   s.description = %q{View_mapper will generate scaffolding for new or existing models, customized for the plugins you use in your app.}
   s.email = %q{pat@patshaughnessy.net}
   s.extra_rdoc_files = [
@@ -38,6 +38,14 @@ Gem::Specification.new do |s|
      "lib/view_mapper/views/auto_complete/templates/layout.html.erb",
      "lib/view_mapper/views/auto_complete/templates/view_edit.html.erb",
      "lib/view_mapper/views/auto_complete/templates/view_new.html.erb",
+     "lib/view_mapper/views/belongs_to/belongs_to_view.rb",
+     "lib/view_mapper/views/belongs_to/templates/migration.rb",
+     "lib/view_mapper/views/belongs_to/templates/model.rb",
+     "lib/view_mapper/views/belongs_to/templates/view_edit.html.erb",
+     "lib/view_mapper/views/belongs_to/templates/view_form.html.erb",
+     "lib/view_mapper/views/belongs_to/templates/view_index.html.erb",
+     "lib/view_mapper/views/belongs_to/templates/view_new.html.erb",
+     "lib/view_mapper/views/belongs_to/templates/view_show.html.erb",
      "lib/view_mapper/views/has_many/has_many_view.rb",
      "lib/view_mapper/views/has_many/templates/helper.rb",
      "lib/view_mapper/views/has_many/templates/layout.html.erb",
@@ -75,6 +83,14 @@ Gem::Specification.new do |s|
      "test/views/auto_complete/expected_templates/standard_routes.rb",
      "test/views/auto_complete/expected_templates/testies.html.erb",
      "test/views/auto_complete/expected_templates/testies_controller.rb",
+     "test/views/belongs_to/belongs_to_test.rb",
+     "test/views/belongs_to/expected_templates/_form.html.erb",
+     "test/views/belongs_to/expected_templates/create_some_other_models.rb",
+     "test/views/belongs_to/expected_templates/edit.html.erb",
+     "test/views/belongs_to/expected_templates/index.html.erb",
+     "test/views/belongs_to/expected_templates/new.html.erb",
+     "test/views/belongs_to/expected_templates/show.html.erb",
+     "test/views/belongs_to/expected_templates/some_other_model.rb",
      "test/views/fake/fake_view.rb",
      "test/views/fake/templates/fake_template1.html.erb",
      "test/views/has_many/expected_templates/_form.html.erb",
@@ -113,6 +129,9 @@ Gem::Specification.new do |s|
      "test/views/auto_complete/expected_templates/expected_routes.rb",
      "test/views/auto_complete/expected_templates/standard_routes.rb",
      "test/views/auto_complete/expected_templates/testies_controller.rb",
+     "test/views/belongs_to/belongs_to_test.rb",
+     "test/views/belongs_to/expected_templates/create_some_other_models.rb",
+     "test/views/belongs_to/expected_templates/some_other_model.rb",
      "test/views/fake/fake_view.rb",
      "test/views/has_many/expected_templates/create_parents.rb",
      "test/views/has_many/expected_templates/parent.rb",

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: view_mapper
 version: !ruby/object:Gem::Version 
-  version: 0.3.1
+  version: 0.3.2
 platform: ruby
 authors: 
 - Pat Shaughnessy
@@ -9,7 +9,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2009-11-25 00:00:00 -05:00
+date: 2010-01-22 00:00:00 -05:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency 
@@ -63,6 +63,14 @@ files:
 - lib/view_mapper/views/auto_complete/templates/layout.html.erb
 - lib/view_mapper/views/auto_complete/templates/view_edit.html.erb
 - lib/view_mapper/views/auto_complete/templates/view_new.html.erb
+- lib/view_mapper/views/belongs_to/belongs_to_view.rb
+- lib/view_mapper/views/belongs_to/templates/migration.rb
+- lib/view_mapper/views/belongs_to/templates/model.rb
+- lib/view_mapper/views/belongs_to/templates/view_edit.html.erb
+- lib/view_mapper/views/belongs_to/templates/view_form.html.erb
+- lib/view_mapper/views/belongs_to/templates/view_index.html.erb
+- lib/view_mapper/views/belongs_to/templates/view_new.html.erb
+- lib/view_mapper/views/belongs_to/templates/view_show.html.erb
 - lib/view_mapper/views/has_many/has_many_view.rb
 - lib/view_mapper/views/has_many/templates/helper.rb
 - lib/view_mapper/views/has_many/templates/layout.html.erb
@@ -100,6 +108,14 @@ files:
 - test/views/auto_complete/expected_templates/standard_routes.rb
 - test/views/auto_complete/expected_templates/testies.html.erb
 - test/views/auto_complete/expected_templates/testies_controller.rb
+- test/views/belongs_to/belongs_to_test.rb
+- test/views/belongs_to/expected_templates/_form.html.erb
+- test/views/belongs_to/expected_templates/create_some_other_models.rb
+- test/views/belongs_to/expected_templates/edit.html.erb
+- test/views/belongs_to/expected_templates/index.html.erb
+- test/views/belongs_to/expected_templates/new.html.erb
+- test/views/belongs_to/expected_templates/show.html.erb
+- test/views/belongs_to/expected_templates/some_other_model.rb
 - test/views/fake/fake_view.rb
 - test/views/fake/templates/fake_template1.html.erb
 - test/views/has_many/expected_templates/_form.html.erb
@@ -160,6 +176,9 @@ test_files:
 - test/views/auto_complete/expected_templates/expected_routes.rb
 - test/views/auto_complete/expected_templates/standard_routes.rb
 - test/views/auto_complete/expected_templates/testies_controller.rb
+- test/views/belongs_to/belongs_to_test.rb
+- test/views/belongs_to/expected_templates/create_some_other_models.rb
+- test/views/belongs_to/expected_templates/some_other_model.rb
 - test/views/fake/fake_view.rb
 - test/views/has_many/expected_templates/create_parents.rb
 - test/views/has_many/expected_templates/parent.rb