feat: add ability to change mock class and sync_mock function helper

Author: ticdenis

aioddd/testing.py

@@ -1,5 +1,5 @@
 import sys
-from typing import Any, Dict, List, Optional, Union
+from typing import Any, Dict, List, Optional, Type, Union
 from unittest.mock import MagicMock, Mock, patch
 
 if sys.version_info.major == 3 and sys.version_info.minor >= 8:
@@ -25,20 +25,42 @@ def sanitize_objects(source: SanitizeObject, affected: SanitizeObject) -> Saniti
     return affected
 
 
-def mock(target: Union[str, object], attributes: Optional[List[str]] = None) -> Mock:
-    target_async_mock = AsyncMock()
+def mock(
+    target: Union[str, object],
+    attributes: Optional[List[str]] = None,
+    target_mock_type: Type[Mock] = AsyncMock,
+    attribute_mock_type: Type[Mock] = AsyncMock,
+) -> Mock:
+    target_async_mock = target_mock_type()
     if not attributes:
         patch(
             target=f'{target.__module__}.{target.__name__}' if isinstance(target, object) else target,  # type: ignore
             side_effect=target_async_mock,
         )
-        return target_async_mock  # type: ignore
+        return target_async_mock
     for attribute in attributes:
-        attribute_async_mock = AsyncMock()
+        attribute_async_mock = attribute_mock_type()
         patch.object(
             target=target,
             attribute=attribute,
             side_effect=attribute_async_mock,
         )
         target_async_mock[attribute] = attribute_async_mock
-    return target_async_mock  # type: ignore
+    return target_async_mock
+
+
+async_mock = mock
+
+
+def sync_mock(
+    target: Union[str, object],
+    attributes: Optional[List[str]] = None,
+    target_mock_type: Type[Mock] = MagicMock,
+    attribute_mock_type: Type[Mock] = MagicMock,
+) -> Mock:
+    return mock(
+        target=target,
+        attributes=attributes,
+        target_mock_type=target_mock_type,
+        attribute_mock_type=attribute_mock_type,
+    )