Enable tracemalloc

Author: gerzse

.github/workflows/integration.yaml

@@ -73,7 +73,7 @@ jobs:
         with:
           python-version: ${{ matrix.python-version }}
           cache: 'pip'
-      - name: run tests
+      - name: Run tests
         run: |
           pip install -U setuptools wheel
           pip install -r requirements.txt

tests/conftest.py

@@ -1,6 +1,7 @@
 import argparse
 import random
 import time
+import tracemalloc
 from typing import Callable, TypeVar
 from unittest import mock
 from unittest.mock import Mock
@@ -72,6 +73,16 @@ def format_usage(self):
         return " | ".join(self.option_strings)
 
 
+@pytest.fixture(scope="session", autouse=True)
+def enable_tracemalloc():
+    """
+    Enable tracemalloc while tests are being executed.
+    """
+    tracemalloc.start()
+    yield
+    tracemalloc.stop()
+
+
 def pytest_addoption(parser):
     parser.addoption(
         "--redis-url",

tests/test_connect.py

@@ -135,7 +135,7 @@ def test_tcp_ssl_version_mismatch(tcp_address):
             tcp_address,
             certfile=certfile,
             keyfile=keyfile,
-            maximum_ssl_version=ssl.PROTOCOL_TLSv1_2,
+            maximum_ssl_version=ssl.TLSVersion.TLSv1_2,
         )