From 28b25092ada76e62bfcadf490ad3cc7e8edd97d9 Mon Sep 17 00:00:00 2001 From: Alexey Masterov Date: Thu, 17 Jul 2025 19:49:49 +0200 Subject: [PATCH] An attempt 5 --- test_runner/random_ops/test_random_ops.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/test_runner/random_ops/test_random_ops.py b/test_runner/random_ops/test_random_ops.py index 9b4c4794cf..167ffdfc2f 100644 --- a/test_runner/random_ops/test_random_ops.py +++ b/test_runner/random_ops/test_random_ops.py @@ -522,9 +522,16 @@ class NeonProject: new_branch = NeonBranch( self, self.neon_api.get_branch_details(self.id, new_branch_def["branch"]["id"]) ) - new_branch.start_benchmark() if new_branch.connection_parameters is None: - raise RuntimeError(f"Neon branch {new_branch.id} does not have connection parameters {new_branch.endpoints}") + if not new_branch.endpoints: + for ep in self.neon_api.get_endpoints(self.id)["endpoints"]: + NeonEndpoint(self, ep) + new_branch.connection_parameters = self.connection_parameters.copy() + for ep in new_branch.endpoints.values(): + if ep.type == "read_write": + new_branch.connection_parameters["host"] = ep.host + break + with psycopg2.connect( host=new_branch.connection_parameters["host"], port=5432,