@@ -120,7 +120,7 @@ def cmd_init_mock(self, r: ClusterNode) -> None:
120
120
def mock_node_resp (node : ClusterNode , response : Any ) -> ClusterNode :
121
121
connection = mock .AsyncMock ()
122
122
connection .is_connected = True
123
- connection .read_response_without_lock .return_value = response
123
+ connection .read_response .return_value = response
124
124
while node ._free :
125
125
node ._free .pop ()
126
126
node ._free .append (connection )
@@ -130,7 +130,7 @@ def mock_node_resp(node: ClusterNode, response: Any) -> ClusterNode:
130
130
def mock_node_resp_exc (node : ClusterNode , exc : Exception ) -> ClusterNode :
131
131
connection = mock .AsyncMock ()
132
132
connection .is_connected = True
133
- connection .read_response_without_lock .side_effect = exc
133
+ connection .read_response .side_effect = exc
134
134
while node ._free :
135
135
node ._free .pop ()
136
136
node ._free .append (connection )
@@ -316,10 +316,10 @@ async def test_execute_command_node_flag_primaries(self, r: RedisCluster) -> Non
316
316
assert await r .ping (target_nodes = RedisCluster .PRIMARIES ) is True
317
317
for primary in primaries :
318
318
conn = primary ._free .pop ()
319
- assert conn .read_response_without_lock .called is True
319
+ assert conn .read_response .called is True
320
320
for replica in replicas :
321
321
conn = replica ._free .pop ()
322
- assert conn .read_response_without_lock .called is not True
322
+ assert conn .read_response .called is not True
323
323
324
324
async def test_execute_command_node_flag_replicas (self , r : RedisCluster ) -> None :
325
325
"""
@@ -333,10 +333,10 @@ async def test_execute_command_node_flag_replicas(self, r: RedisCluster) -> None
333
333
assert await r .ping (target_nodes = RedisCluster .REPLICAS ) is True
334
334
for replica in replicas :
335
335
conn = replica ._free .pop ()
336
- assert conn .read_response_without_lock .called is True
336
+ assert conn .read_response .called is True
337
337
for primary in primaries :
338
338
conn = primary ._free .pop ()
339
- assert conn .read_response_without_lock .called is not True
339
+ assert conn .read_response .called is not True
340
340
341
341
await r .close ()
342
342
@@ -348,7 +348,7 @@ async def test_execute_command_node_flag_all_nodes(self, r: RedisCluster) -> Non
348
348
assert await r .ping (target_nodes = RedisCluster .ALL_NODES ) is True
349
349
for node in r .get_nodes ():
350
350
conn = node ._free .pop ()
351
- assert conn .read_response_without_lock .called is True
351
+ assert conn .read_response .called is True
352
352
353
353
async def test_execute_command_node_flag_random (self , r : RedisCluster ) -> None :
354
354
"""
@@ -359,7 +359,7 @@ async def test_execute_command_node_flag_random(self, r: RedisCluster) -> None:
359
359
called_count = 0
360
360
for node in r .get_nodes ():
361
361
conn = node ._free .pop ()
362
- if conn .read_response_without_lock .called is True :
362
+ if conn .read_response .called is True :
363
363
called_count += 1
364
364
assert called_count == 1
365
365
@@ -372,7 +372,7 @@ async def test_execute_command_default_node(self, r: RedisCluster) -> None:
372
372
mock_node_resp (def_node , "PONG" )
373
373
assert await r .ping () is True
374
374
conn = def_node ._free .pop ()
375
- assert conn .read_response_without_lock .called
375
+ assert conn .read_response .called
376
376
377
377
async def test_ask_redirection (self , r : RedisCluster ) -> None :
378
378
"""
@@ -516,7 +516,7 @@ async def test_reading_from_replicas_in_round_robin(self) -> None:
516
516
with mock .patch .multiple (
517
517
Connection ,
518
518
send_command = mock .DEFAULT ,
519
- read_response_without_lock = mock .DEFAULT ,
519
+ read_response = mock .DEFAULT ,
520
520
_connect = mock .DEFAULT ,
521
521
can_read = mock .DEFAULT ,
522
522
on_connect = mock .DEFAULT ,
@@ -548,7 +548,7 @@ def execute_command_mock_third(self, *args, **options):
548
548
# so we'll mock some of the Connection's functions to allow it
549
549
execute_command .side_effect = execute_command_mock_first
550
550
mocks ["send_command" ].return_value = True
551
- mocks ["read_response_without_lock " ].return_value = "OK"
551
+ mocks ["read_response " ].return_value = "OK"
552
552
mocks ["_connect" ].return_value = True
553
553
mocks ["can_read" ].return_value = False
554
554
mocks ["on_connect" ].return_value = True
@@ -857,8 +857,8 @@ async def test_cluster_delslots(self) -> None:
857
857
node0 = r .get_node (default_host , 7000 )
858
858
node1 = r .get_node (default_host , 7001 )
859
859
assert await r .cluster_delslots (0 , 8192 ) == [True , True ]
860
- assert node0 ._free .pop ().read_response_without_lock .called
861
- assert node1 ._free .pop ().read_response_without_lock .called
860
+ assert node0 ._free .pop ().read_response .called
861
+ assert node1 ._free .pop ().read_response .called
862
862
863
863
await r .close ()
864
864
@@ -1027,7 +1027,7 @@ async def test_cluster_setslot_stable(self, r: RedisCluster) -> None:
1027
1027
node = r .nodes_manager .get_node_from_slot (12182 )
1028
1028
mock_node_resp (node , "OK" )
1029
1029
assert await r .cluster_setslot_stable (12182 ) is True
1030
- assert node ._free .pop ().read_response_without_lock .called
1030
+ assert node ._free .pop ().read_response .called
1031
1031
1032
1032
@skip_if_redis_enterprise ()
1033
1033
async def test_cluster_replicas (self , r : RedisCluster ) -> None :
@@ -1069,7 +1069,7 @@ async def test_readonly(self) -> None:
1069
1069
for res in all_replicas_results .values ():
1070
1070
assert res is True
1071
1071
for replica in r .get_replicas ():
1072
- assert replica ._free .pop ().read_response_without_lock .called
1072
+ assert replica ._free .pop ().read_response .called
1073
1073
1074
1074
await r .close ()
1075
1075
@@ -1082,7 +1082,7 @@ async def test_readwrite(self) -> None:
1082
1082
for res in all_replicas_results .values ():
1083
1083
assert res is True
1084
1084
for replica in r .get_replicas ():
1085
- assert replica ._free .pop ().read_response_without_lock .called
1085
+ assert replica ._free .pop ().read_response .called
1086
1086
1087
1087
await r .close ()
1088
1088
@@ -2441,8 +2441,8 @@ async def test_asking_error(self, r: RedisCluster) -> None:
2441
2441
mock_node_resp_exc (first_node , AskError (ask_msg ))
2442
2442
mock_node_resp (ask_node , "MOCK_OK" )
2443
2443
res = await pipe .get (key ).execute ()
2444
- assert first_node ._free .pop ().read_response_without_lock .await_count
2445
- assert ask_node ._free .pop ().read_response_without_lock .await_count
2444
+ assert first_node ._free .pop ().read_response .await_count
2445
+ assert ask_node ._free .pop ().read_response .await_count
2446
2446
assert res == ["MOCK_OK" ]
2447
2447
2448
2448
async def test_moved_redirection_on_slave_with_default (
@@ -2497,7 +2497,7 @@ async def test_readonly_pipeline_from_readonly_client(
2497
2497
executed_on_replica = False
2498
2498
for node in slot_nodes :
2499
2499
if node .server_type == REPLICA :
2500
- if node ._free .pop ().read_response_without_lock .await_count :
2500
+ if node ._free .pop ().read_response .await_count :
2501
2501
executed_on_replica = True
2502
2502
break
2503
2503
assert executed_on_replica
0 commit comments