From fa31229a2338706455dd754fd57e7f7e8ebf26a7 2014-02-09 10:49:57
From: Jean-Philippe Lang <jp_lang@yahoo.fr>
Date: 2014-02-09 10:49:57
Subject: [PATCH] Check that target tables exist before proceeding.

git-svn-id: http://svn.redmine.org/redmine/trunk@12865 e93f8b46-1217-0410-a6f0-8f06a7374b81

---

diff --git a/lib/tasks/redmine.rake b/lib/tasks/redmine.rake
index cd11796..c8d35af 100644
--- a/lib/tasks/redmine.rake
+++ b/lib/tasks/redmine.rake
@@ -60,9 +60,18 @@ WARNING: All data in the development database is deleted.
 DESC
 
   task :migrate_dbms => :environment do
+    ActiveRecord::Base.establish_connection :development
+    target_tables = ActiveRecord::Base.connection.tables
+    ActiveRecord::Base.remove_connection
+
     ActiveRecord::Base.establish_connection :production
+    tables = ActiveRecord::Base.connection.tables.sort - %w(schema_migrations plugin_schema_info)
+
+    if (tables - target_tables).any?
+      abort "The following table(s) are missing from the target database: #{(tables - target_tables).join(', ')}"
+    end
 
-    (ActiveRecord::Base.connection.tables - %w(schema_migrations plugin_schema_info)).each do |table_name|
+    tables.each do |table_name|
       Source = Class.new(ActiveRecord::Base)
       Target = Class.new(ActiveRecord::Base)
       Target.establish_connection(:development)