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

another things

Browse Source
This commit is contained in:
Aleksandr Sarantsev
2025-06-11 09:19:07 +04:00
parent 652c7203b5
commit 53bdbdf71f
3 changed files with 80 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
test_runner/fixtures/neon_fixtures.py
View File

@@ -2062,11 +2062,16 @@ class NeonStorageController(MetricsGetter, LogUtils):
headers=self.headers(TokenScope.ADMIN),
)
def node_drain(self, node_id):
log.info(f"node_drain({node_id})")
def node_drain(self, node_id: int, drain_all: bool | None = None):
log.info(f"node_drain({node_id}, drain_all={drain_all})")
url = f"{self.api}/control/v1/node/{node_id}/drain"
if drain_all is not None:
url += f"?drain_all={str(drain_all).lower()}"
self.request(
"PUT",
f"{self.api}/control/v1/node/{node_id}/drain",
url,
headers=self.headers(TokenScope.INFRA),
)