diff --git a/test/acceptance/stubbing_method_unnecessarily_test.rb b/test/acceptance/stubbing_method_unnecessarily_test.rb index 6d4615a99..1d5233c9f 100644 --- a/test/acceptance/stubbing_method_unnecessarily_test.rb +++ b/test/acceptance/stubbing_method_unnecessarily_test.rb @@ -37,8 +37,7 @@ def test_should_default_to_allow_stubbing_method_unnecessarily end def test_should_allow_stubbing_method_when_stubbed_method_is_invoked - Mocha.configure { |c| c.stubbing_method_unnecessarily = :prevent } - test_result = run_as_test do + test_result = run_test_with_check(:prevent) do mock = mock('mock') mock.stubs(:public_method) mock.public_method @@ -47,13 +46,17 @@ def test_should_allow_stubbing_method_when_stubbed_method_is_invoked end def stub_method_unnecessarily(treatment = :default) - Mocha.configure { |c| c.stubbing_method_unnecessarily = treatment } unless treatment == :default - run_as_test do + run_test_with_check(treatment) do mock = mock('mock') mock.stubs(:public_method) end end + def run_test_with_check(treatment = :default, &block) + Mocha.configure { |c| c.stubbing_method_unnecessarily = treatment } unless treatment == :default + run_as_test(&block) + end + def violation_message 'stubbing method unnecessarily: #.public_method(any_parameters)' end