Retry migration test on ConnectionError

LLiuJJ · Jan 18, 2023 · 8da3625 · 8da3625
1 parent 5e0fb10
commit 8da3625
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/integration/test_migrate.py b/tests/integration/test_migrate.py
@@ -37,7 +37,8 @@ def migrate_slots(cluster, slots):
  if cursor == 0:
  break
 
- except (redis.ResponseError, redis.TimeoutError) as e:
+ except (redis.ConnectionError, redis.ResponseError,
+ redis.TimeoutError) as e:
  logging.info('raft.scan command failed with error: ' + str(e))
 
  max_retries -= 1