@@ -125,11 +125,12 @@ async def test_retry_transaction_during_unfinished_slot_migration(self, r):
125
125
slot = r .keyslot (key )
126
126
node_migrating , node_importing = _find_source_and_target_node_for_slot (r , slot )
127
127
128
- with patch .object (
129
- ClusterNode , "parse_response"
130
- ) as parse_response , patch .object (
131
- NodesManager , "_update_moved_slots"
132
- ) as manager_update_moved_slots :
128
+ with (
129
+ patch .object (ClusterNode , "parse_response" ) as parse_response ,
130
+ patch .object (
131
+ NodesManager , "_update_moved_slots"
132
+ ) as manager_update_moved_slots ,
133
+ ):
133
134
134
135
def ask_redirect_effect (connection , * args , ** options ):
135
136
if "MULTI" in args :
@@ -167,11 +168,12 @@ async def test_retry_transaction_during_slot_migration_successful(
167
168
slot = r .keyslot (key )
168
169
node_migrating , node_importing = _find_source_and_target_node_for_slot (r , slot )
169
170
170
- with patch .object (
171
- ClusterNode , "parse_response"
172
- ) as parse_response , patch .object (
173
- NodesManager , "_update_moved_slots"
174
- ) as manager_update_moved_slots :
171
+ with (
172
+ patch .object (ClusterNode , "parse_response" ) as parse_response ,
173
+ patch .object (
174
+ NodesManager , "_update_moved_slots"
175
+ ) as manager_update_moved_slots ,
176
+ ):
175
177
176
178
def ask_redirect_effect (conn , * args , ** options ):
177
179
# first call should go here, we trigger an AskError
0 commit comments