From 555a956a66628f0c14440e350fefe62125067058 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jos=C3=A9=20Valim?= <jose.valim@gmail.com>
Date: Wed, 1 Jul 2009 12:07:05 +0200
Subject: Ensure both migration and routes are removed on revoke.

---
 railties/lib/generators/migration.rb               |  8 +++++---
 railties/test/generators/model_generator_test.rb   | 22 ++++++++++++++++++++--
 .../test/generators/resource_generator_test.rb     | 13 +++++++++++--
 3 files changed, 36 insertions(+), 7 deletions(-)

(limited to 'railties')

diff --git a/railties/lib/generators/migration.rb b/railties/lib/generators/migration.rb
index 1981ea6a59..2eb7c0a964 100644
--- a/railties/lib/generators/migration.rb
+++ b/railties/lib/generators/migration.rb
@@ -25,11 +25,13 @@ module Rails
         @migration_file_name  = File.basename(destination).sub(/\.rb$/, '')
         @migration_class_name = @migration_file_name.camelize
 
-        if existing = migration_exists?(migration_dir, @migration_file_name)
-          raise Rails::Generators::Error, "Another migration is already named #{@migration_file_name}: #{existing}"
+        destination = migration_exists?(migration_dir, @migration_file_name)
+
+        if behavior == :invoke
+          raise Error, "Another migration is already named #{@migration_file_name}: #{destination}" if destination
+          destination = File.join(migration_dir, "#{@migration_number}_#{@migration_file_name}.rb")
         end
 
-        destination = File.join(migration_dir, "#{@migration_number}_#{@migration_file_name}.rb")
         template(source, destination, log_status)
       end
 
diff --git a/railties/test/generators/model_generator_test.rb b/railties/test/generators/model_generator_test.rb
index 14cafe7b0d..c0768bba7b 100644
--- a/railties/test/generators/model_generator_test.rb
+++ b/railties/test/generators/model_generator_test.rb
@@ -79,6 +79,24 @@ class ModelGeneratorTest < GeneratorsTestCase
     end
   end
 
+  def test_migration_already_exists_error_message
+    run_generator
+    error = capture(:stderr){ run_generator ["Account"], :behavior => :skip }
+    assert_match /Another migration is already named create_accounts/, error
+  end
+
+  def test_migration_error_is_not_shown_on_revoke
+    run_generator
+    error = capture(:stderr){ run_generator ["Account"], :behavior => :revoke }
+    assert_no_match /Another migration is already named create_accounts/, error
+  end
+
+  def test_migration_is_removed_on_revoke
+    run_generator
+    run_generator ["Account"], :behavior => :revoke
+    assert_no_migration "db/migrate/create_accounts.rb"
+  end
+
   def test_invokes_default_test_framework
     run_generator
     assert_file "test/unit/account_test.rb", /class AccountTest < ActiveSupport::TestCase/
@@ -103,8 +121,8 @@ class ModelGeneratorTest < GeneratorsTestCase
 
   protected
 
-    def run_generator(args=["Account", "name:string", "age:integer"])
-      silence(:stdout) { Rails::Generators::ModelGenerator.start args, :root => destination_root }
+    def run_generator(args=["Account", "name:string", "age:integer"], config={})
+      silence(:stdout) { Rails::Generators::ModelGenerator.start args, config.merge(:root => destination_root) }
     end
 
 end
diff --git a/railties/test/generators/resource_generator_test.rb b/railties/test/generators/resource_generator_test.rb
index 53e9a1e817..a4cbb1894e 100644
--- a/railties/test/generators/resource_generator_test.rb
+++ b/railties/test/generators/resource_generator_test.rb
@@ -100,10 +100,19 @@ class ResourceGeneratorTest < GeneratorsTestCase
     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"])
-      silence(:stdout) { Rails::Generators::ResourceGenerator.start args, :root => destination_root }
+    def run_generator(args=["account"], config={})
+      silence(:stdout) { Rails::Generators::ResourceGenerator.start args, config.merge(:root => destination_root) }
     end
 
 end
-- 
cgit v1.2.3