added custom-name to generator to avoid uninitialized constant error

aasm · Sep 19, 2016 · 673b6f7 · 673b6f7
1 parent fc8b5ef
commit 673b6f7
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/lib/generators/active_record/aasm_generator.rb b/lib/generators/active_record/aasm_generator.rb
@@ -12,7 +12,7 @@ class AASMGenerator < ActiveRecord::Generators::Base
 
       def copy_aasm_migration
         if model_exists?
-          migration_template "migration_existing.rb", "db/migrate/add_aasm_state_to_#{table_name}.rb"
+          migration_template "migration_existing.rb", "db/migrate/add_#{column_name}_to_#{table_name}.rb"
         else
           migration_template "migration.rb", "db/migrate/aasm_create_#{table_name}.rb"
         end

diff --git a/spec/generators/active_record_generator_spec.rb b/spec/generators/active_record_generator_spec.rb
@@ -36,4 +36,9 @@
     assert_migration "db/migrate/add_aasm_state_to_jobs.rb"
   end
 
+  it "add custom aasm_column in existing model" do
+    run_generator %w(job state)
+    assert_migration "db/migrate/add_state_to_jobs.rb"
+  end
+
 end