From d4b64b9ef76c05cd9d219d9ca662dd26bbe60f2f Mon Sep 17 00:00:00 2001 From: Joonas Koivunen Date: Thu, 27 Jul 2023 14:31:53 +0300 Subject: [PATCH] test: allow passing branch-name to endpoint_start --- test_runner/fixtures/neon_fixtures.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/test_runner/fixtures/neon_fixtures.py b/test_runner/fixtures/neon_fixtures.py index eafc061ab9..5e4553f52d 100644 --- a/test_runner/fixtures/neon_fixtures.py +++ b/test_runner/fixtures/neon_fixtures.py @@ -1504,6 +1504,7 @@ class NeonCli(AbstractNeonCli): safekeepers: Optional[List[int]] = None, tenant_id: Optional[TenantId] = None, lsn: Optional[Lsn] = None, + branch_name: Optional[str] = None, ) -> "subprocess.CompletedProcess[str]": args = [ "endpoint", @@ -1517,8 +1518,11 @@ class NeonCli(AbstractNeonCli): args.append(f"--lsn={lsn}") args.extend(["--pg-port", str(pg_port)]) args.extend(["--http-port", str(http_port)]) + if safekeepers is not None: args.extend(["--safekeepers", (",".join(map(str, safekeepers)))]) + if branch_name is not None: + args.extend(["--branch-name", branch_name]) if endpoint_id is not None: args.append(endpoint_id)