X-Git-Url: https://osm.etsi.org/gitweb/?a=blobdiff_plain;f=tests%2Fbase.py;h=292d04a755257dc121135bf35048ed390818eaf7;hb=95c05eaf1dcffffe16a86c80d98dcd19f4aeed45;hp=af386ea1f6d73365cd2b62e91242f50d842e8eca;hpb=8293ed139b3b3dc741d75184de46936e117d923e;p=osm%2FN2VC.git

diff --git a/tests/base.py b/tests/base.py
index af386ea..292d04a 100644
--- 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)