mirrors/scylladb
1
0
Fork 0
mirror of https://github.com/scylladb/scylladb.git synced 2026-05-31 20:16:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

test: manager_client: make con_gen for ManagerClient.__init__ nonoptional

Browse Source 
`ManagerClient` is given a function that is used to create CQL
connections to the Scylla cluster. For some reason it was typed as
`Optional` even though it was never passed `None`. Fix it.
This commit is contained in:
Kamil Braun
2023-07-12 11:38:06 +02:00
parent b4dc3f7cd9
commit 3464877276
1 changed files with 6 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
test/pylib/manager_client.py
View File

@@ -33,7 +33,7 @@ class ManagerClient():
# pylint: disable=too-many-public-methods
def __init__(self, sock_path: str, port: int, use_ssl: bool,
con_gen: Optional[Callable[[List[IPAddress], int, bool], CassandraSession]]) \
con_gen: Callable[[List[IPAddress], int, bool], CassandraSession]) \
-> None:
self.port = port
self.use_ssl = use_ssl
@@ -50,12 +50,11 @@ class ManagerClient():
async def driver_connect(self, server: Optional[ServerInfo] = None) -> None:
"""Connect to cluster"""
if self.con_gen is not None:
targets = [server] if server else await self.running_servers()
servers = [s_info.ip_addr for s_info in targets]
logger.debug("driver connecting to %s", servers)
self.ccluster = self.con_gen(servers, self.port, self.use_ssl)
self.cql = self.ccluster.connect()
targets = [server] if server else await self.running_servers()
servers = [s_info.ip_addr for s_info in targets]
logger.debug("driver connecting to %s", servers)
self.ccluster = self.con_gen(servers, self.port, self.use_ssl)
self.cql = self.ccluster.connect()
def driver_close(self) -> None:
"""Disconnect from cluster"""