db/pg_probackup
1
0
Fork 0
mirror of https://github.com/postgrespro/pg_probackup.git synced 2026-06-21 01:34:15 +02:00
Code Issues Releases Activity

fix auth_test

Browse Source
This commit is contained in:
sfalkon
2017-12-05 20:13:41 +03:00
parent d86778dc84
commit 3f650ed2f1
1 changed files with 24 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/auth_test.py
+10 -11
View File
@@ -1,7 +1,5 @@
""" """
Description: The Test suite check behavior of pg_probackup utility, if password is required for connection to PostgreSQL instance.
The Test suite check behavior of pg_probackup utility,
if password is required for connection to PostgreSQL instance.
- https://confluence.postgrespro.ru/pages/viewpage.action?pageId=16777522 - https://confluence.postgrespro.ru/pages/viewpage.action?pageId=16777522
""" """
@@ -94,7 +92,7 @@ class AuthTest(unittest.TestCase):
try: try:
self.assertIn("ERROR: no password supplied", self.assertIn("ERROR: no password supplied",
"".join(map(lambda x: x.decode("utf-8"), "".join(map(lambda x: x.decode("utf-8"),
self.run_pb_with_auth(self.cmd, '\0\r\n')) run_pb_with_auth([self.pb.probackup_path] + self.cmd, '\0\r\n'))
) )
) )
except (TIMEOUT, ExceptionPexpect) as e: except (TIMEOUT, ExceptionPexpect) as e:
@@ -105,7 +103,7 @@ class AuthTest(unittest.TestCase):
try: try:
self.assertIn("password authentication failed", self.assertIn("password authentication failed",
"".join(map(lambda x: x.decode("utf-8"), "".join(map(lambda x: x.decode("utf-8"),
self.run_pb_with_auth(self.cmd, 'wrong_password\r\n')) run_pb_with_auth([self.pb.probackup_path] + self.cmd, 'wrong_password\r\n'))
) )
) )
except (TIMEOUT, ExceptionPexpect) as e: except (TIMEOUT, ExceptionPexpect) as e:
@@ -116,7 +114,7 @@ class AuthTest(unittest.TestCase):
try: try:
self.assertIn("completed", self.assertIn("completed",
"".join(map(lambda x: x.decode("utf-8"), "".join(map(lambda x: x.decode("utf-8"),
self.run_pb_with_auth(self.cmd, 'password\r\n')) run_pb_with_auth([self.pb.probackup_path] + self.cmd, 'password\r\n'))
) )
) )
except (TIMEOUT, ExceptionPexpect) as e: except (TIMEOUT, ExceptionPexpect) as e:
@@ -125,7 +123,7 @@ class AuthTest(unittest.TestCase):
def test_ctrl_c_event(self): def test_ctrl_c_event(self):
""" Test case: PGPB_AUTH02 - send interrupt signal """ """ Test case: PGPB_AUTH02 - send interrupt signal """
try: try:
self.run_pb_with_auth(self.cmd, kill=True) run_pb_with_auth([self.pb.probackup_path] + self.cmd, kill=True)
except TIMEOUT: except TIMEOUT:
self.fail("Error: CTRL+C event ignored") self.fail("Error: CTRL+C event ignored")
@@ -170,9 +168,10 @@ class AuthTest(unittest.TestCase):
except ProbackupException as e: except ProbackupException as e:
self.fail(e) self.fail(e)
def run_pb_with_auth(self, password=None, kill=False):
def run_pb_with_auth(cmd, password=None, kill=False):
try: try:
with spawn(" ".join([self.pb.probackup_path] + self.cmd), timeout=10) as probackup: with spawn(" ".join(cmd), timeout=10) as probackup:
result = probackup.expect("Password for user .*:", 5) result = probackup.expect("Password for user .*:", 5)
if kill: if kill:
probackup.kill(signal.SIGINT) probackup.kill(signal.SIGINT)
@@ -202,7 +201,7 @@ def modify_pg_hba(node):
def create_pgpass(path, line): def create_pgpass(path, line):
with open(path, 'w') as passfile: with open(path, 'w+') as passfile:
# host:port:db:username:password # host:port:db:username:password
passfile.write(line) passfile.write(line)
os.chmod(path, 600) os.chmod(path, 0600)