allow unstubing of instances that are a sub class

lib/rspec/mocks/space.rb

@@ -49,7 +49,15 @@ def remove_any_instance_recorder_for(klass)
       end
 
       def proxies_of(klass)
-        proxies_by_klass[klass].map { |id| proxies[id] }
+        proxies_by_klass.inject([]) do |klass_proxies, (proxy_klass, object_ids)|
+          more_proxies =
+            if proxy_klass.ancestors.include?(klass) || klass.ancestors.include?(proxy_klass)
+              object_ids.map { |id| proxies[id] }
+            else
+              []
+            end
+          klass_proxies + more_proxies
+        end
       end
 
       def proxy_for(object)

spec/rspec/mocks/any_instance_spec.rb

@@ -266,6 +266,8 @@ class RSpec::SampleRspecTestClass;end
       end
 
       context "unstub implementation" do
+        let(:sub_klass) { Class.new(klass) }
+
         it "replaces the stubbed method with the original method" do
           klass.any_instance.stub(:existing_method)
           klass.any_instance.unstub(:existing_method)
@@ -287,6 +289,14 @@ class RSpec::SampleRspecTestClass;end
           expect(obj.existing_method).to eq(:existing_method_return_value)
         end
 
+        it "removes stubs from sub class after invokation when super class was originally stubbed" do
+          klass.any_instance.stub(:existing_method).and_return(:any_instance_value)
+          obj = sub_klass.new
+          obj.existing_method
+          klass.any_instance.unstub(:existing_method)
+          expect(obj.existing_method).to eq(:existing_method_return_value)
+        end
+
         it "does not remove any stubs on the local instance" do
           klass.any_instance.stub(:existing_method).and_return(:any_instance_value)
           obj = klass.new

spec/rspec/mocks/space_spec.rb

@@ -3,17 +3,24 @@
 module RSpec::Mocks
   describe Space do
 
-    describe "#proxies_of_any_instance_recorder(klass)" do
+    describe "#proxies_of(klass)" do
       let(:space) { Space.new }
 
       before do
         space.proxy_for(Object.new)
-        space.proxy_for(String.new)
+        space.proxy_for(Object.new)
+        space.proxy_for(Array.new)
         space.proxy_for(String.new)
       end
 
       it 'returns the proxies for objects of the same type as klass' do
-        expect(space.proxies_of(String).size).to eq 2
+        expect(space.proxies_of(Object).size).to eq 4
+        expect(space.proxies_of(Object)).to be_all { |obj| obj.is_a? RSpec::Mocks::PartialMockProxy }
+      end
+
+      it 'returns the proxies of super classes of klass' do
+        expect(space.proxies_of(String).size).to eq 3
+        expect(space.proxies_of(String)).to be_all { |obj| obj.is_a? RSpec::Mocks::PartialMockProxy }
       end
     end