spy_rb 0.1.5 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 8961d7ee67921ac362c6fc641cc406cfc794262b
-  data.tar.gz: 950d59327fceab861d83df830251e847e4474c0b
+  metadata.gz: 9c9735d3d8cbb2057082d9e5bc41ca566de15ac3
+  data.tar.gz: ab3225f2107d4adb711b0d357dcfc51ebad01cb5
 SHA512:
-  metadata.gz: bd5e5a986047c5ad17c8f05807f2846f6bc142825af545b9eb954d9db7a429628b97f2da61e6abd45635c446b035c4e7f8c74c2cba9eff72ef48477d8332a1b6
-  data.tar.gz: 0a893d712ea97c103e9b62795fe6312ba763b85877037f74f38891dccbe8af9aacf74facc030ee023428563e9599fc23bb68184a48a21e36fd3d57795cb1137a
+  metadata.gz: 15f2c77adff4bff6479b40b103e2d0283752e9626ecef92181822ccf42cf3ea6d682e586763340b982a4db2d238a698dc0430016ca3b60d8210a38374bd2d373
+  data.tar.gz: defdc2587687c8deaffff5baa84bd880680bdb5e040e9eb9c255d3af3108ee33e4bfa051cf0ffa9711d09ebe8de102ec74f32ebfeca71b6d776f53b1ad59c130

data/lib/spy/api.rb

@@ -10,14 +10,15 @@ module Spy
     def on(*args)
       case args.length
       when 2
-        return core.add_spy *(args << :method)
+        spied, msg = *args
+        return core.add_spy(spied, spied.method(msg))
       end
       raise ArgumentError
     end
 
     # TODO docs
-    def on_any_instance(mod, msg)
-      core.add_spy(mod, msg, :instance_method)
+    def on_any_instance(spied, msg)
+      core.add_spy(spied, spied.instance_method(msg))
     end
 
     # Stops spying on the method and restores its original functionality
@@ -37,9 +38,11 @@ module Spy
       when 1
         return core.remove_all_spies if args.first == :all
       when 2
-        return core.remove_spy *(args << :method)
+        spied, msg = *args
+        return core.remove_spy(spied, spied.method(msg))
       when 3
-        return core.remove_spy *args
+        spied, msg, method_type = *args
+        return core.remove_spy(spied, spied.send(method_type, msg))
       end
       raise ArgumentError
     end

data/lib/spy/collection.rb

@@ -31,13 +31,19 @@ module Spy
     # Add a slicker interface that abstracts away Collection::Entry
     module SpyHelper
       def <<(spy)
-        entry = Collection::Entry.new(spy.receiver, spy.msg, spy.method_type)
+        receiver = spy.original.is_a?(Method) ? spy.original.receiver : nil
+        name = spy.original.name
+        klass = spy.original.class
+        entry = Collection::Entry.new(receiver, name, klass)
         entry.value = spy
         insert entry
       end
 
-      def pop(receiver, msg, method_type)
-        remove Collection::Entry.new(receiver, msg, method_type)
+      def pop(method)
+        receiver = method.is_a?(Method) ? method.receiver : nil
+        name = method.name
+        klass = method.class
+        remove Collection::Entry.new(receiver, name, klass)
       end
     end
 

data/lib/spy/core.rb

@@ -8,14 +8,14 @@ module Spy
       @spy_collection ||= Collection.new
     end
 
-    def add_spy(receiver, msg, method_type)
-      spy = Instance.new(receiver, msg, method_type)
+    def add_spy(spied, method)
+      spy = Instance.new(spied, method)
       spy_collection << spy
       spy.start
     end
 
-    def remove_spy(receiver, msg, method_type)
-      spy = spy_collection.pop(receiver, msg, method_type)
+    def remove_spy(spied, method)
+      spy = spy_collection.pop(method)
       spy.stop
     end
 

data/lib/spy/instance/strategy/intercept.rb

@@ -0,0 +1,34 @@
+module Spy
+  class Instance
+    module Strategy
+      class Intercept
+        def initialize(spy, intercept_target)
+          @spy = spy
+          @intercept_target = intercept_target
+        end
+
+        def apply
+          spy = @spy
+          @intercept_target.class_eval do
+            define_method spy.original.name do |*args|
+              spy.before_call(*args)
+              if spy.original.is_a?(UnboundMethod)
+                spy.original.bind(self).call(*args)
+              else
+                spy.original.call(*args)
+              end
+            end
+            send(spy.visibility, spy.original.name)
+          end
+        end
+
+        def undo
+          spy = @spy
+          @intercept_target.class_eval do
+            remove_method spy.original.name
+          end
+        end
+      end
+    end
+  end
+end

data/lib/spy/instance/strategy/wrap.rb

@@ -0,0 +1,35 @@
+module Spy
+  class Instance
+    module Strategy
+      class Wrap
+        def initialize(spy)
+          @spy = spy
+        end
+
+        def apply
+          spy = @spy
+          spy.original.owner.class_eval do
+            define_method spy.original.name do |*args|
+              spy.before_call(*args)
+              if spy.original.is_a?(UnboundMethod)
+                spy.original.bind(self).call(*args)
+              else
+                spy.original.call(*args)
+              end
+            end
+            send(spy.visibility, spy.original.name)
+          end
+        end
+
+        def undo
+          spy = @spy
+          spy.original.owner.class_eval do
+            remove_method spy.original.name
+            define_method spy.original.name, spy.original
+            send(spy.visibility, spy.original.name)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/spy/instance/strategy.rb

@@ -0,0 +1,26 @@
+require 'spy/instance/strategy/wrap'
+require 'spy/instance/strategy/intercept'
+
+module Spy
+  class Instance
+    module Strategy
+      class << self
+        def factory_build(spy)
+          if spy.original.is_a?(Method)
+            if spy.original.owner == spy.spied.singleton_class
+              Strategy::Wrap.new(spy)
+            else
+              Strategy::Intercept.new(spy, spy.spied.singleton_class)
+            end
+          else
+            if spy.original.owner == spy.spied
+              Strategy::Wrap.new(spy)
+            else
+              Strategy::Intercept.new(spy, spy.spied)
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/spy/instance.rb

@@ -1,5 +1,6 @@
 require 'spy/callbacks/with_args'
 require 'spy/callbacks/when'
+require 'spy/instance/strategy'
 
 # An instance of a spied method
 # - Holds a reference to the original method
@@ -7,42 +8,24 @@ require 'spy/callbacks/when'
 # - Provides hooks for callbacks
 module Spy
   class Instance
-    attr_reader :receiver, :method_type, :msg, :original, :call_count, :visibility
+    attr_reader :original, :spied, :strategy, :call_count, :visibility
 
-    def initialize(receiver, msg, method_type)
-      @msg = msg
-      @receiver = receiver
-      @method_type = method_type
+    def initialize(spied, original)
+      @spied = spied
+      @original = original
+      @visibility = extract_visibility
       @before_filters = []
       @call_count = 0
-
-      # Cache the original method for unwrapping later
-      @original = @receiver.send(method_type, msg)
-      @visibility = extract_visibility
+      @strategy = Strategy.factory_build(self)
     end
 
     def start
-      context = self
-      original.owner.instance_eval do
-        define_method context.msg do |*args|
-          context.before_call(*args)
-          if context.original.respond_to? :bind
-            result = context.original.bind(self).call(*args)
-          else
-            result = context.original.call(*args)
-          end
-          result
-        end
-        send(context.visibility, context.msg)
-      end
+      @strategy.apply
       self
     end
 
     def stop
-      context = self
-      original.owner.instance_eval do
-        define_method context.msg, context.original
-      end
+      @strategy.undo
       self
     end
 
@@ -69,11 +52,11 @@ module Spy
       owner = @original.owner
       [:public, :protected, :private].each do |vis|
         query = "#{vis}_method_defined?"
-        if owner.respond_to?(query) && owner.send(query, @msg)
+        if owner.respond_to?(query) && owner.send(query, @original.name)
           return vis
         end
       end
-      raise NoMethodError, "couldn't find method #{@msg} belonging to #{owner}"
+      raise NoMethodError, "couldn't find method #{@original.name} belonging to #{owner}"
     end
   end
 end

data/lib/spy/version.rb

@@ -1,3 +1,3 @@
 module Spy
-  VERSION = '0.1.5'
+  VERSION = '0.2.0'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: spy_rb
 version: !ruby/object:Gem::Version
-  version: 0.1.5
+  version: 0.2.0
 platform: ruby
 authors:
 - Josh Bodah
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-12-18 00:00:00.000000000 Z
+date: 2015-02-27 00:00:00.000000000 Z
 dependencies: []
 description: Spy brings everything that's great about Sinon.JS to Ruby. Mocking frameworks
   work by stubbing out functionality. Spy works by listening in on functionality and
@@ -28,6 +28,9 @@ files:
 - lib/spy/core.rb
 - lib/spy/errors.rb
 - lib/spy/instance.rb
+- lib/spy/instance/strategy.rb
+- lib/spy/instance/strategy/intercept.rb
+- lib/spy/instance/strategy/wrap.rb
 - lib/spy/version.rb
 homepage: https://github.com/jbodah/spy_rb
 licenses:
@@ -54,4 +57,3 @@ signing_key:
 specification_version: 4
 summary: SinonJS-style Test Spies for Ruby
 test_files: []
-has_rdoc: