include Rails::Generators::Migration

vlado · Mar 9, 2012 · e42ceb2 · e42ceb2
1 parent 5659f58
commit e42ceb2
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/lib/generators/rails_temporary_data_generator.rb b/lib/generators/rails_temporary_data_generator.rb
@@ -1,4 +1,8 @@
+require 'rails/generators'
+require 'rails/generators/migration'
+
 class RailsTemporaryDataGenerator < Rails::Generators::Base
+  include Rails::Generators::Migration
 
   def create_migration_file
     migration_template 'migration.rb', 'db/migrate/create_temporary_data.rb'