1 files changed, 106 insertions, 0 deletions
diff --git a/railties/test/generators/resource_generator_test.rb b/railties/test/generators/resource_generator_test.rb
new file mode 100644
index 0000000000..876ddbf84e
--- /dev/null
+++ b/railties/test/generators/resource_generator_test.rb
@@ -0,0 +1,106 @@
+require 'abstract_unit'
+require 'generators/generators_test_helper'
+require 'generators/rails/resource/resource_generator'
+
+class ResourceGeneratorTest < GeneratorsTestCase
+
+  def setup
+    super
+    routes = Rails::Generators::ResourceGenerator.source_root
+    routes = File.join(routes, "..", "..", "app", "templates", "config", "routes.rb")
+    destination = File.join(destination_root, "config")
+
+    FileUtils.mkdir_p(destination)
+    FileUtils.cp File.expand_path(routes), destination
+  end
+
+  def test_help_with_inherited_options
+    content = run_generator ["--help"]
+    assert_match /ActiveRecord options:/, content
+    assert_match /TestUnit options:/, content
+  end
+
+  def test_files_from_inherited_invocation
+    run_generator
+
+    %w(
+      app/models/account.rb
+      test/unit/account_test.rb
+      test/fixtures/accounts.yml
+    ).each { |path| assert_file path }
+
+    assert_migration "db/migrate/create_accounts.rb"
+  end
+
+  def test_inherited_invocations_with_attributes
+    run_generator ["account", "name:string"]
+    assert_migration "db/migrate/create_accounts.rb", /t.string :name/
+  end
+
+  def test_resource_controller_with_pluralized_class_name
+    run_generator
+    assert_file "app/controllers/accounts_controller.rb", /class AccountsController < ApplicationController/
+    assert_file "test/functional/accounts_controller_test.rb", /class AccountsControllerTest < ActionController::TestCase/
+
+    assert_file "app/helpers/accounts_helper.rb", /module AccountsHelper/
+    assert_file "test/unit/helpers/accounts_helper_test.rb", /class AccountsHelperTest < ActionView::TestCase/
+  end
+
+  def test_resource_controller_with_actions
+    run_generator ["account", "--actions", "index", "new"]
+
+    assert_file "app/controllers/accounts_controller.rb" do |controller|
+      assert_instance_method controller, :index
+      assert_instance_method controller, :new
+    end
+
+    assert_file "app/views/accounts/index.html.erb"
+    assert_file "app/views/accounts/new.html.erb"
+  end
+
+  def test_resource_routes_are_added
+    run_generator
+
+    assert_file "config/routes.rb" do |route|
+      assert_match /map\.resources :accounts$/, route
+    end
+  end
+
+  def test_singleton_resource
+    run_generator ["account", "--singleton"]
+
+    assert_file "config/routes.rb" do |route|
+      assert_match /map\.resource :account$/, route
+    end
+  end
+
+  def test_plural_names_are_singularized
+    content = run_generator ["accounts"]
+    assert_file "app/models/account.rb", /class Account < ActiveRecord::Base/
+    assert_file "test/unit/account_test.rb", /class AccountTest/
+    assert_match /Plural version of the model detected, using singularized version. Override with --force-plural./, content
+  end
+
+  def test_plural_names_can_be_forced
+    content = run_generator ["accounts", "--force-plural"]
+    assert_file "app/models/accounts.rb", /class Accounts < ActiveRecord::Base/
+    assert_file "test/unit/accounts_test.rb", /class AccountsTest/
+    assert_no_match /Plural version of the model detected/, content
+  end
+
+  def test_route_is_removed_on_revoke
+    run_generator
+    run_generator ["account"], :behavior => :revoke
+
+    assert_file "config/routes.rb" do |route|
+      assert_no_match /map\.resources :accounts$/, route
+    end
+  end
+
+  protected
+
+    def run_generator(args=["account"], config={})
+      silence(:stdout) { Rails::Generators::ResourceGenerator.start args, config.merge(:destination_root => destination_root) }
+    end
+
+end