summaryrefslogtreecommitdiff
path: root/t/unit/transport/test_redis.py
diff options
context:
space:
mode:
Diffstat (limited to 't/unit/transport/test_redis.py')
-rw-r--r--t/unit/transport/test_redis.py11
1 files changed, 10 insertions, 1 deletions
diff --git a/t/unit/transport/test_redis.py b/t/unit/transport/test_redis.py
index 029f7901..9905f6ee 100644
--- a/t/unit/transport/test_redis.py
+++ b/t/unit/transport/test_redis.py
@@ -6,6 +6,7 @@ from collections import defaultdict
from itertools import count
from queue import Empty
from queue import Queue as _Queue
+from typing import TYPE_CHECKING, Optional, Type
from unittest.mock import ANY, Mock, call, patch
import pytest
@@ -16,6 +17,9 @@ from kombu.transport import virtual
from kombu.utils import eventio # patch poll
from kombu.utils.json import dumps
+if TYPE_CHECKING:
+ from types import TracebackType
+
def _redis_modules():
@@ -231,7 +235,12 @@ class Pipeline:
def __enter__(self):
return self
- def __exit__(self, *exc_info):
+ def __exit__(
+ self,
+ exc_type: Optional[Type[BaseException]],
+ exc_val: Optional[BaseException],
+ exc_tb: Optional['TracebackType']
+ ) -> None:
pass
def __getattr__(self, key):