Merge pull request #55 from rspec/fix-case-equality-by-mocking-===

Mock === on the modal_class when defining a mock

Author: JonRowe

lib/rspec/active_model/mocks/mocks.rb

@@ -124,6 +124,11 @@ def self.primary_key; :id; end
                :blank? => false}.merge(stubs)
 
       double("#{model_class.name}_#{stubs[:id]}", stubs).tap do |m|
+        if model_class.method(:===).owner == Module && !stubs.has_key?(:===)
+          allow(model_class).to receive(:===).and_wrap_original do |original, other|
+            m === other || original.call(other)
+          end
+        end
         msingleton = class << m; self; end
         msingleton.class_eval do
           include ActiveModelInstanceMethods

spec/rspec/active_model/mocks/mock_model_spec.rb

@@ -157,6 +157,50 @@
     end
   end
 
+  describe "#===" do
+    it "works with a case statement" do
+      case mock_model(MockableModel)
+      when MockableModel then true
+      else
+        raise
+      end
+
+      case :not_mockable_model
+      when MockableModel then raise
+      else
+        true
+      end
+    end
+
+    it "won't break previous stubs" do
+      allow(MockableModel).to receive(:===).with("string") { true }
+      mock_model(MockableModel)
+
+      case "string"
+      when MockableModel then true
+      else
+        raise
+      end
+    end
+
+    it "won't override class definitions" do
+      another_mockable_model =
+        Class.new(MockableModel) do
+          def self.===(other)
+            true
+          end
+        end
+
+      mock_model(another_mockable_model)
+
+      case "string"
+      when another_mockable_model then true
+      else
+        raise
+      end
+    end
+  end
+
   describe "#kind_of?" do
     before(:each) do
       @model = mock_model(SubMockableModel)

spec/rspec/active_model/mocks/stub_model_spec.rb

@@ -161,6 +161,21 @@ def model_class
       end
       expect(model).to be(@block_arg)
     end
+  end
+
+  describe "#===" do
+    it "works with a case statement" do
+      case stub_model(MockableModel)
+      when MockableModel then true
+      else
+        raise
+      end
 
+      case :not_stub_model
+      when MockableModel then raise
+      else
+        true
+      end
+    end
   end
 end