Refs #6133; Fix specs

2 files changed, 10 insertions, 2 deletions

diff --git a/db/migrate/20171106111448_update_import_message_criticity_type_to_string.rb b/db/migrate/20171106111448_update_import_message_criticity_type_to_string.rb
index c14450387..a08a10b9a 100644
--- a/db/migrate/20171106111448_update_import_message_criticity_type_to_string.rb
+++ b/db/migrate/20171106111448_update_import_message_criticity_type_to_string.rb
@@ -7,10 +7,10 @@ class UpdateImportMessageCriticityTypeToString < ActiveRecord::Migration
       when 0 then "info"
       when 1 then "warning"
       when 2 then "error"
-      else 
+      else
         "info"
       end
     end
-    ImportMessage.all.each { |im| im.update_attribute(:criticity, change_criticity_value(im.criticity)) }
+    Import::Message.all.each { |im| im.update_attribute(:criticity, change_criticity_value(im.criticity)) }
   end
 end
diff --git a/db/migrate/20180306152953_update_imports_names.rb b/db/migrate/20180306152953_update_imports_names.rb
new file mode 100644
index 000000000..086d5ce5a
--- /dev/null
+++ b/db/migrate/20180306152953_update_imports_names.rb
@@ -0,0 +1,8 @@
+class UpdateImportsNames < ActiveRecord::Migration
+  def change
+    Import::Base.all.pluck(:type).uniq.each do |type|
+      next if type =~ /^Imports/
+      Import::Base.where(type: type).update_all type: "Import::#{type.gsub 'Import', ''}"
+    end
+  end
+end