Added postgres test

arecker · Sep 19, 2015 · 869d295 · 869d295
1 parent bfc09f0
commit 869d295
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 4 deletions.
diff --git a/msg/db.py b/msg/db.py
@@ -12,7 +12,7 @@ def _run_command(self, cmd):
 
 
 class PostgresController(BaseDatabaseController):
-    pass
+    user = 'postgres'
 
 
 class MysqlController(BaseDatabaseController):

diff --git a/tests/mocks.py b/tests/mocks.py
@@ -61,8 +61,8 @@ def __init__(self):
         self._files = MockFiles(self._callback)
         self._operations = MockOperations(self._callback)
 
-    def sudo(self, cmd):
-        self._shell_command(cmd, True)
+    def sudo(self, cmd, user=None):
+        self._shell_command(cmd, True, user=user)
 
     def run(self, cmd):
         self._shell_command(cmd, False)
@@ -71,11 +71,13 @@ def _callback(self, obj):
         self.history.append(obj)
         self.last = obj
 
-    def _shell_command(self, cmd, sudo):
+    def _shell_command(self, cmd, sudo, user=None):
         self.last = {
             'command': cmd,
             'sudo': sudo
         }
+        if user:
+            self.last['user'] = user
         self.history.append(self.last)
 
 

diff --git a/tests/test_db.py b/tests/test_db.py
@@ -10,3 +10,14 @@ def test_run_command_without_user(self):
             self.fail('able to run an incomplete controller')
         except NotImplementedError as e:
             self.assertEqual(e.message, '\'user\' required')
+
+
+class PostgresDatabaseController(MockFabricTestCase):
+    def test_run_command_without_user(self):
+        d = db.PostgresController()
+        d._run_command('drop everything')
+        self.assertEqual(self.mock.last, {
+            'command': 'drop everything',
+            'sudo': True,
+            'user': 'postgres'
+        })