feat: instances are now hashable (#44)

Author: Michael Kryukov

mongomock_motor/__init__.py

@@ -192,6 +192,9 @@ def __eq__(self, other):
     def __getattr__(self, name):
         return getattr(self.__collection, name)
 
+    def __hash__(self):
+        return hash(self.__collection)
+
     def find(self, *args, **kwargs) -> AsyncCursor:
         return AsyncCursor(self.__collection.find(*args, **kwargs))
 
@@ -269,6 +272,9 @@ def __getattr__(self, name):
 
         return self.get_collection(name)
 
+    def __hash__(self):
+        return hash(self.__database)
+
 
 @masquerade_class('motor.motor_asyncio.AsyncIOMotorClient')
 @with_async_methods(
@@ -317,6 +323,9 @@ def __getattr__(self, name):
 
         return self.get_database(name)
 
+    def __hash__(self):
+        return hash(self.__client)
+
 
 @contextmanager
 def enabled_gridfs_integration():

tests/test_hashable.py

@@ -0,0 +1,24 @@
+import pytest
+
+from mongomock_motor import AsyncMongoMockClient
+
+
+@pytest.mark.anyio
+async def test_hashing_is_possible():
+    hash(AsyncMongoMockClient())
+    hash(AsyncMongoMockClient()['database'])
+    hash(AsyncMongoMockClient()['database']['collection'])
+
+
+@pytest.mark.anyio
+async def test_hashing_is_stable():
+    assert hash(AsyncMongoMockClient()) == hash(AsyncMongoMockClient())
+    assert hash(AsyncMongoMockClient('localhost')) == hash(
+        AsyncMongoMockClient('not.localhost')
+    )
+    assert hash(AsyncMongoMockClient()['database1']) == hash(
+        AsyncMongoMockClient()['database1']
+    )
+    assert hash(AsyncMongoMockClient()['database1']) != hash(
+        AsyncMongoMockClient()['database2']
+    )