adapt changes to new redis-py errors

Author: DvirDukhan

tests/flow/includes.py

@@ -202,7 +202,7 @@ def run_test_multiproc(env, routing_hint, n_procs, fn, args=tuple()):
     procs = []
 
     def tmpfn():
-        con = env.getConnectionByKey(routing_hint, None)
+        con = env.getConnectionByKey(routing_hint, 'SET')
         fn(con, *args)
         return 1
 
@@ -222,26 +222,26 @@ def load_file_content(file_name):
         return f.read()
 
 
-def check_error_message(env, con, error_msg, *command, error_msg_is_substr=False):
+def check_error_message(env, con, error_msg, *command, error_msg_is_substr=False, error_type=redis.exceptions.ResponseError):
     try:
         con.execute_command(*command)
         env.assertFalse(True, message=get_caller_pos())
     except Exception as exception:
-        env.assertEqual(type(exception), redis.exceptions.ResponseError, message=get_caller_pos())
+        env.assertEqual(type(exception), error_type, message=get_caller_pos())
         if error_msg_is_substr:
             # We only verify that the given error_msg is a substring of the entire error message.
             env.assertTrue(str(exception).find(error_msg) >= 0, message=get_caller_pos())
         else:
             env.assertEqual(error_msg, str(exception), message=get_caller_pos())
 
 
-def check_error(env, con, *command):
+def check_error(env, con, *command, error_type=redis.exceptions.ResponseError):
     try:
         con.execute_command(*command)
         env.assertFalse(True, message=get_caller_pos())
     except Exception as e:
         exception = e
-        env.assertEqual(type(exception), redis.exceptions.ResponseError, message=get_caller_pos())
+        env.assertTrue(issubclass(type(exception), error_type), message=get_caller_pos())
 
 
 # Returns a dict with all the fields of a certain section from INFO MODULES command

tests/flow/tests_commands.py

@@ -147,10 +147,10 @@ def test_modelexecute_errors(env):
 
     # The following 2 commands should raise an error on cluster mode (keys are not on the same shard)
     if env.isCluster():
-        check_error_message(env, con, "CROSSSLOT Keys in request don't hash to the same slot",
-                            'AI.MODELEXECUTE', 'm{1}', 'INPUTS', 2, 'a{1}', 'b', 'OUTPUTS', 1, 'c{1}')
-        check_error_message(env, con, "CROSSSLOT Keys in request don't hash to the same slot",
-                            'AI.MODELEXECUTE', 'm{1}', 'INPUTS', 2, 'a{1}', 'b{1}', 'OUTPUTS', 1, 'c')
+        check_error_message(env, con, "Keys in request don't hash to the same slot",
+                            'AI.MODELEXECUTE', 'm{1}', 'INPUTS', 2, 'a{1}', 'b', 'OUTPUTS', 1, 'c{1}', error_type=redis.exceptions.ClusterCrossSlotError)
+        check_error_message(env, con, "Keys in request don't hash to the same slot",
+                            'AI.MODELEXECUTE', 'm{1}', 'INPUTS', 2, 'a{1}', 'b{1}', 'OUTPUTS', 1, 'c', error_type=redis.exceptions.ClusterCrossSlotError)
 
 
 def test_keys_syntax(env):
@@ -319,10 +319,10 @@ def test_pytorch_scriptexecute_errors(env):
 
     if env.isCluster():
         # cross shard
-        check_error_message(env, con, "CROSSSLOT Keys in request don't hash to the same slot", 'AI.SCRIPTEXECUTE', 'ket{1}', 'bar', 'KEYS', 1 , '{2}', 'INPUTS', 2, 'a{1}', 'b{1}', 'OUTPUTS', 1, 'c{1}')
+        check_error_message(env, con, "Keys in request don't hash to the same slot", 'AI.SCRIPTEXECUTE', 'ket{1}', 'bar', 'KEYS', 1 , '{2}', 'INPUTS', 2, 'a{1}', 'b{1}', 'OUTPUTS', 1, 'c{1}', error_type=redis.exceptions.ClusterCrossSlotError)
 
         # key doesn't exist
-        check_error_message(env, con, "CROSSSLOT Keys in request don't hash to the same slot", 'AI.SCRIPTEXECUTE', 'ket{1}', 'bar', 'KEYS', 1 , '{1}', 'INPUTS', 2, 'a{1}', 'b{2}', 'OUTPUTS', 1, 'c{1}')
+        check_error_message(env, con, "Keys in request don't hash to the same slot", 'AI.SCRIPTEXECUTE', 'ket{1}', 'bar', 'KEYS', 1 , '{1}', 'INPUTS', 2, 'a{1}', 'b{2}', 'OUTPUTS', 1, 'c{1}', error_type=redis.exceptions.ClusterCrossSlotError)
 
 
 def test_pytorch_scriptexecute_variadic_errors(env):

tests/flow/tests_dag_errors.py

@@ -213,7 +213,7 @@ def test_dag_crossslot_violation_errors(env):
             'PERSIST', '1', 'resultTensor:{2}', '|>',
             'AI.TENSORSET', 'resultTensor:{2}', 'FLOAT', 1, 30,
         )
-        check_error_message(env, con, "CROSSSLOT Keys in request don't hash to the same slot", *command)
+        check_error_message(env, con, "Keys in request don't hash to the same slot", *command, error_type=redis.exceptions.ClusterCrossSlotError)
 
         # ERR CROSSSLOT violation (model key has a different hash tag than the LOAD and PERSIST tensors)
         command = (
@@ -224,7 +224,7 @@ def test_dag_crossslot_violation_errors(env):
             'INPUTS', 1, 'transactionTensor:{1}',
             'OUTPUTS', 1, 'resultTensor:{1}',
         )
-        check_error_message(env, con, "CROSSSLOT Keys in request don't hash to the same slot", *command)
+        check_error_message(env, con, "Keys in request don't hash to the same slot", *command, error_type=redis.exceptions.ClusterCrossSlotError)
 
         command = (
             'AI.DAGEXECUTE', 'LOAD', '1', 'referenceTensor:{1}',
@@ -235,7 +235,7 @@ def test_dag_crossslot_violation_errors(env):
             'INPUTS', 1, 'transactionTensor:{1}',
             'OUTPUTS', 1, 'resultTensor:{1}',
         )
-        check_error_message(env, con, "CROSSSLOT Keys in request don't hash to the same slot", *command)
+        check_error_message(env, con, "Keys in request don't hash to the same slot", *command, error_type=redis.exceptions.ClusterCrossSlotError)
 
 
 def test_dag_tensorget_tensorset_errors(env):