lib/spy/subroutine.rb in spy-0.2.2 vs lib/spy/subroutine.rb in spy-0.2.3

@@ -303,17 +303,19 @@
       # retrieve all the spies from a given object
       # @param base_object
       # @param singleton_method [Boolean] (true) only get singleton_method_spies
       # @return [Array<Subroutine>]
       def get_spies(base_object, singleton_methods = true)
-        if singleton_methods
-        all_methods = base_object.public_methods(false) +
-          base_object.protected_methods(false) +
-          base_object.private_methods(false)
-        else
-          all_methods = base_object.instance_methods(false) +
-            base_object.private_instance_methods(false)
-        end
+        all_methods =
+          if singleton_methods
+            base_object.public_methods(false) +
+              base_object.protected_methods(false) +
+              base_object.private_methods(false)
+          else
+            base_object.public_instance_methods(false) +
+              base_object.protected_instance_methods(false) +
+              base_object.private_instance_methods(false)
+          end
 
         all_methods.map do |method_name|
           Agency.instance.find(get_spy_id(base_object.method(method_name)))
         end.compact
       end