Use AsyncMock

modoboa/policyd/tests.py

@@ -1,7 +1,8 @@
 """Policy daemon related tests."""
 
 import asyncio
-from mock import patch
+from aiosmtplib import send
+from unittest.mock import AsyncMock
 from multiprocessing import Process
 import socket
 import time
@@ -56,8 +57,7 @@ class PolicyDaemonTestCase(
         self.admin.role = "SuperAdmins"
 
         db.connections.close_all()
-        patcher = patch("aiosmtplib.send")
-        self.send_mock = patcher.start()
+        self.send_mock = AsyncMock(send)
         self.process = Process(target=start_policy_daemon)
         self.process.daemon = True
         self.process.start()