fixed llapi+gears

Author: DvirDukhan

tests/flow/tests_gears_llapi.py

@@ -7,7 +7,7 @@
 
 
 def verify_gears_loaded(env):
-    con = env.getConnection()
+    con = get_connection(env, '{1}')
     modules = con.execute_command("MODULE", "LIST")
     if b'rg' in [module[1] for module in modules]:
         return True
@@ -416,7 +416,7 @@ def FlattenTensor(record):
 GB("CommandReader").map(FlattenTensor).register(trigger="FlattenTensor_test")
     '''
 
-    con = env.getConnection()
+    con = get_connection(env, '{1}')
     ret = con.execute_command('rg.pyexecute', script)
     env.assertEqual(ret, b'OK')
     ret = con.execute_command('rg.trigger', 'FlattenTensor_test')

tests/flow/tests_llapi.py

@@ -11,7 +11,7 @@
 def with_test_module(f):
     @wraps(f)
     def wrapper(env, *args, **kwargs):
-        con = env.getConnection()
+        con = get_connection(env, '{1}')
         modules = con.execute_command("MODULE", "LIST")
         if b'RAI_llapi' in [module[1] for module in modules]:
             return f(env, *args, **kwargs)
@@ -29,7 +29,7 @@ def wrapper(env, *args, **kwargs):
 @with_test_module
 def test_basic_check(env):
 
-    con = env.getConnection()
+    con = get_connection(env, '{1}')
     ret = con.execute_command("RAI_llapi.basic_check")
     env.assertEqual(ret, b'OK')