rust/neon
1
0
Fork 0
mirror of https://github.com/neondatabase/neon.git synced 2026-05-16 04:30:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

An attempt 5

Browse Source
This commit is contained in:
Alexey Masterov
2025-07-17 19:49:49 +02:00
parent b77a1fae04
commit 28b25092ad
1 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
test_runner/random_ops/test_random_ops.py
View File

@@ -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,