RDBC-471 Changed the concept of _RefEq and wrote test

Author: poissoncorp

pyravendb/store/document_session.py

@@ -27,12 +27,11 @@ def __init__(self, ref):
             return
         self.ref = ref
 
+    # As we split the hashable and unhashable items into separate collections, we only compare _RefEq to other _RefEq
     def __eq__(self, other):
-        return (
-            True
-            if id(self.ref) == id(other)
-            else (False if id(self.ref) == id(other) else id(self.ref) == id(other.ref))
-        )
+        if isinstance(other, _RefEq):
+            return id(self.ref) == id(other.ref)
+        raise TypeError("Expected _RefEq type object")
 
     def __hash__(self):
         return id(self.ref)

pyravendb/tests/session_tests/test_documents_by_entity.py

@@ -0,0 +1,23 @@
+from pyravendb.tests.test_base import TestBase
+from pyravendb.store.document_session import _RefEq
+from dataclasses import dataclass
+
+
+@dataclass()
+class TestData:
+    entry: str
+
+
+class TestDocumentsByEntity(TestBase):
+    def setUp(self):
+        super(TestDocumentsByEntity, self).setUp()
+
+    def test_ref_eq(self):
+        data = TestData(entry="classified")
+
+        wrapped_data_1 = _RefEq(data)
+        wrapped_data_2 = _RefEq(wrapped_data_1)
+
+        self.assertTrue(wrapped_data_2 == wrapped_data_1)
+        with self.assertRaises(TypeError):
+            wrapped_data_1 == data