better_receive 0.1.1 → 0.2.0

data/lib/better_receive/mock.rb

@@ -1,29 +1,66 @@
 module BetterReceive
   class Mock
-    attr_reader :expectation, :object
 
     def initialize(object)
       @object = object
     end
 
-    def verify(selector, options, &block)
-      object.should respond_to(selector)
+    def responds_to_and_receives(selector, options, &block)
+      better_respond_to(selector)
 
       mock_method(selector, options, &block)
     end
 
+    def rspec_verify
+    end
+
     private
 
-    def mock_method(selector, options, &block)
-      mock_proxy = object.send(:__mock_proxy)
+    attr_reader :object
 
-      location = options[:expected_from] || caller(1)[2]
-      @expectation = mock_proxy.add_message_expectation(location, selector, options, &block)
+    def better_respond_to(selector)
+      case object
+      when RSpec::Mocks::AnyInstance::Recorder
+        any_instance_should_respond_to selector
+      else
+        object.should respond_to(selector)
+      end
+    end
+
+    def any_instance_should_respond_to(selector)
+      klass = object.instance_variable_get(:@klass)
+      unless klass.method_defined?(selector)
+        raise "Expected instances of #{klass.name} to respond to :#{selector}"
+      end
     end
 
     def respond_to(selector)
       RSpec::Matchers::BuiltIn::RespondTo.new(selector)
     end
 
+    def mock_method(selector, options, &block)
+      case object
+      when RSpec::Mocks::AnyInstance::Recorder
+        any_instance_should_receive selector, &block
+      else
+        location = options[:expected_from] || caller(1)[2]
+        mock_proxy.add_message_expectation(location, selector, options, &block)
+      end
+    end
+
+    def any_instance_should_receive(selector, &block)
+      RSpec::Mocks::space.add(object)
+
+      object.instance_variable_set(:@expectation_set, true)
+      object.send(:observe!, selector)
+
+      expectation_chain = RSpec::Mocks::AnyInstance::PositiveExpectationChain.new(selector, &block)
+      object.message_chains.add(selector, expectation_chain)
+    end
+
+    def mock_proxy
+      @mock_proxy ||= object.send(:__mock_proxy)
+    end
+
   end
 end

data/lib/better_receive/version.rb

@@ -1,3 +1,3 @@
 module BetterReceive
-  VERSION = "0.1.1"
+  VERSION = "0.2.0"
 end

data/lib/better_receive.rb

@@ -3,7 +3,7 @@ require "better_receive/mock"
 
 module BetterReceive
   def better_receive(method_name, options={}, &block)
-    Mock.new(self).verify(method_name, options, &block)
+    Mock.new(self).responds_to_and_receives(method_name, options, &block)
   end
 end
 

data/spec/lib/better_receive_spec.rb

@@ -60,6 +60,40 @@ describe BetterReceive do
           foo.better_receive(:bar, passed: true, &block_param)
         end
       end
+
+      context "on .any_instance" do
+        context "when the method is defined" do
+          context "and the method is called" do
+            it 'does not raise an error' do
+              expect {
+                Foo.any_instance.better_receive(:bar)
+                foo.bar
+              }.to_not raise_error
+            end
+          end
+
+          context 'and the method is not called' do
+            it 'does raise an error' do
+              expect {
+                Foo.any_instance.better_receive(:bar)
+                RSpec::Mocks::space.verify_all
+              }.to raise_error(RSpec::Mocks::MockExpectationError) { |error|
+                error.message.should == "Exactly one instance should have received the following message(s) but didn't: bar"
+              }
+            end
+          end
+        end
+
+        context 'when the method is not defined' do
+          it 'raises an error' do
+            expect {
+              Foo.any_instance.better_receive(:baz)
+            }.to raise_error { |error|
+              error.message.should == "Expected instances of Foo to respond to :baz"
+            }
+          end
+        end
+      end
     end
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: better_receive
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.2.0
   prerelease: 
 platform: ruby
 authors:
@@ -10,7 +10,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-01-13 00:00:00.000000000 Z
+date: 2013-01-15 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rspec