Fix #4 (for real this time?)

2 files changed, 13 insertions, 1 deletions

diff --git a/fabtools/postgres.py b/fabtools/postgres.py
index 05f3a75..a13e64a 100644
--- a/fabtools/postgres.py
+++ b/fabtools/postgres.py
@@ -14,7 +14,8 @@ def _run_as_pg(command):
     """
     Run command as 'postgres' user
     """
-    return sudo(command, user='postgres', shell=False)
+    with cd('/var/lib/postgresql'):
+        return sudo('sudo -u postgres %s' % command)
 
 
 def user_exists(name):
diff --git a/fabtools/tests/fabfiles/postgres.py b/fabtools/tests/fabfiles/postgres.py
index 4eaa3d0..8c834e5 100644
--- a/fabtools/tests/fabfiles/postgres.py
+++ b/fabtools/tests/fabfiles/postgres.py
@@ -12,6 +12,17 @@ def postgresql():
     """
     require.postgres.server()
 
+    # Test low-level operations
+    assert not fabtools.postgres.user_exists('alice')
+    assert not fabtools.postgres.user_exists('bob')
+    fabtools.postgres.create_user('alice', password='1234')
+    assert fabtools.postgres.user_exists('alice')
+    assert not fabtools.postgres.user_exists('bob')
+    fabtools.postgres.create_user('bob', password='5678')
+    assert fabtools.postgres.user_exists('alice')
+    assert fabtools.postgres.user_exists('bob')
+
+    # Test high-level operations
     require.postgres.user('pguser', 'foo')
     assert fabtools.postgres.user_exists('pguser')