diff --git a/lib/tasks/user.rake b/lib/tasks/user.rake index 481704db..b323ff63 100644 --- a/lib/tasks/user.rake +++ b/lib/tasks/user.rake @@ -36,13 +36,13 @@ namespace :user do exit end - if u[:role] == "super_admin" - user.remove_role(:user) - user.set_role(:super_admin) - elsif u[:role] == "admin" - user.set_role(:admin) + unless Role.exists?(name: u[:role], provider: u[:provider]) + puts "Invalid Role - Role does not exist" + exit end + user.set_role(u[:role]) + puts "Account succesfully created." puts "Email: #{u[:email]}" puts "Password: #{u[:password]}"