Include AR version in generated migrations

shioyama · Mar 8, 2017 · ac3dfbb · ac3dfbb
1 parent d4a9da9
commit ac3dfbb
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/lib/generators/mobility/templates/create_string_translations.rb b/lib/generators/mobility/templates/create_string_translations.rb
@@ -1,4 +1,4 @@
-class CreateStringTranslations < ActiveRecord::Migration
+class CreateStringTranslations < ActiveRecord::Migration[<%= ActiveRecord::Migration.current_version %>]
 
   def change
     create_table :mobility_string_translations do |t|

diff --git a/lib/generators/mobility/templates/create_text_translations.rb b/lib/generators/mobility/templates/create_text_translations.rb
@@ -1,4 +1,4 @@
-class CreateTextTranslations < ActiveRecord::Migration
+class CreateTextTranslations < ActiveRecord::Migration[<%= ActiveRecord::Migration.current_version %>]
 
   def change
     create_table :mobility_text_translations do |t|

diff --git a/spec/generators/install_generator_spec.rb b/spec/generators/install_generator_spec.rb
@@ -21,7 +21,7 @@
         directory "db" do
           directory "migrate" do
             migration "create_text_translations" do
-              contains "class CreateTextTranslations"
+              contains "class CreateTextTranslations < ActiveRecord::Migration[5.0]"
               contains "def change"
               contains "create_table :mobility_text_translations"
               contains "t.text    :value"
@@ -39,7 +39,7 @@
         directory "db" do
           directory "migrate" do
             migration "create_string_translations" do
-              contains "class CreateStringTranslations"
+              contains "class CreateStringTranslations < ActiveRecord::Migration[5.0]"
               contains "def change"
               contains "create_table :mobility_string_translations"
               contains "t.string  :value"