db/pg_probackup
db/pg_probackup
1
0
Fork 0
mirror of https://github.com/postgrespro/pg_probackup.git synced 2024-12-02 09:53:24 +02:00
Code Issues Releases Activity

fix auth_test

Browse Source
This commit is contained in:
s.logvinenko 2017-11-29 11:48:26 +03:00
parent a2b802ca8f
commit c09e805697
1 changed files with 22 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
tests/auth_test.py
View File

@ -1,6 +1,7 @@
import os
import unittest
import tempfile
import signal
from subprocess import Popen, PIPE, TimeoutExpired
from .helpers.ptrack_helpers import ProbackupTest, ProbackupException
@ -44,6 +45,7 @@ class AuthTest(unittest.TestCase):
@classmethod
def tearDownClass(cls):
cls.node.cleanup()
cls.pb.del_test_dir(module_name, '')
def setUp(self):
@ -69,23 +71,23 @@ class AuthTest(unittest.TestCase):
def test_wrong_password(self):
try:
run_pb_with_auth(self.cmd,'wrong_password')
except:
pass
except ProbackupException as e:
self.fail(repr(e))
@unittest.skip
def test_ctrl_c_event(self):
try:
run_pb_with_auth(self.cmd,signal='CTRL_C_EVENT')
run_pb_with_auth(self.cmd,kill=True)
except:
pass
def run_pb_with_auth(cmd, password=None, signal=None):
def run_pb_with_auth(cmd, password=None, kill=False):
probackup = Popen(cmd, stdin=PIPE, stdout=PIPE, stderr=PIPE)
if probackup.stdout.readline() == "Password:":
if probackup.stdout.readline() == "Password as user :":
if signal:
probackup.send_signal(signal)
os.kill(probackup.pid, signal.SIGINT)
try:
out, err = probackup.communicate(password, 10)
return probackup.returncode, out, err
@ -111,3 +113,17 @@ def add_backup_user(node):
GRANT EXECUTE ON FUNCTION pg_ptrack_get_and_clear(oid, oid) TO backup;
'''
node.psql("postgres", query.replace('\n',''))
# dirty hack
def modify_pg_hba(node):
hba_conf = os.path.join(node.data_dir, "pg_hba.conf")
with open(hba_conf, "w") as conf:
conf.write("# TYPE\tDATABASE\tUSER\tADDRESS\t\tMETHOD\n"
"local\tall\t\tall\t\t\ttrust\n"
"host\tall\t\tall\t127.0.0.1/32\tmd5\n"
"host\tall\t\tall\t::1/128\t\ttrust\n"
# replication
"local\treplication\tall\t\t\ttrust\n"
"host\treplication\tall\t127.0.0.1/32\ttrust\n"
"host\treplication\tall\t::1/128\t\ttrust\n")