@@ -4,20 +4,6 @@ class MockableClass def self.find id return :original_return end - - def public_method - private_method - protected_method - end - -protected - - def protected_method; end - -private - - def private_method; end - end describe "A partial mock" do @@ -40,25 +26,4 @@ describe "A partial mock" do MockableClass.msg_3 end - it 'should make public proxies for public methods' do - object = MockableClass.new - object.should_receive(:public_method) - object.public_methods.include?('public_method').should == true - object.public_method - end - - it 'should make private proxies for private methods' do - object = MockableClass.new - object.should_receive(:private_method) - object.private_methods.include?('private_method').should == true - object.public_method - end - - it 'should make protected proxies for protected methods' do - object = MockableClass.new - object.should_receive(:protected_method) - object.protected_methods.include?('protected_method').should == true - object.public_method - end - end examples/pure/partial_mock_example.rb @@ -102,5 +102,41 @@ module Spec lambda { o.stub!(:bar) }.should_not raise_error(NoMethodError) end end + + describe "Method visibility when using partial mocks" do + class MockableClass + def public_method + private_method + protected_method + end + protected + def protected_method; end + private + def private_method; end + end + + before(:each) do + @object = MockableClass.new + end + + it 'should keep public methods public' do + @object.should_receive(:public_method) + @object.public_methods.should include('public_method') + @object.public_method + end + + it 'should keep private methods private' do + @object.should_receive(:private_method) + @object.private_methods.should include('private_method') + @object.public_method + end + + it 'should keep protected methods protected' do + @object.should_receive(:protected_method) + @object.protected_methods.should include('protected_method') + @object.public_method + end + + end end end spec/spec/mocks/partial_mock_spec.rb de3f6cdb87e2a3a2502485ddf38717b9fc55efd7 Pat Maddox pat.maddox@gmail.com http://github.com/dchelimsky/rspec/commit/7af69e32cd778ab28b33e59c8c69238851554e95 7af69e32cd778ab28b33e59c8c69238851554e95 2008-05-22T09:51:41-07:00 2008-05-22T09:51:41-07:00 Moved specs around, little bit of cleanup 848ec3a513cf206122dde0e30e8b481308db4f2c Pat Maddox pat.maddox@gmail.com