mirror of
https://github.com/outbackdingo/patroni.git
synced 2026-01-27 18:20:05 +00:00
183adb7848
* Implement proper tests for `multiprocessing.set_start_method()` * Exclude some watchdog code from coverage (it is used only for behave tests) * properly use os.path.join for windows compatibility * import DCS modules in `features/environment.py` on demand. It allows to run behave tests against chosen DCS without installing all dependencies. * remove some unused behave code * fix some minor issues in the dcs.kubernetes module
50 lines
1.4 KiB
Python
50 lines
1.4 KiB
Python
from behave import step, then
|
|
import time
|
|
|
|
|
|
def polling_loop(timeout, interval=1):
|
|
"""Returns an iterator that returns values until timeout has passed. Timeout is measured from start of iteration."""
|
|
start_time = time.time()
|
|
iteration = 0
|
|
end_time = start_time + timeout
|
|
while time.time() < end_time:
|
|
yield iteration
|
|
iteration += 1
|
|
time.sleep(interval)
|
|
|
|
|
|
@step('I start {name:w} with watchdog')
|
|
def start_patroni_with_watchdog(context, name):
|
|
return context.pctl.start(name, custom_config={'watchdog': True})
|
|
|
|
|
|
@step('{name:w} watchdog has been pinged after {timeout:d} seconds')
|
|
def watchdog_was_pinged(context, name, timeout):
|
|
for _ in polling_loop(timeout):
|
|
if context.pctl.get_watchdog(name).was_pinged:
|
|
return True
|
|
return False
|
|
|
|
|
|
@then('{name:w} watchdog has been closed')
|
|
def watchdog_was_closed(context, name):
|
|
assert context.pctl.get_watchdog(name).was_closed
|
|
|
|
|
|
@step('I reset {name:w} watchdog state')
|
|
def watchdog_reset_pinged(context, name):
|
|
context.pctl.get_watchdog(name).reset()
|
|
|
|
|
|
@then('{name:w} watchdog is triggered after {timeout:d} seconds')
|
|
def watchdog_was_triggered(context, name, timeout):
|
|
for _ in polling_loop(timeout):
|
|
if context.pctl.get_watchdog(name).was_triggered:
|
|
return True
|
|
assert False
|
|
|
|
|
|
@step('{name:w} hangs for {timeout:d} seconds')
|
|
def patroni_hang(context, name, timeout):
|
|
return context.pctl.patroni_hang(name, timeout)
|