mirror of
https://github.com/outbackdingo/patroni.git
synced 2026-01-27 18:20:05 +00:00
f77073c8e1
- get rid from sleeps - reduce retry_timeout - avoid graceful Patroni shut down while DCS is "paused", just kill Patroni and after that gracefully stop postgres - don't try to delete Pod when Patroni is killed. If K8s API is paused it takes ages The run time on my laptop is reduced from 2m to 1m28s.
53 lines
1.9 KiB
Python
53 lines
1.9 KiB
Python
import json
|
|
import time
|
|
|
|
from behave import step, then
|
|
|
|
|
|
@step('I configure and start {name:w} with a tag {tag_name:w} {tag_value:w}')
|
|
def start_patroni_with_a_name_value_tag(context, name, tag_name, tag_value):
|
|
return context.pctl.start(name, custom_config={'tags': {tag_name: tag_value}})
|
|
|
|
|
|
@then('There is a {label} with "{content}" in {name:w} data directory')
|
|
def check_label(context, label, content, name):
|
|
value = (context.pctl.read_label(name, label) or '').replace('\n', '\\n')
|
|
assert content in value, "\"{0}\" in {1} doesn't contain {2}".format(value, label, content)
|
|
|
|
|
|
@step('I create label with "{content:w}" in {name:w} data directory')
|
|
def write_label(context, content, name):
|
|
context.pctl.write_label(name, content)
|
|
|
|
|
|
@step('"{name}" key in DCS has {key:w}={value} after {time_limit:d} seconds')
|
|
def check_member(context, name, key, value, time_limit):
|
|
time_limit *= context.timeout_multiplier
|
|
max_time = time.time() + int(time_limit)
|
|
dcs_value = None
|
|
while time.time() < max_time:
|
|
try:
|
|
response = json.loads(context.dcs_ctl.query(name))
|
|
dcs_value = str(response.get(key))
|
|
if dcs_value == value:
|
|
return
|
|
except Exception:
|
|
pass
|
|
time.sleep(1)
|
|
assert False, "{0} does not have {1}={2} (found {3}) in dcs after {4} seconds".format(name, key, value,
|
|
dcs_value, time_limit)
|
|
|
|
|
|
@step('there is a non empty {key:w} key in DCS after {time_limit:d} seconds')
|
|
def check_initialize(context, key, time_limit):
|
|
time_limit *= context.timeout_multiplier
|
|
max_time = time.time() + int(time_limit)
|
|
while time.time() < max_time:
|
|
try:
|
|
if context.dcs_ctl.query(key):
|
|
return
|
|
except Exception:
|
|
pass
|
|
time.sleep(1)
|
|
assert False, "There is no {0} in dcs after {1} seconds".format(key, time_limit)
|