From be287648d05a16f5684a2f462ee68ad85218517e Mon Sep 17 00:00:00 2001
From: M1nd3r <petrsedlacek.km@seznam.cz>
Date: Thu, 3 Jul 2025 18:22:09 +0200
Subject: [PATCH] chore(core, tests): add MockAsync and await_result_patched
 [no changelog]

---
 core/tests/mock.py | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/core/tests/mock.py b/core/tests/mock.py
index f5dcb7d484..810d2c8a38 100644
--- a/core/tests/mock.py
+++ b/core/tests/mock.py
@@ -24,6 +24,19 @@ class Mock:
         return self.return_value
 
 
+class MockAsync(Mock):
+    def __call__(self, *args, **kwargs) -> Any:
+        self.calls.append((args, kwargs))
+
+        if self.raises is not None:
+            raise self.raises
+
+        async def _async_return() -> Any:
+            return self.return_value
+
+        return _async_return()
+
+
 class patch:
     MOCK_OBJECT = object()
     NO_VALUE = object()