From 8cc0aa90b1c426accf9a4120455ba8c1114d3b91 Mon Sep 17 00:00:00 2001 From: Alexander Kukushkin Date: Tue, 26 May 2015 10:39:16 +0200 Subject: [PATCH 1/3] Fix typos when passing connection parameters to psycopg2. --- helpers/postgresql.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/helpers/postgresql.py b/helpers/postgresql.py index 4726ddc4..5138662d 100644 --- a/helpers/postgresql.py +++ b/helpers/postgresql.py @@ -169,7 +169,8 @@ class Postgresql: if member.hostname == self.name: continue try: - member_conn = psycopg2.connect(parseurl(member.address)) + r = parseurl(member.address) + member_conn = psycopg2.connect(**r) member_conn.autocommit = True member_cursor = member_conn.cursor() member_cursor.execute( From 56daec0d6c94a48ac1826ab6935efde615980500 Mon Sep 17 00:00:00 2001 From: Alexander Kukushkin Date: Tue, 26 May 2015 15:53:57 +0200 Subject: [PATCH 2/3] Api: Bugfix for current_xlog_location in slaves. --- helpers/api.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/helpers/api.py b/helpers/api.py index f0d47e10..83aa619b 100644 --- a/helpers/api.py +++ b/helpers/api.py @@ -37,7 +37,9 @@ class RestApiHandler(BaseHTTPRequestHandler): cursor = self.server.cursor() cursor.execute("""SELECT to_char(pg_postmaster_start_time(), 'YYYY-MM-DD HH24:MI:SS.MS TZ'), pg_is_in_recovery(), - pg_current_xlog_location(), + CASE WHEN pg_is_in_recovery() + THEN null + ELSE pg_current_xlog_location() END, pg_last_xlog_receive_location(), pg_last_xlog_replay_location(), pg_is_in_recovery() AND pg_is_xlog_replay_paused()""") From 9968a66048157ba74d261b71766a8aaeae0d331b Mon Sep 17 00:00:00 2001 From: Alexander Kukushkin Date: Wed, 27 May 2015 11:08:06 +0200 Subject: [PATCH 3/3] helpers/postgresql.py Replace os.system with subprocess.call --- tests/test_governor.py | 7 ++++--- tests/test_postgresql.py | 7 ++++--- 2 files changed, 8 insertions(+), 6 deletions(-) diff --git a/tests/test_governor.py b/tests/test_governor.py index 34b46043..941d6d45 100644 --- a/tests/test_governor.py +++ b/tests/test_governor.py @@ -1,14 +1,15 @@ import os import psycopg2 -import unittest import requests +import subprocess import sys import time +import unittest import yaml from governor import Governor, main, sigchld_handler, sigterm_handler from test_ha import true, false -from test_postgresql import Postgresql, os_system, psycopg2_connect +from test_postgresql import Postgresql, subprocess_call, psycopg2_connect from test_etcd import requests_get, requests_put, requests_delete if sys.hexversion >= 0x03000000: @@ -38,7 +39,7 @@ class TestGovernor(unittest.TestCase): def set_up(self): self.touched = False - os.system = os_system + subprocess.call = subprocess_call psycopg2.connect = psycopg2_connect requests.get = requests_get requests.put = requests_put diff --git a/tests/test_postgresql.py b/tests/test_postgresql.py index 2e30b602..dc6fb512 100644 --- a/tests/test_postgresql.py +++ b/tests/test_postgresql.py @@ -1,13 +1,14 @@ import os import psycopg2 -import unittest import shutil +import subprocess +import unittest from helpers.etcd import Cluster, Member from helpers.postgresql import Postgresql -def os_system(cmd): +def subprocess_call(cmd, shell=False): return 0 @@ -89,7 +90,7 @@ class TestPostgresql(unittest.TestCase): super(TestPostgresql, self).__init__(method_name) def set_up(self): - os.system = os_system + subprocess.call = subprocess_call self.p = Postgresql({'name': 'test0', 'data_dir': 'data/test0', 'listen': '127.0.0.1, 127.0.0.2:5432', 'connect_address': '127.0.0.2:5432', 'replication': { 'username': 'replicator', 'password': 'rep-pass', 'network': '127.0.0.1/32'}, 'parameters': {'foo': 'bar'}, 'recovery_conf': {'foo': 'bar'}}) psycopg2.connect = psycopg2_connect