Blob Blame History Raw
From 4a80b454a164278a4751c761b64305a325abc5f0 Mon Sep 17 00:00:00 2001
From: xtreak <tir.karthi@gmail.com>
Date: Thu, 27 Jun 2019 09:52:26 +0000
Subject: [PATCH] Use MagicMock to make sure create_server doesn't use
 AsyncMock on Python 3.8 and above

---
 aiosmtpd/tests/test_main.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/aiosmtpd/tests/test_main.py b/aiosmtpd/tests/test_main.py
index a821c6a..c61b05c 100644
--- a/aiosmtpd/tests/test_main.py
+++ b/aiosmtpd/tests/test_main.py
@@ -10,7 +10,7 @@
 from contextlib import ExitStack
 from functools import partial
 from io import StringIO
-from unittest.mock import patch
+from unittest.mock import patch, MagicMock
 
 try:
     import pwd
@@ -150,7 +150,7 @@ def setUp(self):
         pfunc = partial(patch.object, self.loop)
         resources = ExitStack()
         self.addCleanup(resources.close)
-        self.create_server = resources.enter_context(pfunc('create_server'))
+        self.create_server = resources.enter_context(pfunc('create_server', MagicMock()))
         self.run_until_complete = resources.enter_context(
             pfunc('run_until_complete'))
         self.add_signal_handler = resources.enter_context(