X-Git-Url: https://osm.etsi.org/gitweb/?a=blobdiff_plain;f=tests%2Fbase.py;h=292d04a755257dc121135bf35048ed390818eaf7;hb=6637bf37c99d012ccd51823501dd7325ba3d6840;hp=af386ea1f6d73365cd2b62e91242f50d842e8eca;hpb=a09dd5038d226fbcda54a79bdddc995054857198;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)