jeremymcanally-stump 0.0.1 → 0.0.2

data/lib/stump/proxy.rb

@@ -0,0 +1,103 @@
+class Object
+  # Creates a proxy method on an object.  In this setup, it places an expectation on an object (like a mock)
+  # but still calls the original method.  So if you want to make sure the method is called and still return
+  # its value, or simply want to invoke the side effects of a method and return a stubbed value, then you can
+  # do that.
+  #
+  # ==== Examples
+  #
+  #     class Parrot
+  #       def speak!
+  #         puts @words
+  #       end
+  #
+  #       def say_this(words)
+  #         @words = words
+  #         "I shall say #{words}!"
+  #       end
+  #     end
+  #
+  #     # => test/your_test.rb
+  #     sqawky = Parrot.new
+  #     sqawky.proxy!(:say_this)
+  #     # Proxy method still calls original method...
+  #     sqawky.say_this("hey")   # => "I shall say hey!"
+  #     sqawky.speak!            # => "hey"
+  #
+  #     sqawky.proxy!(:say_this, "herro!")
+  #     # Even though we return a stubbed value...
+  #     sqawky.say_this("these words")   # => "herro!"
+  #     # ...the side effects are still there.
+  #     sqawky.speak!                    # => "these words"
+  #
+  # TODO: This implementation is still very rough.  Needs refactoring and refining.  Won't 
+  # work on ActiveRecord attributes, for example.
+  #
+  def proxy!(method, options = {}, &block)
+    Stump::Mocks.add([self, method])
+    
+    if respond_to?(method)
+      proxy_existing_method(method, options, &block)
+    else
+      proxy_missing_method(method, options, &block)
+    end
+  end
+  
+  protected
+  def proxy_existing_method(method, options = {}, &block)
+    method_alias = "__old_#{method}"
+    
+    meta_eval do
+      module_eval("alias #{method_alias} #{method}")
+    end
+        
+    behavior = if options[:return]
+                  lambda do |*args| 
+                    raise ArgumentError if args.length != method(method_alias.to_sym).arity
+                    
+                    Stump::Mocks.verify([self, method])
+
+                    if method(method_alias.to_sym).arity == 0
+                      send(method_alias)
+                    else
+                      send(method_alias, *args)
+                    end
+
+                    return options[:return]
+                  end
+                else
+                  lambda do |*args| 
+                    raise ArgumentError if args.length != method(method_alias.to_sym).arity
+
+                    Stump::Mocks.verify([self, method])
+                    
+                    if method(method_alias.to_sym).arity == 0
+                      return send(method_alias)
+                    else
+                      return send(method_alias, *args)
+                    end
+                  end
+                end
+
+    meta_def method, &behavior
+  end
+  
+  def proxy_missing_method(method, options = {}, &block)
+    behavior = if options[:return]
+                  lambda do |*args|
+                    Stump::Mocks.verify([self, method])
+                    
+                    method_missing(method, args)
+                    return options[:return]
+                  end
+                else
+                  lambda do |*args|
+                    Stump::Mocks.verify([self, method])
+      
+                    method_missing(method, args)
+                  end
+                end
+    
+    meta_def method, &behavior
+  end
+end

data/stump.gemspec

@@ -1,7 +1,7 @@
 Gem::Specification.new do |s|
   s.name     = "stump"
-  s.version  = "0.0.1"
-  s.date     = "2008-10-07"
+  s.version  = "0.0.2"
+  s.date     = "2008-10-30"
   s.summary  = "Stubbing and mocking that isn't painful."
   s.email    = "jeremy@entp.com"
   s.homepage = "http://github.com/jeremymcanally/stump"
@@ -26,6 +26,7 @@ Gem::Specification.new do |s|
     "lib/stump/mock.rb",
     "lib/stump/mocks.rb",
     "lib/stump/stub.rb",
+    "lib/stump/proxy.rb",
     "lib/stump/version.rb",
     "lib/stump/core_ext/test_case.rb",
     "setup.rb"
@@ -40,4 +41,4 @@ Gem::Specification.new do |s|
 
   s.rdoc_options = ["--main", "README.rdoc"]
   s.extra_rdoc_files = ["History.txt", "Manifest.txt", "README.rdoc"]
-end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: jeremymcanally-stump
 version: !ruby/object:Gem::Version 
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors: 
 - Jeremy McAnally
@@ -9,7 +9,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2008-10-07 00:00:00 -07:00
+date: 2008-10-30 00:00:00 -07:00
 default_executable: 
 dependencies: []
 
@@ -38,6 +38,7 @@ files:
 - lib/stump/mock.rb
 - lib/stump/mocks.rb
 - lib/stump/stub.rb
+- lib/stump/proxy.rb
 - lib/stump/version.rb
 - lib/stump/core_ext/test_case.rb
 - setup.rb