stubble 0.1.0 → 0.1.1

data/Rakefile

@@ -2,6 +2,9 @@ require 'rake'
 require 'rake/testtask'
 require 'rake/rdoctask'
 
+require "rubygems"
+require "rake/gempackagetask"
+
 desc 'Default: run unit tests.'
 task :default => :test
 
@@ -22,20 +25,6 @@ Rake::RDocTask.new(:rdoc) do |rdoc|
   rdoc.rdoc_files.include('lib/**/*.rb')
 end
 
-require "rubygems"
-require "rake/gempackagetask"
-require "rake/rdoctask"
-
-require "rake/testtask"
-Rake::TestTask.new do |t|
-  t.libs << "test"
-  t.test_files = FileList["test/**/*_test.rb"]
-  t.verbose = true
-end
-
-
-task :default => ["test"]
-
 # This builds the actual gem. For details of what all these options
 # mean, and other ones you can add, check the documentation here:
 #
@@ -45,7 +34,7 @@ spec = Gem::Specification.new do |s|
 
   # Change these as appropriate
   s.name              = "stubble"
-  s.version           = "0.1.0"
+  s.version           = "0.1.1"
   s.summary           = "(experimental) simple stubbing"
   s.author            = "Matthew Rudy Jacobs"
   s.email             = "MatthewRudyJacobs@gmail.com"

data/lib/stubble.rb

@@ -32,8 +32,14 @@ module Stubble
       @tracking.delete(method_name)
     end
     
-    def stub!(method_name, values)
-      @stubs[method_name] = values
+    def raise!(method_name, exception)
+      @stubs[method_name] ||= []
+      @stubs[method_name] << RaiseStub.new(exception)
+    end
+    
+    def stub!(method_name, value)
+      @stubs[method_name] ||= []
+      @stubs[method_name] << ReturnStub.new(value)
     end
     
     def unstub!(method_name)
@@ -57,6 +63,26 @@ module Stubble
   end
   module_function :add_stubble!
   
+  class ReturnStub
+    def initialize(value)
+      @value = value
+    end
+    
+    def perform
+      return @value
+    end
+  end
+  
+  class RaiseStub
+    def initialize(exception)
+      @exception = exception
+    end
+    
+    def perform
+      raise @exception
+    end
+  end
+  
   module StubMethods
     
     def track!(method_name)
@@ -65,7 +91,7 @@ module Stubble
         class_eval <<-RUBY
           def #{method_name}_with_stubble(*args, &block)
             if stubbed_return = self._stubble.called!(#{method_name.inspect}, args, block)
-              stubbed_return
+              stubbed_return.perform
             else
               #{method_name}_without_stubble(*args, &block)
             end
@@ -81,7 +107,15 @@ module Stubble
     
     def stub!(method_name, values)
       track!(method_name)
-      self._stubble.stub!(method_name, values)
+      values.each do |value|
+        self._stubble.stub!(method_name, value)
+      end
+    end
+    
+    def raise!(method_name, exception)
+      track!(method_name)
+      
+      self._stubble.raise!(method_name, exception)
     end
     
     def unstub!(method_name)

data/test/stubble_test.rb

@@ -122,6 +122,37 @@ class StubbleTest < Test::Unit::TestCase
     assert_equal 2, this.some_method
   end
   
+  test "stub! - can be done progressively" do
+    this = StubThis.new
+    assert_equal 1, this.some_method
+    
+    Stubble.add_stubble!(this)
+    this.stub!(:some_method, [:a])
+    this.stub!(:some_method, [:b, :c])
+    
+    assert_equal :a, this.some_method
+    assert_equal :b, this.some_method
+    assert_equal :c, this.some_method
+    
+    assert_equal 2, this.some_method
+  end
+  
+  class SomeMagicError < RuntimeError ; end
+  
+  test "raise!" do
+     this = StubThis.new
+     assert_equal 1, this.some_method
+
+     Stubble.add_stubble!(this)
+     this.raise!(:some_method, SomeMagicError)
+     
+     assert_raise(SomeMagicError) do
+       this.some_method
+     end
+
+     assert_equal 2, this.some_method
+   end
+  
   test "unstub! - cancels a stub" 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: 27
+  hash: 25
   prerelease: false
   segments: 
   - 0
   - 1
-  - 0
-  version: 0.1.0
+  - 1
+  version: 0.1.1
 platform: ruby
 authors: 
 - Matthew Rudy Jacobs
@@ -15,7 +15,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2010-12-17 00:00:00 +08:00
+date: 2011-02-01 00:00:00 +08:00
 default_executable: 
 dependencies: []