lib/generators/migration/migration_generator.rb in generator-spec-0.3.4 vs lib/generators/migration/migration_generator.rb in generator-spec-0.3.5

@@ -1,13 +1,15 @@
-class MigrationGenerator < Rails::Generators::Base
-  desc "Creates a migration" 
+require 'rspec_for_generators/migration_helper'
 
-  argument :name, :type => :string
+class MigrationGenerator < Rails::Generators::NamedBase 
+  include MigrationHelper
 
+  desc "Creates a migration"
+
   def self.source_root
     @source_root ||= File.expand_path("../templates", __FILE__)
   end
     
   def create_migration
-    template "#{name}.erb", 'db/migrations/20100203_create_users.rb'
+    migration "#{name} name:string age:integer admin:boolean"
   end
 end