Improve PR #386

* Add more test * Check to kvmi-ok is asynchronous
2025-06-21 00:22:56 +00:00 · 2016-01-04 16:30:06 +01:00
parent fba049767a
commit d184b65800
5 changed files with 58 additions and 9 deletions
--- a/tests/utils.py
+++ b/tests/utils.py
@ -50,6 +50,11 @@ class _asyncio_patch:
        future = asyncio.Future()
        if "return_value" in self.kwargs:
            future.set_result(self.kwargs["return_value"])
+        elif "side_effect" in self.kwargs:
+            if isinstance(self.kwargs["side_effect"], Exception):
+                future.set_exception(self.kwargs["side_effect"])
+            else:
+                raise NotImplementedError
        else:
            future.set_result(True)
        return future