From bb920f9474266d2d2a0d583ef99d00ac1995c40e Mon Sep 17 00:00:00 2001
From: ItsDrike <itsdrike@protonmail.com>
Date: Sun, 30 Jun 2024 23:02:19 +0200
Subject: [PATCH] Add test helpers.py

---
 tests/helpers.py | 105 +++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 105 insertions(+)
 create mode 100644 tests/helpers.py

diff --git a/tests/helpers.py b/tests/helpers.py
new file mode 100644
index 0000000..d21d6c6
--- /dev/null
+++ b/tests/helpers.py
@@ -0,0 +1,105 @@
+from __future__ import annotations
+
+import asyncio
+import unittest.mock
+from typing import Any, Generic, TYPE_CHECKING, TypeVar
+
+from typing_extensions import ParamSpec
+
+if TYPE_CHECKING:
+    from collections.abc import Callable, Coroutine
+
+T = TypeVar("T")
+P = ParamSpec("P")
+T_Mock = TypeVar("T_Mock", bound=unittest.mock.Mock)
+
+__all__ = [
+    "synchronize",
+    "UnpropagatingMockMixin",
+    "CustomMockMixin",
+]
+
+
+def synchronize(f: Callable[P, Coroutine[Any, Any, T]]) -> Callable[P, T]:
+    """Take an asynchronous function, and return a synchronous alternative.
+
+    This is needed because we sometimes want to test asynchronous behavior in a synchronous test function,
+    where we can't simply await something. This function uses `asyncio.run` and generates a wrapper
+    around the original asynchronous function, that awaits the result in a blocking synchronous way,
+    returning the obtained value.
+    """
+
+    def wrapper(*args: P.args, **kwargs: P.kwargs) -> T:
+        return asyncio.run(f(*args, **kwargs))
+
+    return wrapper
+
+
+class UnpropagatingMockMixin(Generic[T_Mock]):
+    """Provides common functionality for our :class:`~unittest.mock.Mock` classes.
+
+    By default, mock objects propagate themselves by returning a new instance of the same mock
+    class, with same initialization attributes. This is done whenever we're accessing new
+    attributes that mock class.
+
+    This propagation makes sense for simple mocks without any additional restrictions, however when
+    dealing with limited mocks to some ``spec_set``, it doesn't usually make sense to propagate
+    those same ``spec_set`` restrictions, since we generally don't have attributes/methods of a
+    class be of/return the same class.
+
+    This mixin class stops this propagation, and instead returns instances of specified mock class,
+    defined in :attr:`.child_mock_type` class variable, which is by default set to
+    :class:`~unittest.mock.MagicMock`, as it can safely represent most objects.
+
+    .. note:
+        This propagation handling will only be done for the mock classes that inherited from this
+        mixin class. That means if the :attr:`.child_mock_type` is one of the regular mock classes,
+        and the mock is propagated, a regular mock class is returned as that new attribute. This
+        regular class then won't have the same overrides, and will therefore propagate itself, like
+        any other mock class would.
+
+        If you wish to counteract this, you can set the :attr:`.child_mock_type` to a mock class
+        that also inherits from this mixin class, perhaps to your class itself, overriding any
+        propagation recursively.
+    """
+
+    child_mock_type: T_Mock = unittest.mock.MagicMock
+
+    # Since this is a mixin class, we can access some attributes defined in mock classes safely.
+    # Define the types of these variables here, for proper static type analysis.
+    _mock_sealed: bool
+    _extract_mock_name: Callable[[], str]
+
+    def _get_child_mock(self, **kwargs) -> T_Mock:
+        """Make :attr:`.child_mock_type`` instances instead of instances of the same class.
+
+        By default, this method creates a new mock instance of the same original class, and passes
+        over the same initialization arguments. This overrides that behavior to instead create an
+        instance of :attr:`.child_mock_type` class.
+        """
+        # Mocks can be sealed, in which case we wouldn't want to allow propagation of any kind
+        # and rather raise an AttributeError, informing that given attr isn't accessible
+        if self._mock_sealed:
+            mock_name = self._extract_mock_name()
+            obj_name = f"{mock_name}.{kwargs['name']}" if "name" in kwargs else f"{mock_name}()"
+            raise AttributeError(f"Can't access {obj_name}, mock is sealed.")
+
+        # Propagate any other children as simple `unittest.mock.Mock` instances
+        # rather than `self.__class__` instances
+        return self.child_mock_type(**kwargs)
+
+
+class CustomMockMixin(UnpropagatingMockMixin):
+    """Provides common functionality for our custom mock types.
+
+    * Stops propagation of same ``spec_set`` restricted mock in child mocks
+      (see :class:`.UnpropagatingMockMixin` for more info)
+    * Allows using the ``spec_set`` attribute as class attribute
+    """
+
+    spec_set = None
+
+    def __init__(self, **kwargs):
+        if "spec_set" in kwargs:
+            self.spec_set = kwargs.pop("spec_set")
+        super().__init__(spec_set=self.spec_set, **kwargs)  # type: ignore # Mixin class, this __init__ is valid