Fixed bug for setting AdminUser to another class name

gregbell · gregbell · commit 01e8b8160118 · 2011-05-11T07:23:06.000-07:00
diff --git a/lib/generators/active_admin/devise/devise_generator.rb b/lib/generators/active_admin/devise/devise_generator.rb
@@ -35,9 +35,9 @@ def set_namespace_for_path
       end
 
       def add_default_user_to_migration
-        inject_into_file  Dir["db/migrate/*_devise_create_admin_users.rb"].first, 
-                          "# Create a default admin user\n    #{class_name}.create!(:email => 'admin@example.com', :password => 'password', :password_confirmation => 'password')\n\n    ",
-                          :before => "add_index :admin_users, :email"
+        inject_into_file  Dir["db/migrate/*_devise_create_#{table_name}.rb"].first, 
+                          "# Create a default user\n    #{class_name}.create!(:email => 'admin@example.com', :password => 'password', :password_confirmation => 'password')\n\n    ",
+                          :before => "add_index :#{table_name}, :email"
       end
 
     end
