Issue #355 Use setproctitle to hide command line passwords

1 files changed, 20 insertions, 2 deletions

diff --git a/tests/test_main.py b/tests/test_main.py
index e25f7f08..6e8d14b0 100644
--- a/tests/test_main.py
+++ b/tests/test_main.py
@@ -1,5 +1,5 @@
 import pytest
-from pgcli.main import need_completion_refresh
+from pgcli.main import need_completion_refresh, obscure_process_password
 
 
 @pytest.mark.parametrize('sql', [
@@ -7,4 +7,22 @@ from pgcli.main import need_completion_refresh
     'SELECT * FROM foo; DROP TABLE foo',
 ])
 def test_need_completion_refresh(sql):
-    assert need_completion_refresh(sql)
\ No newline at end of file
+    assert need_completion_refresh(sql)
+
+def test_obscure_process_password():
+    import setproctitle
+    original_title = setproctitle.getproctitle()
+
+    setproctitle.setproctitle("pgcli user=root password=secret host=localhost")
+    obscure_process_password()
+    title = setproctitle.getproctitle()
+    expected = "pgcli user=root password=xxxx host=localhost"
+    assert title == expected
+
+    setproctitle.setproctitle("pgcli postgres://root:secret@localhost/db")
+    obscure_process_password()
+    title = setproctitle.getproctitle()
+    expected = "pgcli postgres://root:xxxx@localhost/db"
+    assert title == expected
+
+    setproctitle.setproctitle(original_title)