fix compat

propan/test/kafka.py

@@ -1,7 +1,12 @@
+import sys
 from datetime import datetime
 from types import MethodType
 from typing import Any, Dict, Optional
-from unittest.mock import AsyncMock
+
+if sys.version_info < (3, 8):
+    from asyncmock import AsyncMock
+else:
+    from unittest.mock import AsyncMock
 
 from aiokafka.structs import ConsumerRecord
 

propan/test/nats.py

@@ -1,6 +1,11 @@
+import sys
 from types import MethodType
 from typing import Any, Dict, Optional, Union
-from unittest.mock import AsyncMock
+
+if sys.version_info < (3, 8):
+    from asyncmock import AsyncMock
+else:
+    from unittest.mock import AsyncMock
 from uuid import uuid4
 
 from nats.aio.msg import Msg

propan/test/rabbit.py

@@ -1,8 +1,13 @@
 import re
+import sys
 from contextlib import asynccontextmanager
 from types import MethodType
 from typing import Any, Optional, Union
-from unittest.mock import AsyncMock
+
+if sys.version_info < (3, 8):
+    from asyncmock import AsyncMock
+else:
+    from unittest.mock import AsyncMock
 from uuid import uuid4
 
 import aiormq

propan/test/redis.py

@@ -1,7 +1,12 @@
 import re
+import sys
 from types import MethodType
 from typing import Any, Dict, Optional, Union
-from unittest.mock import AsyncMock
+
+if sys.version_info < (3, 8):
+    from asyncmock import AsyncMock
+else:
+    from unittest.mock import AsyncMock
 
 from typing_extensions import TypeAlias
 

propan/test/sqs.py

@@ -1,8 +1,13 @@
 import json
+import sys
 from hashlib import md5
 from types import MethodType
 from typing import Any, Dict, Optional
-from unittest.mock import AsyncMock
+
+if sys.version_info < (3, 8):
+    from asyncmock import AsyncMock
+else:
+    from unittest.mock import AsyncMock
 from uuid import uuid4
 
 from propan import SQSBroker