added print

FalkorDB · Sep 26, 2024 · fb38ae0 · fb38ae0
1 parent 6f7730b
commit fb38ae0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/omnistrate_tests/classes/omnistrate_fleet_instance.py b/omnistrate_tests/classes/omnistrate_fleet_instance.py
@@ -528,7 +528,7 @@ def create_connection(
                     cluster_error_retry_attempts=20,
                     retry=retry.Retry(
                         retries=20,
-                        backoff=backoff.ExponentialBackoff(base=3),
+                        backoff=backoff.ExponentialBackoff(base=5),
                         supported_errors=(
                             ConnectionRefusedError,
                             ConnectionError,

diff --git a/omnistrate_tests/test_update_memory.py b/omnistrate_tests/test_update_memory.py
@@ -203,15 +203,14 @@ def test_zero_downtime(
             # Write some data to the DB
             try:
                 graph.query("CREATE (n:Person {name: 'Alice'})")
-            except:
+                graph.ro_query("MATCH (n:Person {name: 'Alice'}) RETURN n")
+            except (ConnectionError, ReadOnlyError) as e:
                 logging.info("THE CREATE COMMAND FAILED")
                 print("THE CREATE COMMAND FAILED")
                 db.connection.close()
                 db = instance.create_connection(ssl=ssl, force_reconnect=True)
                 graph = db.select_graph("test")
                 continue
-            graph.ro_query("MATCH (n:Person {name: 'Alice'}) RETURN n")
-
             time.sleep(3)
     except Exception as e:
         logging.exception(e)