expects_chain 0.0.1

data/lib/expects_chain/expectation.rb

@@ -0,0 +1,38 @@
+module ExpectsChain
+  class Expectation
+    def initialize type, object, *args
+      @type = type
+      @object = object
+      @expectations = args 
+    end
+
+    def returns value
+      set_up_call_chain value
+      value
+    end
+
+    private
+
+    def set_up_call_chain value
+      current_mock = value
+      first_expectation = @expectations.shift
+      @expectations.reverse.each do |expectation|
+        old_mock, current_mock = current_mock, mocker.send(@type)
+        set_expectation(current_mock, expectation, old_mock)
+      end
+      set_expectation(@object, first_expectation, current_mock)
+    end
+
+    def set_expectation obj, method, ret
+      if method.kind_of?(Array) && method.first.kind_of?(Symbol)
+        mocker.send("#{@type}_object_with_arguments".to_sym, obj, method.shift, method, ret)
+      else
+        mocker.send("#{@type}_object".to_sym, obj, method, ret)
+      end
+    end
+
+    def mocker
+      @_mocker ||= Mockers::Mocha
+    end
+  end
+end

data/lib/expects_chain/mockers/mocha.rb

@@ -0,0 +1,29 @@
+module ExpectsChain
+  module Mockers
+    module Mocha
+      def self.mock_object obj, method, ret
+        obj.expects(method).at_least_once.returns(ret)
+      end
+
+      def self.mock_object_with_arguments obj, method, attrs, ret
+        obj.expects(method).with(*attrs).at_least_once.returns(ret)
+      end
+
+      def self.stub_object
+        obj.stubs(method).returns(ret)
+      end
+
+      def self.stub_object_with_arguments obj, method, attrs, ret
+        obj.stubs(method).with(*attrs).at_least_once.returns(ret)
+      end
+
+      def self.mock
+        ::Mocha::Mock.new(nil)
+      end
+
+      def self.stub
+        ::Mocha::Mock.new(nil)
+      end
+    end
+  end
+end

data/lib/expects_chain.rb

@@ -0,0 +1,20 @@
+# encoding: utf-8
+
+require 'expects_chain/expectation'
+require 'expects_chain/mockers/mocha'
+
+module ExpectsChain
+  module Methods
+    def expects_chain *args
+      ExpectsChain::Expectation.new(:mock, self, *args) 
+    end
+
+    def stubs_chain *args
+      ExpectsChain::Expectation.new(:stub, self, *args) 
+    end
+  end
+end
+
+class Object
+  include ExpectsChain::Methods 
+end

metadata

@@ -0,0 +1,48 @@
+--- !ruby/object:Gem::Specification
+name: expects_chain
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Tomasz Pewiński
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-04-21 00:00:00.000000000 Z
+dependencies: []
+description: RSpec extension that allows you to set chained expectations on objects.
+  Currently supports mocha
+email: pewniak747@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/expects_chain.rb
+- lib/expects_chain/expectation.rb
+- lib/expects_chain/mockers/mocha.rb
+homepage: http://github.com/pewniak747/expects_chain
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.11
+signing_key: 
+specification_version: 3
+summary: RSpec extension that allows you to set chained expectations on objects.
+test_files: []