view_mapper 0.3.3 → 0.3.4

data/test/views/paperclip/expected_templates/{create_testies.rb → create_test_models.rb}

@@ -1,6 +1,6 @@
-class CreateTesties < ActiveRecord::Migration
+class CreateTestModels < ActiveRecord::Migration
   def self.up
-    create_table :testies do |t|
+    create_table :test_models do |t|
       t.string :first_name
       t.string :last_name
       t.string :address
@@ -19,6 +19,6 @@ class CreateTesties < ActiveRecord::Migration
   end
 
   def self.down
-    drop_table :testies
+    drop_table :test_models
   end
 end

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

@@ -1,6 +1,6 @@
-<h1>Editing testy</h1>
+<h1>Editing test_model</h1>
 
-<% form_for(@testy, :html => { :multipart => true }) do |f| %>
+<% form_for(@test_model, :html => { :multipart => true }) do |f| %>
   <%= f.error_messages %>
   <p>
     <%= f.label :first_name %><br />
@@ -31,5 +31,5 @@
   </p>
 <% end %>
 
-<%= link_to 'Show', @testy %> |
-<%= link_to 'Back', testies_path %>
+<%= link_to 'Show', @test_model %> |
+<%= link_to 'Back', test_models_path %>

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

@@ -1,4 +1,4 @@
-<h1>Listing testies</h1>
+<h1>Listing test_models</h1>
 
 <table>
   <tr>
@@ -8,19 +8,19 @@
     <th>Some flag</th>
   </tr>
 
-<% @testies.each do |testy| %>
+<% @test_models.each do |test_model| %>
   <tr>
-    <td><%=h testy.first_name %></td>
-    <td><%=h testy.last_name %></td>
-    <td><%=h testy.address %></td>
-    <td><%=h testy.some_flag %></td>
-    <td><%= link_to 'Show', testy %></td>
-    <td><%= link_to 'Edit', edit_testy_path(testy) %></td>
-    <td><%= link_to 'Destroy', testy, :confirm => 'Are you sure?', :method => :delete %></td>
+    <td><%=h test_model.first_name %></td>
+    <td><%=h test_model.last_name %></td>
+    <td><%=h test_model.address %></td>
+    <td><%=h test_model.some_flag %></td>
+    <td><%= link_to 'Show', test_model %></td>
+    <td><%= link_to 'Edit', edit_test_model_path(test_model) %></td>
+    <td><%= link_to 'Destroy', test_model, :confirm => 'Are you sure?', :method => :delete %></td>
   </tr>
 <% end %>
 </table>
 
 <br />
 
-<%= link_to 'New testy', new_testy_path %>
+<%= link_to 'New test_model', new_test_model_path %>

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

@@ -1,6 +1,6 @@
-<h1>New testy</h1>
+<h1>New test_model</h1>
 
-<% form_for(@testy, :html => { :multipart => true }) do |f| %>
+<% form_for(@test_model, :html => { :multipart => true }) do |f| %>
   <%= f.error_messages %>
   <p>
     <%= f.label :first_name %><br />
@@ -31,4 +31,4 @@
   </p>
 <% end %>
 
-<%= link_to 'Back', testies_path %>
+<%= link_to 'Back', test_models_path %>

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

@@ -1,33 +1,33 @@
 <p>
   <b>First name:</b>
-  <%=h @testy.first_name %>
+  <%=h @test_model.first_name %>
 </p>
 
 <p>
   <b>Last name:</b>
-  <%=h @testy.last_name %>
+  <%=h @test_model.last_name %>
 </p>
 
 <p>
   <b>Address:</b>
-  <%=h @testy.address %>
+  <%=h @test_model.address %>
 </p>
 
 <p>
   <b>Some flag:</b>
-  <%=h @testy.some_flag %>
+  <%=h @test_model.some_flag %>
 </p>
 
 <p>
   <b>Avatar:</b>
-  <%= link_to @testy.avatar_file_name, @testy.avatar.url %><br>
+  <%= link_to @test_model.avatar_file_name, @test_model.avatar.url %><br>
 </p>
 
 <p>
   <b>Avatar2:</b>
-  <%= link_to @testy.avatar2_file_name, @testy.avatar2.url %><br>
+  <%= link_to @test_model.avatar2_file_name, @test_model.avatar2.url %><br>
 </p>
 
 
-<%= link_to 'Edit', edit_testy_path(@testy) %> |
-<%= link_to 'Back', testies_path %>
+<%= link_to 'Edit', edit_test_model_path(@test_model) %> |
+<%= link_to 'Back', test_models_path %>

data/test/views/paperclip/expected_templates/{testy.rb → test_model.rb}

@@ -1,4 +1,4 @@
-class Testy < ActiveRecord::Base
+class TestModel < ActiveRecord::Base
   has_attached_file :avatar
   has_attached_file :avatar2
 end

data/test/views/paperclip/paperclip_view_test.rb

@@ -11,274 +11,302 @@ class PaperclipViewTest < Test::Unit::TestCase
   attr_reader :table_name
   attr_reader :options
 
-  context "A view_for generator instantiated for a test model" do
+  context "When the paperclip plugin is installed and there's a model with paperclip attachments" do
     setup do
-      setup_test_model(true)
+      ActiveRecord::Base.send(:include, MockPaperclip)
+      paperclip_src = <<SRC
+def self.attachment_definitions
+  { :avatar => {:validations => []}, :avatar2 => {:validations => []} }
+end
+SRC
+      ClassFactory :test_model, :class_eval => paperclip_src do |t|
+        t.string  :first_name
+        t.string  :last_name
+        t.string  :address
+        t.boolean :some_flag
+        t.string  :avatar_file_name
+        t.string  :avatar_content_type
+        t.integer :avatar_file_size
+        t.string  :avatar_updated_at
+        t.string  :avatar2_file_name
+        t.string  :avatar2_content_type
+        t.integer :avatar2_file_size
+        t.string  :avatar2_updated_at
+      end
     end
 
-    should "detect the existing attachments when no attachment is specified" do
-      gen = new_generator_for_test_model('view_for', ['--view', 'paperclip'])
-      assert_contains gen.attachments, 'avatar'
-      assert_contains gen.attachments, 'avatar2'
-    end
+    context "A view_for generator instantiated for a test model" do
 
-    should "use the specified attachments if provided" do
-      gen = new_generator_for_test_model('view_for', ['--view', 'paperclip:avatar'])
-      assert_equal [ 'avatar' ], gen.attachments
-    end
+      should "detect the existing attachments when no attachment is specified" do
+        gen = new_generator_for_test_model('view_for', ['--view', 'paperclip'])
+        assert_contains gen.attachments, 'avatar'
+        assert_contains gen.attachments, 'avatar2'
+      end
 
-    should "return an error message with a bad paperclip param" do
-      Rails::Generator::Base.logger.expects('error').with('Attachment \'blah\' does not exist.')
-      new_generator_for_test_model('view_for', ['--view', 'paperclip:blah'])
-    end
-  end
+      should "use the specified attachments if provided" do
+        gen = new_generator_for_test_model('view_for', ['--view', 'paperclip:avatar'])
+        assert_equal [ 'avatar' ], gen.attachments
+      end
 
-  context "A view_for generator instantiated for a test model missing Paperclip columns" do
-    setup do
-      setup_test_model(false)
+      should "return an error message with a bad paperclip param" do
+        Rails::Generator::Base.logger.expects('error').with('Attachment \'blah\' does not exist.')
+        new_generator_for_test_model('view_for', ['--view', 'paperclip:blah'])
+      end
     end
 
-    should "return an error message with a bad paperclip param" do
-      Rails::Generator::Base.logger.expects('error').with('Column \'avatar_file_name\' does not exist. First run script/generate paperclip testy avatar.')
-      new_generator_for_test_model('view_for', ['--view', 'paperclip:avatar'])
-    end
-  end
+    context "A view_for generator instantiated for a test model missing Paperclip columns" do
+      setup do
+        paperclip_src = <<SRC
+def self.attachment_definitions
+  { :avatar => {:validations => []}, :avatar2 => {:validations => []} }
+end
+SRC
+        ClassFactory :test_model, :class_eval => paperclip_src
+      end
 
-  context "A test model with no attachments" do
-    setup do
-      setup_test_model(true)
-      Testy.class_eval do
-        def self.attachment_definitions
-          {}
-        end
+      should "return an error message with a bad paperclip param" do
+        Rails::Generator::Base.logger.expects('error').with('Column \'avatar_file_name\' does not exist. First run script/generate paperclip test_model avatar.')
+        new_generator_for_test_model('view_for', ['--view', 'paperclip:avatar'])
       end
     end
 
-    teardown do
-      Testy.class_eval do
-        def self.attachment_definitions
-          { :avatar => {:validations => []}, :avatar2 => {:validations => []} }
-        end
+    context "A test model with no attachments" do
+      setup do
+        ClassFactory :test_model
       end
-    end
 
-    should "return a warning when run with view_for when no attachment exists and not run any actions" do
-      Rails::Generator::Base.logger.expects('warning').with('No paperclip attachments exist on the specified class.')
-      Rails::Generator::Commands::Create.any_instance.expects(:directory).never
-      Rails::Generator::Commands::Create.any_instance.expects(:template).never
-      Rails::Generator::Commands::Create.any_instance.expects(:route_resources).never
-      Rails::Generator::Commands::Create.any_instance.expects(:file).never
-      Rails::Generator::Commands::Create.any_instance.expects(:route).never
-      Rails::Generator::Commands::Create.any_instance.expects(:dependency).never
-      Rails::Generator::Base.logger.stubs(:error)
-      Rails::Generator::Base.logger.stubs(:route)
-      @generator_script = Rails::Generator::Scripts::Generate.new
-      @generator_script.run(generator_script_cmd_line('view_for', ['--view', 'paperclip']))
-    end
+      should "return a warning when run with view_for when no attachment exists and not run any actions" do
+        Rails::Generator::Base.logger.expects('warning').with('No paperclip attachments exist on the specified class.')
+        Rails::Generator::Commands::Create.any_instance.expects(:directory).never
+        Rails::Generator::Commands::Create.any_instance.expects(:template).never
+        Rails::Generator::Commands::Create.any_instance.expects(:route_resources).never
+        Rails::Generator::Commands::Create.any_instance.expects(:file).never
+        Rails::Generator::Commands::Create.any_instance.expects(:route).never
+        Rails::Generator::Commands::Create.any_instance.expects(:dependency).never
+        Rails::Generator::Base.logger.stubs(:error)
+        Rails::Generator::Base.logger.stubs(:route)
+        @generator_script = Rails::Generator::Scripts::Generate.new
+        @generator_script.run(generator_script_cmd_line('view_for', ['--view', 'paperclip']))
+      end
 
-    should "return a warning when run with scaffold_for_view when no attachment exists and not run any actions" do
-      Rails::Generator::Base.logger.expects('warning').with('No paperclip attachments specified.')
-      Rails::Generator::Commands::Create.any_instance.expects(:directory).never
-      Rails::Generator::Commands::Create.any_instance.expects(:template).never
-      Rails::Generator::Commands::Create.any_instance.expects(:route_resources).never
-      Rails::Generator::Commands::Create.any_instance.expects(:file).never
-      Rails::Generator::Commands::Create.any_instance.expects(:route).never
-      Rails::Generator::Commands::Create.any_instance.expects(:dependency).never
-      Rails::Generator::Base.logger.stubs(:error)
-      Rails::Generator::Base.logger.stubs(:route)
-      @generator_script = Rails::Generator::Scripts::Generate.new
-      @generator_script.run(generator_script_cmd_line('scaffold_for_view', ['--view', 'paperclip']))
+      should "return a warning when run with scaffold_for_view when no attachment exists and not run any actions" do
+        Rails::Generator::Base.logger.expects('warning').with('No paperclip attachments specified.')
+        Rails::Generator::Commands::Create.any_instance.expects(:directory).never
+        Rails::Generator::Commands::Create.any_instance.expects(:template).never
+        Rails::Generator::Commands::Create.any_instance.expects(:route_resources).never
+        Rails::Generator::Commands::Create.any_instance.expects(:file).never
+        Rails::Generator::Commands::Create.any_instance.expects(:route).never
+        Rails::Generator::Commands::Create.any_instance.expects(:dependency).never
+        Rails::Generator::Base.logger.stubs(:error)
+        Rails::Generator::Base.logger.stubs(:route)
+        @generator_script = Rails::Generator::Scripts::Generate.new
+        @generator_script.run(generator_script_cmd_line('scaffold_for_view', ['--view', 'paperclip']))
+      end
     end
-  end
 
-  context "A scaffold_for_view generator with no model" do
-    setup do
-      ActiveRecord::Base.send(:include, MockPaperclip)
-    end
+    context "A scaffold_for_view generator with no model" do
 
-    should "use the specified attachments if provided" do
-      gen = new_generator_for_test_model('scaffold_for_view', ['--view', 'paperclip:avatar3,avatar4'])
-      assert_contains gen.attachments, 'avatar3'
-      assert_contains gen.attachments, 'avatar4'
-    end
+      should "use the specified attachments if provided" do
+        gen = new_generator_for_test_model('scaffold_for_view', ['--view', 'paperclip:avatar3,avatar4'])
+        assert_contains gen.attachments, 'avatar3'
+        assert_contains gen.attachments, 'avatar4'
+      end
 
-    should "not detect the existing attachments when no attachment is specified" do
-      Rails::Generator::Base.logger.expects('warning').with('No paperclip attachments specified.')
-      gen = new_generator_for_test_model('scaffold_for_view', ['--view', 'paperclip'])
-      assert_equal [], gen.attachments
+      should "not detect the existing attachments when no attachment is specified" do
+        Rails::Generator::Base.logger.expects('warning').with('No paperclip attachments specified.')
+        gen = new_generator_for_test_model('scaffold_for_view', ['--view', 'paperclip'])
+        assert_equal [], gen.attachments
+      end
     end
-  end
 
-  context "A view_for gen with no model" do
-    should "not return an error message when paperclip is installed" do
-      Rails::Generator::Base.logger.expects('error').never
-      ActiveRecord::Base.send(:include, MockPaperclip)
-      new_generator_for_test_model('view_for', ['--view', 'paperclip:avatar'])
-    end
+    context "A view_for generator instantiated for a test model with two attachments" do
+      setup do
+        @gen = new_generator_for_test_model('view_for', ['--view', 'paperclip'])
+      end
 
-    should "return an error message when paperclip is not installed" do
-      Rails::Generator::Base.logger.expects('error').with('The Paperclip plugin does not appear to be installed.')
-      ActiveRecord::Base.stubs(:methods).returns([])
-      new_generator_for_test_model('view_for', ['--view', 'paperclip:avatar'])
-    end
-  end
+      should "return the proper source root" do
+        assert_equal File.expand_path(File.dirname(__FILE__) + '/../../..//lib/view_mapper/views/paperclip/templates'), ViewMapper::PaperclipView.source_root
+      end
 
-  context "A view_for generator instantiated for a test model with two attachments" do
-    setup do
-      setup_test_model(true)
-      @gen = new_generator_for_test_model('view_for', ['--view', 'paperclip'])
-    end
+      should "have the proper value for source_roots_for_view" do
+        assert_equal [
+          File.expand_path(File.dirname(__FILE__) + '/../../..//lib/view_mapper/views/paperclip/templates'),
+          @gen.class.lookup('scaffold').path + '/templates',
+          @gen.class.lookup('model').path + '/templates'
+        ], @gen.source_roots_for_view
+      end
 
-    should "return the proper source root" do
-      assert_equal File.expand_path(File.dirname(__FILE__) + '/../../..//lib/view_mapper/views/paperclip/templates'), ViewMapper::PaperclipView.source_root
+      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
+          @attachments = @gen.attachments
+          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
     end
 
-    should "have the proper value for source_roots_for_view" do
-      assert_equal [
-        File.expand_path(File.dirname(__FILE__) + '/../../..//lib/view_mapper/views/paperclip/templates'),
-        @gen.class.lookup('scaffold').path + '/templates',
-        @gen.class.lookup('model').path + '/templates'
-      ], @gen.source_roots_for_view
+    context "A Rails generator script run on a test model" do
+      setup do
+        @generator_script = Rails::Generator::Scripts::Generate.new
+      end
+
+      should "not perform any actions when run on the view_for generator with an invalid paperclip field" do
+        Rails::Generator::Commands::Create.any_instance.expects(:directory).never
+        Rails::Generator::Commands::Create.any_instance.expects(:template).never
+        Rails::Generator::Commands::Create.any_instance.expects(:route_resources).never
+        Rails::Generator::Commands::Create.any_instance.expects(:file).never
+        Rails::Generator::Commands::Create.any_instance.expects(:route).never
+        Rails::Generator::Commands::Create.any_instance.expects(:dependency).never
+        Rails::Generator::Base.logger.stubs(:error)
+        Rails::Generator::Base.logger.stubs(:route)
+        @generator_script.run(generator_script_cmd_line('view_for', ['--view', 'paperclip:blah']))
+      end
+
+      should "create a normal view_for manifest when the view_for generator is run with a valid attachment" do
+
+        directories = [
+          'app/controllers/',
+          'app/helpers/',
+          'app/views/test_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/test_models/index.html.erb',
+          'view_show.html.erb'  => 'app/views/test_models/show.html.erb',
+          'view_new.html.erb'   => 'app/views/test_models/new.html.erb',
+          'view_edit.html.erb'  => 'app/views/test_models/edit.html.erb',
+          'layout.html.erb'     => 'app/views/layouts/test_models.html.erb',
+          'style.css'           => 'public/stylesheets/scaffold.css',
+          'controller.rb'       => 'app/controllers/test_models_controller.rb',
+          'functional_test.rb'  => 'test/functional/test_models_controller_test.rb',
+          'helper.rb'           => 'app/helpers/test_models_helper.rb',
+          'helper_test.rb'      => 'test/unit/helpers/test_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('test_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', 'paperclip:avatar']))
+      end
+
+      should "create a manifest containing model actions when the scaffold_for_view generator is run with a valid attachment" do
+
+        directories = [
+          'app/models/',
+          'app/controllers/',
+          'app/helpers/',
+          'app/views/test_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/test_models/index.html.erb',
+          'view_show.html.erb'  => 'app/views/test_models/show.html.erb',
+          'view_new.html.erb'   => 'app/views/test_models/new.html.erb',
+          'view_edit.html.erb'  => 'app/views/test_models/edit.html.erb',
+          'layout.html.erb'     => 'app/views/layouts/test_models.html.erb',
+          'style.css'           => 'public/stylesheets/scaffold.css',
+          'controller.rb'       => 'app/controllers/test_models_controller.rb',
+          'functional_test.rb'  => 'test/functional/test_models_controller_test.rb',
+          'helper.rb'           => 'app/helpers/test_models_helper.rb',
+          'helper_test.rb'      => 'test/unit/helpers/test_models_helper_test.rb',
+          'model.rb'            => 'app/models/test_model.rb',
+          'unit_test.rb'        => 'test/unit/test_model_test.rb',
+          'fixtures.yml'        => 'test/fixtures/test_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('test_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 => "CreateTestModels" },
+          :migration_file_name => "create_test_models"
+        )
+
+        @generator_script.run(generator_script_cmd_line('scaffold_for_view', ['--view', 'paperclip:avatar']))
+      end
     end
 
-    view_for_templates = %w{ new edit index show }
-    view_for_templates.each do | template |
-      should "render the #{template} template as expected" do
+    context "A scaffold_for_view generator instantiated for a test model with an avatar attachment" do
+      setup do
+        @gen = new_generator_for_test_model('scaffold_for_view', ['--view', 'paperclip:avatar,avatar2'])
+      end
+
+      should "render the model template as expected" do
+        @class_name = @gen.class_name
         @attributes = @gen.attributes
-        @singular_name = @gen.singular_name
-        @plural_name = @gen.plural_name
         @attachments = @gen.attachments
-        template_file = File.open(@gen.source_path("view_#{template}.html.erb"))
+        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/#{template}.html.erb"))
+        expected_file = File.open(File.join(File.dirname(__FILE__), "expected_templates/test_model.rb"))
         assert_equal expected_file.read, result
       end
-    end
-  end
-
-  context "A Rails generator script run on a testy model" do
-    setup do
-      setup_test_model(true)
-      @generator_script = Rails::Generator::Scripts::Generate.new
-    end
 
-    should "not perform any actions when run on the view_for generator with an invalid paperclip field" do
-      Rails::Generator::Commands::Create.any_instance.expects(:directory).never
-      Rails::Generator::Commands::Create.any_instance.expects(:template).never
-      Rails::Generator::Commands::Create.any_instance.expects(:route_resources).never
-      Rails::Generator::Commands::Create.any_instance.expects(:file).never
-      Rails::Generator::Commands::Create.any_instance.expects(:route).never
-      Rails::Generator::Commands::Create.any_instance.expects(:dependency).never
-      Rails::Generator::Base.logger.stubs(:error)
-      Rails::Generator::Base.logger.stubs(:route)
-      @generator_script.run(generator_script_cmd_line('view_for', ['--view', 'paperclip:blah']))
-    end
-
-    should "create a normal view_for manifest when the view_for generator is run with a valid attachment" do
-
-      directories = [
-        'app/controllers/',
-        'app/helpers/',
-        'app/views/testies',
-        '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/testies/index.html.erb',
-        'view_show.html.erb'  => 'app/views/testies/show.html.erb',
-        'view_new.html.erb'   => 'app/views/testies/new.html.erb',
-        'view_edit.html.erb'  => 'app/views/testies/edit.html.erb',
-        'layout.html.erb'     => 'app/views/layouts/testies.html.erb',
-        'style.css'           => 'public/stylesheets/scaffold.css',
-        'controller.rb'       => 'app/controllers/testies_controller.rb',
-        'functional_test.rb'  => 'test/functional/testies_controller_test.rb',
-        'helper.rb'           => 'app/helpers/testies_helper.rb',
-        'helper_test.rb'      => 'test/unit/helpers/testies_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('testies')
-      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', 'paperclip:avatar']))
-    end
-
-    should "create a manifest containing model actions when the scaffold_for_view generator is run with a valid attachment" do
-
-      directories = [
-        'app/models/',
-        'app/controllers/',
-        'app/helpers/',
-        'app/views/testies',
-        '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/testies/index.html.erb',
-        'view_show.html.erb'  => 'app/views/testies/show.html.erb',
-        'view_new.html.erb'   => 'app/views/testies/new.html.erb',
-        'view_edit.html.erb'  => 'app/views/testies/edit.html.erb',
-        'layout.html.erb'     => 'app/views/layouts/testies.html.erb',
-        'style.css'           => 'public/stylesheets/scaffold.css',
-        'controller.rb'       => 'app/controllers/testies_controller.rb',
-        'functional_test.rb'  => 'test/functional/testies_controller_test.rb',
-        'helper.rb'           => 'app/helpers/testies_helper.rb',
-        'helper_test.rb'      => 'test/unit/helpers/testies_helper_test.rb',
-        'model.rb'            => 'app/models/testy.rb',
-        'unit_test.rb'        => 'test/unit/testy_test.rb',
-        'fixtures.yml'        => 'test/fixtures/testies.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('testies')
-      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 => "CreateTesties" },
-        :migration_file_name => "create_testies"
-      )
-
-      @generator_script.run(generator_script_cmd_line('scaffold_for_view', ['--view', 'paperclip:avatar']))
+      should "render the migration template as expected" do
+        @class_name = @gen.class_name
+        @attributes = @gen.attributes
+        @attachments = @gen.attachments
+        @migration_name = 'CreateTestModels'
+        @table_name = @gen.table_name
+        @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_test_models.rb"))
+        assert_equal expected_file.read, result
+      end
     end
   end
 
-  context "A scaffold_for_view generator instantiated for a test model with an avatar attachment" do
+  context "A view_for generator" do
     setup do
-      @gen = new_generator_for_test_model('scaffold_for_view', ['--view', 'paperclip:avatar,avatar2'])
+      paperclip_src = <<SRC
+def self.attachment_definitions
+  { :avatar => {:validations => []}, :avatar2 => {:validations => []} }
+end
+SRC
+      ClassFactory :test_model, :class_eval => paperclip_src do |t|
+        t.string  :avatar_file_name
+        t.string  :avatar_content_type
+        t.integer :avatar_file_size
+        t.string  :avatar_updated_at
+        t.string  :avatar2_file_name
+        t.string  :avatar2_content_type
+        t.integer :avatar2_file_size
+        t.string  :avatar2_updated_at
+      end
     end
 
-    should "render the model template as expected" do
-      @class_name = @gen.class_name
-      @attributes = @gen.attributes
-      @attachments = @gen.attachments
-      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/testy.rb"))
-      assert_equal expected_file.read, result
+    should "not return an error message when paperclip is installed" do
+      Rails::Generator::Base.logger.expects('error').never
+      ActiveRecord::Base.send(:include, MockPaperclip)
+      new_generator_for_test_model('view_for', ['--view', 'paperclip:avatar'])
     end
 
-    should "render the migration template as expected" do
-      @class_name = @gen.class_name
-      @attributes = @gen.attributes
-      @attachments = @gen.attachments
-      @migration_name = 'CreateTesties'
-      @table_name = @gen.table_name
-      @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_testies.rb"))
-      assert_equal expected_file.read, result
+    should "return an error message when paperclip is not installed" do
+      Rails::Generator::Base.logger.expects('error').with('The Paperclip plugin does not appear to be installed.')
+      ActiveRecord::Base.stubs(:methods).returns([])
+      new_generator_for_test_model('view_for', ['--view', 'paperclip:avatar'])
     end
   end
+
 end