orangutan 0.0.5 → 0.0.6

data/README

@@ -82,6 +82,20 @@ calls.each {|call| puts "#{call.name}.#{call.method}(#{call.args.join(',')})" }
 
 # Recorded call sequences can also be used for assertions in combination with or instead of checking expectations.
 
+# Stubbing methods on object instances
+
+register_object :math, Math   # registers the object (this just gives the object a name for referencing in calls)
+stub_method :math, :cos       # replaces the implementation of the cos method with a recorder
+so_when(:math).receives(:cos).with(0).return("i just can't take it anymore")
+
+# Registering as a mock framework with rspec
+
+require 'orangutan/mock_adapter'
+Spec::Runner.configure do |config|
+  # turn off mocking
+  config.mock_with Orangutan::MockAdapter
+end
+
 Questions:
 
 * Why Orangutan?
@@ -94,8 +108,4 @@ Definately not.  What a ridiculous question.  I'm appalled.
 
 * What's Chantek?
 
-Chantek is a famous orangutan that can solve sudokus and the rubik's cube - http://en.wikipedia.org/wiki/Chantek
-
-* Why do I not need to register this as a mock framework?
-
-Most frameworks such as rspec's mocks, mocha and stubba replace methods on existing classes.  Because this can't be done with this library - you can only create pure stub objects and setup method behaviours, there's nothing that needs to be undone at the end of a test/example.
+Chantek is a famous orangutan that can solve sudokus and the rubik's cube - http://en.wikipedia.org/wiki/Chantek

data/lib/orangutan/chantek.rb

@@ -1,7 +1,9 @@
+require 'orangutan/object_extensions'
 require 'orangutan/stub_base'
 require 'orangutan/expectation'
 require 'orangutan/call'
 require 'orangutan/reflector'
+require 'orangutan/stub_include'
 
 module Orangutan 
   module Chantek
@@ -16,6 +18,37 @@ module Orangutan
       @expectations = {}
       @stubs= {}
       @events = {}
+      @stubbed_methods = []
+    end
+
+    def register_object name, object
+      @stubs[name] ||= object
+    end
+
+    def stub_method name, method
+      object = @stubs[name]
+      raise "could not find an object registered with the name #{name}" unless object
+      chantek = self
+      object.instance_eval do
+        @parent = chantek
+        @name = name
+      end
+      @stubbed_methods << [name, method]
+      object.eigen_eval do
+        alias_method("__unstubbed_#{method}", method)
+      end
+      implement_method object.eigen, method
+    end
+    
+    def restore_method name, method
+      @stubs[name].eigen_eval do
+        alias_method(method, "__unstubbed_#{method}")
+      end
+      @stubbed_methods.delete [name, method]
+    end
+
+    def restore_methods
+      @stubbed_methods.each {|tuple| restore_method tuple[0], tuple[1]}
     end
 
     def stub name, params={}
@@ -39,6 +72,8 @@ module Orangutan
 
     def implement_method clazz, name
       clazz.instance_eval do
+        include StubInclude
+
         define_method name do |*args|
           yield_container, return_container = __react__(name, args)
           yield_container.value.each {|v| yield *v } if yield_container && block_given?

data/lib/orangutan/mock_adapter.rb

@@ -11,6 +11,7 @@ module Orangutan
     end
 
     def teardown_mocks_for_rspec
+      restore_methods
     end
   end
 end

data/lib/orangutan/object_extensions.rb

@@ -0,0 +1,11 @@
+class Object
+  def eigen
+    class << self
+      self
+    end
+  end
+  
+  def eigen_eval(&block)
+    eigen.instance_eval(&block)
+  end
+end

data/lib/orangutan/stub_base.rb

@@ -1,7 +1,11 @@
+require 'orangutan/stub_include'
+
 module Orangutan
-  class StubBase
+  class StubBase    
     instance_methods.each { |m| undef_method m unless m =~ /^__/ }
 
+    include StubInclude
+
     def initialize name, parent, recursive=false
       @name, @parent, @recursive = name, parent, recursive
     end
@@ -13,22 +17,5 @@ module Orangutan
       end
       __return__(method, return_container)
     end
-  private
-    def __return__ method, return_container
-      return *return_container.value if return_container
-      return @parent.stub(:"#{@name}/#{method}", :recursive => true) if @recursive
-      nil
-    end
-  
-    def __react__ method, args
-      yield_container, return_container = nil, nil
-      @parent.calls << Call.new(@name, method, args)
-      first_match = @parent.first_match(@name, method, args)
-      if first_match
-        first_match.raiser.execute if first_match.raiser
-        yield_container, return_container =  first_match.yield_container, first_match.return_container
-      end
-      return yield_container, return_container
-    end
   end
 end

data/lib/orangutan/stub_include.rb

@@ -0,0 +1,20 @@
+module Orangutan
+  module StubInclude
+    def __return__ method, return_container
+      return *return_container.value if return_container
+      return @parent.stub(:"#{@name}/#{method}", :recursive => true) if @recursive
+      nil
+    end
+  
+    def __react__ method, args
+      yield_container, return_container = nil, nil
+      @parent.calls << Call.new(@name, method, args)
+      first_match = @parent.first_match(@name, method, args)
+      if first_match
+        first_match.raiser.execute if first_match.raiser
+        yield_container, return_container =  first_match.yield_container, first_match.return_container
+      end
+      return yield_container, return_container
+    end
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: orangutan
 version: !ruby/object:Gem::Version 
-  version: 0.0.5
+  version: 0.0.6
 platform: ruby
 authors: 
 - Mark Ryall
@@ -9,7 +9,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2009-12-23 00:00:00 +11:00
+date: 2010-03-06 00:00:00 +11:00
 default_executable: 
 dependencies: []
 
@@ -29,9 +29,11 @@ files:
 - lib/orangutan/container.rb
 - lib/orangutan/expectation.rb
 - lib/orangutan/mock_adapter.rb
+- lib/orangutan/object_extensions.rb
 - lib/orangutan/raiser.rb
 - lib/orangutan/reflector.rb
 - lib/orangutan/stub_base.rb
+- lib/orangutan/stub_include.rb
 - lib/orangutan.rb
 - README
 - MIT-LICENSE