stubble 0.1.1 → 0.1.2

data/Rakefile

@@ -34,7 +34,7 @@ spec = Gem::Specification.new do |s|
 
   # Change these as appropriate
   s.name              = "stubble"
-  s.version           = "0.1.1"
+  s.version           = "0.1.2"
   s.summary           = "(experimental) simple stubbing"
   s.author            = "Matthew Rudy Jacobs"
   s.email             = "MatthewRudyJacobs@gmail.com"

data/lib/stubble.rb

@@ -87,16 +87,18 @@ module Stubble
     
     def track!(method_name)
       self._stubble.track!(method_name)
-      unless methods.include?("#{method_name}_with_stubble")
+
+      unstubbled_method_name = "__unstubbled__#{method_name}"
+      unless methods.include?(unstubbled_method_name)
         class_eval <<-RUBY
-          def #{method_name}_with_stubble(*args, &block)
+          alias #{unstubbled_method_name} #{method_name} 
+          def #{method_name}(*args, &block)
             if stubbed_return = self._stubble.called!(#{method_name.inspect}, args, block)
               stubbed_return.perform
             else
-              #{method_name}_without_stubble(*args, &block)
+              #{unstubbled_method_name}(*args, &block)
             end
           end
-          alias_method_chain #{method_name.inspect}, :stubble
         RUBY
       end
     end

data/test/stubble_test.rb

@@ -16,6 +16,10 @@ class StubbleTest < Test::Unit::TestCase
       @hits +=1
     end
     
+    def with_punctuation!
+      "punctuated"
+    end
+    
     def some_method_with_arguments(times, shout="boom")
       shout * times
     end
@@ -108,6 +112,20 @@ class StubbleTest < Test::Unit::TestCase
     assert_equal 1, this._stubble[:some_method].length # no new
   end
   
+  test "we can stub! punctuated methods" do
+    this = StubThis.new
+    assert_equal "punctuated", this.with_punctuation!
+    
+    Stubble.add_stubble!(this)
+    this.stub!(:with_punctuation!, [:a, :b, :c])
+    
+    assert_equal :a, this.with_punctuation!
+    assert_equal :b, this.with_punctuation!
+    assert_equal :c, this.with_punctuation!
+    
+    assert_equal "punctuated", this.with_punctuation!
+  end
+  
   test "stub! - defines the next values to be returned" do
     this = StubThis.new
     assert_equal 1, this.some_method

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: stubble
 version: !ruby/object:Gem::Version 
-  hash: 25
+  hash: 31
   prerelease: false
   segments: 
   - 0
   - 1
-  - 1
-  version: 0.1.1
+  - 2
+  version: 0.1.2
 platform: ruby
 authors: 
 - Matthew Rudy Jacobs