File tree Expand file tree Collapse file tree 2 files changed +7
-5
lines changed Expand file tree Collapse file tree 2 files changed +7
-5
lines changed Original file line number Diff line number Diff line change @@ -323,14 +323,13 @@ async def initialize(self) -> "RedisCluster":
323
323
if self ._initialize :
324
324
async with self ._lock :
325
325
if self ._initialize :
326
- self ._initialize = False
327
326
try :
328
327
await self .nodes_manager .initialize ()
329
328
await self .commands_parser .initialize (
330
329
self .nodes_manager .default_node
331
330
)
331
+ self ._initialize = False
332
332
except BaseException :
333
- self ._initialize = True
334
333
await self .nodes_manager .close ()
335
334
await self .nodes_manager .close ("startup_nodes" )
336
335
raise
@@ -343,6 +342,7 @@ async def close(self) -> None:
343
342
if not self ._initialize :
344
343
self ._initialize = True
345
344
await self .nodes_manager .close ()
345
+ await self .nodes_manager .close ("startup_nodes" )
346
346
347
347
async def __aenter__ (self ) -> "RedisCluster" :
348
348
return await self .initialize ()
Original file line number Diff line number Diff line change @@ -680,13 +680,15 @@ async def test_not_require_full_coverage_cluster_down_error(
680
680
else :
681
681
raise e
682
682
683
- async def test_can_run_concurrent_commands (self , r : RedisCluster ) -> None :
684
- assert await r .ping (target_nodes = RedisCluster .ALL_NODES ) is True
683
+ async def test_can_run_concurrent_commands (self , request : FixtureRequest ) -> None :
684
+ url = request .config .getoption ("--redis-url" )
685
+ rc = RedisCluster .from_url (url )
685
686
assert all (
686
687
await asyncio .gather (
687
- * (r . ping ( target_nodes = RedisCluster .ALL_NODES ) for _ in range (100 ))
688
+ * (rc . echo ( "i" , target_nodes = RedisCluster .ALL_NODES ) for i in range (100 ))
688
689
)
689
690
)
691
+ await rc .close ()
690
692
691
693
692
694
@pytest .mark .onlycluster
You can’t perform that action at this time.
0 commit comments