Added test for out of order receive and fix bug creating task

[osm/N2VC.git] / tests / base.py

diff --git a/tests/base.py b/tests/base.py
index af386ea..292d04a 100644 (file)
--- a/tests/base.py
+++ b/tests/base.py
@@ -44,3 +44,8 @@ class CleanModel():
         await self.model.disconnect()
         await self.controller.destroy_model(self.model.info.uuid)
         await self.controller.disconnect()
+
+
+class AsyncMock(mock.MagicMock):
+    async def __call__(self, *args, **kwargs):
+        return super().__call__(*args, **kwargs)