2 files changed, 130 insertions, 0 deletions
diff --git a/railties/test/application/rake/migrations_test.rb b/railties/test/application/rake/migrations_test.rb
new file mode 100644
index 0000000000..4ec6f1b67c
--- /dev/null
+++ b/railties/test/application/rake/migrations_test.rb
@@ -0,0 +1,85 @@
+require "isolation/abstract_unit"
+
+module ApplicationTests
+  module RakeTests
+    class RakeMigrationsTest < Test::Unit::TestCase
+      def setup
+        build_app
+        boot_rails
+        FileUtils.rm_rf("#{app_path}/config/environments")
+      end
+
+      def teardown
+        teardown_app
+      end
+      
+      test 'model and migration generator with change syntax' do
+         Dir.chdir(app_path) do
+           `rails generate model user username:string password:string`
+           `rails generate migration add_email_to_users email:string`
+         end
+
+         output = Dir.chdir(app_path){ `rake db:migrate` }
+         assert_match(/create_table\(:users\)/, output)
+         assert_match(/CreateUsers: migrated/, output)
+         assert_match(/add_column\(:users, :email, :string\)/, output)
+         assert_match(/AddEmailToUsers: migrated/, output)
+
+         output = Dir.chdir(app_path){ `rake db:rollback STEP=2` }
+         assert_match(/drop_table\("users"\)/, output)
+         assert_match(/CreateUsers: reverted/, output)
+         assert_match(/remove_column\("users", :email\)/, output)
+         assert_match(/AddEmailToUsers: reverted/, output)
+       end
+
+       test 'migration status when schema migrations table is not present' do
+         output = Dir.chdir(app_path){ `rake db:migrate:status` }
+         assert_equal "Schema migrations table does not exist yet.\n", output
+       end
+
+       test 'test migration status' do
+         Dir.chdir(app_path) do
+           `rails generate model user username:string password:string`
+           `rails generate migration add_email_to_users email:string`
+         end
+
+         Dir.chdir(app_path) { `rake db:migrate`}
+         output = Dir.chdir(app_path) { `rake db:migrate:status` }
+
+         assert_match(/up\s+\d{14}\s+Create users/, output)
+         assert_match(/up\s+\d{14}\s+Add email to users/, output)
+
+         Dir.chdir(app_path) { `rake db:rollback STEP=1` }
+         output = Dir.chdir(app_path) { `rake db:migrate:status` }
+
+         assert_match(/up\s+\d{14}\s+Create users/, output)
+         assert_match(/down\s+\d{14}\s+Add email to users/, output)
+       end
+
+       test 'test migration status after rollback and redo' do
+         Dir.chdir(app_path) do
+           `rails generate model user username:string password:string`
+           `rails generate migration add_email_to_users email:string`
+         end
+
+         Dir.chdir(app_path) { `rake db:migrate`}
+         output = Dir.chdir(app_path) { `rake db:migrate:status` }
+
+         assert_match(/up\s+\d{14}\s+Create users/, output)
+         assert_match(/up\s+\d{14}\s+Add email to users/, output)
+
+         Dir.chdir(app_path) { `rake db:rollback STEP=2` }
+         output = Dir.chdir(app_path) { `rake db:migrate:status` }
+
+         assert_match(/down\s+\d{14}\s+Create users/, output)
+         assert_match(/down\s+\d{14}\s+Add email to users/, output)
+
+         Dir.chdir(app_path) { `rake db:migrate:redo` }
+         output = Dir.chdir(app_path) { `rake db:migrate:status` }
+
+         assert_match(/up\s+\d{14}\s+Create users/, output)
+         assert_match(/up\s+\d{14}\s+Add email to users/, output)
+       end
+    end
+  end
+end
diff --git a/railties/test/application/rake/notes_test.rb b/railties/test/application/rake/notes_test.rb
new file mode 100644
index 0000000000..659cbfec0f
--- /dev/null
+++ b/railties/test/application/rake/notes_test.rb
@@ -0,0 +1,45 @@
+require "isolation/abstract_unit"
+
+module ApplicationTests
+  module RakeTests
+    class RakeNotesTest < Test::Unit::TestCase
+      def setup 
+        build_app
+        require "rails/all"
+      end
+    
+      def teardown
+        teardown_app
+      end
+
+      test 'notes' do
+
+        app_file "app/views/home/index.html.erb", "<% # TODO: note in erb %>"
+        app_file "app/views/home/index.html.haml", "-# TODO: note in haml"
+        app_file "app/views/home/index.html.slim", "/ TODO: note in slim"
+
+        boot_rails
+        require 'rake'
+        require 'rdoc/task'
+        require 'rake/testtask'
+
+        Rails.application.load_tasks
+   
+        Dir.chdir(app_path) do
+          output = `bundle exec rake notes`
+        
+          assert_match /note in erb/, output
+          assert_match /note in haml/, output
+          assert_match /note in slim/, output
+        end
+      
+      end
+    
+      private
+      def boot_rails
+        super
+        require "#{app_path}/config/environment"
+      end
+    end
+  end
+end