tests: fix import eqmock

cenobites · Oct 27, 2024 · 379d2c7 · 379d2c7
1 parent 75d26e0
commit 379d2c7
Show file tree

Hide file tree

Showing 2 changed files with 50 additions and 0 deletions.
diff --git a/examples/minimal-async/tests/utils.py b/examples/minimal-async/tests/utils.py
@@ -0,0 +1,25 @@
+from typing import Any
+
+
+class EqMock:
+    value: Any = None
+
+    def __init__(self, remember: bool = False) -> None:
+        self.remember: bool = remember
+
+    def __eq__(self, other: Any) -> bool:  # noqa: ANN401
+        if self.remember and self.value is not None:
+            return self.value == other
+        else:
+            assert other, other
+
+            if self.remember:
+                self.value = other
+
+        return True
+
+    def __repr__(self) -> str:
+        return repr(self.value) if self.remember else super().__repr__()
+
+    def __str__(self) -> str:
+        return str(self.value) if self.remember else super().__str__()
diff --git a/examples/minimal/tests/utils.py b/examples/minimal/tests/utils.py
@@ -0,0 +1,25 @@
+from typing import Any
+
+
+class EqMock:
+    value: Any = None
+
+    def __init__(self, remember: bool = False) -> None:
+        self.remember: bool = remember
+
+    def __eq__(self, other: Any) -> bool:  # noqa: ANN401
+        if self.remember and self.value is not None:
+            return self.value == other
+        else:
+            assert other, other
+
+            if self.remember:
+                self.value = other
+
+        return True
+
+    def __repr__(self) -> str:
+        return repr(self.value) if self.remember else super().__repr__()
+
+    def __str__(self) -> str:
+        return str(self.value) if self.remember else super().__str__()