diff --git a/test/controllers/custom_registrations_controller_test.rb b/test/controllers/custom_registrations_controller_test.rb index 31a93b87b6..10e81e4826 100644 --- a/test/controllers/custom_registrations_controller_test.rb +++ b/test/controllers/custom_registrations_controller_test.rb @@ -32,4 +32,9 @@ class CustomRegistrationsControllerTest < ActionController::TestCase put :update, { user: { } } assert @controller.update_block_called?, "update failed to yield resource to provided block" end + + test "yield resource to block on new" do + get :new + assert @controller.new_block_called?, "new failed to yield resource to provided block" + end end diff --git a/test/rails_app/app/controllers/custom/registrations_controller.rb b/test/rails_app/app/controllers/custom/registrations_controller.rb index 9f1699c8ef..3864661f2d 100644 --- a/test/rails_app/app/controllers/custom/registrations_controller.rb +++ b/test/rails_app/app/controllers/custom/registrations_controller.rb @@ -1,4 +1,10 @@ class Custom::RegistrationsController < Devise::RegistrationsController + def new + super do |resource| + @new_block_called = true + end + end + def create super do |resource| @create_block_called = true @@ -18,4 +24,8 @@ def create_block_called? def update_block_called? @update_block_called == true end + + def new_block_called? + @new_block_called == true + end end