mirror of
https://github.com/outbackdingo/patroni.git
synced 2026-08-25 14:53:37 +00:00
Oleksii Kliukin
fa1a7687e5
Make sure the step definition does not include "command" worlds. Use the table name that includes current timestamp in the tests.
91 lines
3.3 KiB
Python
91 lines
3.3 KiB
Python
from lettuce import world, steps
|
|
import time
|
|
import requests
|
|
|
|
|
|
@steps
|
|
class PatroniAPISteps(object):
|
|
|
|
def __init__(self, environ):
|
|
self.env = environ
|
|
self.response = None
|
|
self.status_code = None
|
|
|
|
# there is no way we can find out if the node has already
|
|
# started as a leader without checking the DCS. We cannot
|
|
# just rely on the database availability, since there is
|
|
# a short gap between the time PostgreSQL becomes available
|
|
# and Patroni assuming the leader role.
|
|
@staticmethod
|
|
def is_a_leader(step, name, time_limit):
|
|
'''(\w+) is a leader after (\d+) seconds'''
|
|
max_time = time.time() + int(time_limit)
|
|
while (world.etcd_ctl.query("leader") != name):
|
|
time.sleep(1)
|
|
if time.time() > max_time:
|
|
assert False, "{0} is not a leader in etcd after {1} seconds".format(name, time_limit)
|
|
|
|
@staticmethod
|
|
def sleep_for_n_seconds(step, value):
|
|
'''I sleep for (\d+) seconds'''
|
|
time.sleep(int(value))
|
|
|
|
def do_get(self, step, url):
|
|
'''I issue a GET request to (https?://(?:\w|\.|:|/)+)'''
|
|
try:
|
|
r = requests.get(url)
|
|
except requests.exceptions.RequestException:
|
|
self.code = None
|
|
self.response = None
|
|
else:
|
|
self.status_code = r.status_code
|
|
try:
|
|
self.response = r.json()
|
|
except ValueError:
|
|
self.response = r.content
|
|
|
|
def do_post_empty(self, step, url):
|
|
'''I issue an empty POST request to (https?://(?:\w|\.|:|/)+)'''
|
|
self.do_post(step, url, None)
|
|
|
|
def do_post(self, step, url, data):
|
|
'''I issue a POST request to (https?://(?:\w|\.|:|/)+) with ((?:\s*\w+\s*=\s*\w+\s*,?)+)'''
|
|
post_data = {}
|
|
if data:
|
|
post_components = data.split(',')
|
|
for pc in post_components:
|
|
if '=' in pc:
|
|
k, v = pc.split('=', 2)
|
|
post_data[k.strip()] = v.strip()
|
|
try:
|
|
r = requests.post(url, json=post_data)
|
|
except requests.exceptions.RequestException:
|
|
self.code = None
|
|
self.response = None
|
|
else:
|
|
self.status_code = r.status_code
|
|
try:
|
|
self.response = r.json()
|
|
except ValueError:
|
|
self.response = r.content
|
|
|
|
def check_response(self, step, component, data):
|
|
'''I receive a response (\w+) (.*)'''
|
|
if component == 'code':
|
|
assert self.status_code == int(data), "status code {0} != {1}".format(self.status_code, int(data))
|
|
elif component == 'text':
|
|
assert self.response == data.strip('"'), "response {0} does not contain {1}".format(self.response, data)
|
|
else:
|
|
assert component in self.response, "{0} is not part of the response".format(component)
|
|
assert self.response[component] == data, "{0} does not contain {1}".format(component, data)
|
|
|
|
def replication_works(self, step, time_limit):
|
|
'''replication works after (\d+) seconds'''
|
|
step.behave_as("""
|
|
When I add the table test_{0} to postgres0
|
|
Then table test_{0} is present on postgres1 after {1} seconds
|
|
""".format(int(time.time()), time_limit))
|
|
|
|
|
|
PatroniAPISteps(world)
|