expects_chain 0.1.1 → 0.2.0

data/lib/expects_chain/expectation.rb

@@ -7,10 +7,17 @@ module ExpectsChain
     end
 
     def returns value
+      @rtype = :returns
       set_up_call_chain value
       value
     end
 
+    def raises exception
+      @rtype = :raises
+      set_up_call_chain exception 
+      exception
+    end
+
     def self.mocker
       if @_mocker.nil?
         framework = RSpec.configuration.mock_framework.framework_name.to_s.capitalize.to_sym
@@ -25,21 +32,21 @@ module ExpectsChain
     def set_up_call_chain value
       current_mock = value
       first_expectation = @expectations.shift
-      @expectations.reverse.each do |expectation|
+      @expectations.reverse.each_with_index do |expectation, idx|
         old_mock, current_mock = current_mock, mocker.send(@type)
-        set_expectation(current_mock, expectation, old_mock)
+        set_expectation(current_mock, expectation, old_mock, idx == 0)
       end
       set_expectation(@object, first_expectation, current_mock)
     end
 
-    def set_expectation obj, method, ret
-      type = (@type == :mock ? :expects : :stubs)
-      rtype = :returns
+    def set_expectation obj, method, ret, last = false
+      type, rtype = (@type == :mock ? :expects : :stubs), @rtype
+      new_mock = mocker.new(obj)
       if method.kind_of?(Array) && method.first.kind_of?(Symbol)
-        mocker.new(obj).send(type, method.shift).with(method).send(rtype, ret)
-      else
-        mocker.new(obj).send(type, method).send(rtype, ret)
+        new_mock.send(type, method.shift).with(method)
+      else new_mock.send(type, method)
       end
+      new_mock.send(last ? rtype : :returns, ret)
     end
 
     def mocker

data/lib/expects_chain/mockers/base.rb

@@ -9,6 +9,12 @@ module ExpectsChain
         @object = @object.with(*args)
         self
       end
+
+      private
+
+      def generate_exception object
+        object.is_a?(Class) ? object.new : object
+      end
     end
   end
 end

data/lib/expects_chain/mockers/mocha.rb

@@ -15,6 +15,10 @@ module ExpectsChain
         @object = @object.returns(ret)
       end
 
+      def raises ex
+        @object.raises(generate_exception(ex))
+      end
+
       def self.mock
         ::Mocha::Mock.new(nil)
       end

data/lib/expects_chain/mockers/rspec.rb

@@ -12,7 +12,11 @@ module ExpectsChain
       end
 
       def returns ret
-        @object = @object.and_return(ret)
+        @object.and_return(ret)
+      end
+
+      def raises ex
+        @object.and_raise(generate_exception(ex))
       end
 
       def self.mock

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: expects_chain
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.2.0
   prerelease: 
 platform: ruby
 authors: