Add a basic notion of Blacklight::Document equality

projectblacklight · Aug 3, 2017 · eae5950 · eae5950
1 parent 3aa180a
commit eae5950
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/app/models/concerns/blacklight/document/active_model_shim.rb b/app/models/concerns/blacklight/document/active_model_shim.rb
@@ -31,6 +31,13 @@ def id
       self[self.class.unique_key]
     end
 
+    def ==(other)
+      super ||
+        (other.instance_of?(self.class) &&
+          id &&
+          other.id == id)
+    end
+
     def _read_attribute(attr)
       self[attr]
     end

diff --git a/spec/models/blacklight/document/active_model_shim_spec.rb b/spec/models/blacklight/document/active_model_shim_spec.rb
@@ -30,4 +30,14 @@ def documents
       expect(MockDocument.find(1).id).to be 1
     end
   end
+
+  describe "#==" do
+    it 'is equal for the same id' do
+      expect(MockDocument.new(id: 1) == MockDocument.new(id: 1)).to eq true
+    end
+
+    it 'is not equal if the ids differ' do
+      expect(MockDocument.new(id: 1) == MockDocument.new(id: 2)).to eq false
+    end
+  end
 end