Spy#mock_all

Author: ryanong

lib/spy.rb

@@ -110,11 +110,23 @@ def off_const(base_module, *constant_names)
     def mock(klass, *stubs)
       new_mock = Mock.new(klass).new
       if stubs.size > 0
-        Spy.on(new_mock, *stubs)
+        on(new_mock, *stubs)
       end
       new_mock
     end
 
+    def mock_all(klass, *stubs)
+      mock_klass = Mock.new(klass)
+      new_mock = mock_klass.new
+
+      spies = stubs.size > 0 ? on(new_mock, *stubs) : []
+
+      unstubbed_methods = mock_klass.mocked_methods - spies.map(&:method_name)
+      on(new_mock, *unstubbed_methods) if unstubbed_methods.size > 0
+
+      new_mock
+    end
+
     # unhook all methods
     def teardown
       Agency.instance.dissolve!

lib/spy/mock.rb

@@ -5,6 +5,7 @@ module Spy
   # original class it will raise an error.
   module Mock
     CLASSES_NOT_TO_OVERRIDE = [Enumerable, Numeric, Comparable, Class, Module, Object]
+    METHODS_NOT_TO_OVERRIDE = [:initialize, :method]
 
     def initialize
     end
@@ -57,8 +58,10 @@ def new(klass)
       def included(mod)
         method_classes = classes_to_override_methods(mod)
 
+        mocked_methods = []
         [:public, :protected, :private].each do |visibility|
           get_inherited_methods(method_classes, visibility).each do |method_name|
+            mocked_methods << method_name
             args = args_for_method(mod.instance_method(method_name))
 
             mod.class_eval <<-DEF_METHOD, __FILE__, __LINE__+1
@@ -70,6 +73,10 @@ def #{method_name}(#{args})
             DEF_METHOD
           end
         end
+
+        mod.define_singleton_method(:mocked_methods) do
+          mocked_methods
+        end
       end
 
       private
@@ -86,11 +93,12 @@ def classes_to_override_methods(mod)
       end
 
       def get_inherited_methods(klass_ancestors, visibility)
-        get_methods_method = "#{visibility}_instance_methods".to_sym
-        instance_methods = klass_ancestors.map(&get_methods_method)
+        instance_methods = klass_ancestors.map do |klass|
+         klass.send("#{visibility}_instance_methods".to_sym, false)
+        end
         instance_methods.flatten!
         instance_methods.uniq!
-        instance_methods - Object.send(get_methods_method)
+        instance_methods - METHODS_NOT_TO_OVERRIDE
       end
 
       def args_for_method(method)

test/integration/test_mocking.rb

@@ -0,0 +1,25 @@
+require 'test_helper'
+
+class TestMocking < MiniTest::Unit::TestCase
+  def teardown
+    Spy::Agency.instance.dissolve!
+  end
+
+  def test_spy_on_mock_does_not_raise
+    mock = Spy.mock(Pen)
+    spy = Spy.on(mock, :write).and_return(:awesome)
+    assert_equal :awesome, mock.write("hello")
+    assert spy.has_been_called?
+  end
+
+  def test_spy_mock_shortcuts
+    mock = Spy.mock(Pen, :another, write_hello: :goodbye)
+    assert_nil mock.another
+    assert_equal :goodbye, mock.write_hello
+  end
+
+  def test_spy_mock_all
+    mock = Spy.mock_all(Pen)
+    assert_nil mock.another
+  end
+end

test/integration/test_subroutine_spying.rb

@@ -47,11 +47,4 @@ def test_spy_on_double_does_not_raise
     assert_equal :dumb, double.doubles_are
     assert spy.has_been_called?
   end
-
-  def test_spy_on_mock_does_not_raise
-    mock = Spy.mock(Pen)
-    spy = Spy.on(mock, :write).and_return(:awesome)
-    assert_equal :awesome, mock.write("hello")
-    assert spy.has_been_called?
-  end
 end

test/spy/test_mock.rb

@@ -8,7 +8,8 @@ def write_hello(other)
     end
 
     def setup
-      @pen = Spy.mock(BluePen)
+      @pen_mock = Mock.new(BluePen)
+      @pen = @pen_mock.new
     end
 
     def teardown
@@ -67,6 +68,14 @@ def test_that_and_call_original_works
       end
     end
 
+    def test_mocked_methods
+      pen_methods = Pen.public_instance_methods(false) +
+        Pen.protected_instance_methods(false) +
+        Pen.private_instance_methods(false)
+      pen_methods.delete(:initialize)
+      assert_equal pen_methods.sort, @pen_mock.mocked_methods.sort
+    end
+
     def test_base_class_methods_are_not_stubbed
       (Object.instance_methods - [:tap, :pretty_print_inspect]).each do |method_name|
         object_method = Object.instance_method(method_name)