|
3 | 3 | import time |
4 | 4 | import weakref |
5 | 5 | from collections import defaultdict |
6 | | -from typing import Dict, Tuple, Any, List, Optional, Union |
| 6 | +from typing import Dict, Tuple, Any, List, Optional, Union, Set, Type |
7 | 7 |
|
8 | 8 | import redis |
9 | 9 |
|
@@ -85,18 +85,32 @@ def get_server(key: str, version: VersionType, server_type: ServerType) -> "Fake |
85 | 85 |
|
86 | 86 | class FakeBaseConnectionMixin(object): |
87 | 87 | def __init__( |
88 | | - self, *args: Any, version: VersionType = (7, 0), server_type: ServerType = "redis", **kwargs: Any |
| 88 | + self, |
| 89 | + *args: Any, |
| 90 | + version: VersionType = (7, 0), |
| 91 | + server_type: ServerType = "redis", |
| 92 | + server: Optional[FakeServer] = None, |
| 93 | + client_class: Type[redis.Redis] = redis.Redis, |
| 94 | + lua_modules: Optional[Set[str]] = None, |
| 95 | + writer=None, |
| 96 | + connected: bool = True, |
| 97 | + path: Optional[str] = None, |
| 98 | + **kwargs: Any, |
89 | 99 | ) -> None: |
| 100 | + """ |
| 101 | + Initializes the class and sets up the required attributes and configurations for the server and client interaction. |
| 102 | +
|
| 103 | + """ |
90 | 104 | self.client_name: Optional[str] = None |
91 | 105 | self.server_key: str |
92 | 106 | self._sock = None |
93 | 107 | self._selector: Optional[FakeSelector] = None |
94 | | - self._server = kwargs.pop("server", None) |
95 | | - self._client_class = kwargs.pop("client_class", redis.Redis) |
96 | | - self._lua_modules = kwargs.pop("lua_modules", set()) |
97 | | - self._writer = kwargs.pop("writer", None) |
98 | | - path = kwargs.pop("path", None) |
99 | | - connected = kwargs.pop("connected", True) |
| 108 | + self._server = server |
| 109 | + self._client_class = client_class |
| 110 | + self._lua_modules = lua_modules |
| 111 | + self._writer = writer |
| 112 | + path = path |
| 113 | + connected = connected |
100 | 114 | if self._server is None: |
101 | 115 | if path: |
102 | 116 | self.server_key = path |
|
0 commit comments