Additional function to fs module

rrmngmnt/filesystem.py

@@ -89,3 +89,48 @@ def create_script(self, content, path):
                 raise errors.CommandExecutionFailure(
                     executor, cmd, rc, err,
                 )
+
+    def mkdir(self, path):
+        """
+        Create directory on host
+
+        :param path: directory path
+        :type path: str
+        :return: True, if action succeed, otherwise False
+        :rtype: bool
+        """
+        return self.host.run_command(
+            ['mkdir', path]
+        )[0] == 0
+
+    def chown(self, path, username, groupname):
+        """
+        Change owner of file or directory
+
+        :param path: file or directory path
+        :type path: str
+        :param username: change user owner to username
+        :type username: str
+        :param groupname: change group owner to groupname
+        :type groupname: str
+        :return: True, if action succeed, otherwise False
+        :rtype: bool
+        """
+        return self.host.run_command(
+            ['chown', '%s:%s' % (username, groupname), path]
+        )[0] == 0
+
+    def chmod(self, path, mode):
+        """
+        Change permission of directory or file
+
+        :param path: file or directory path
+        :type path: str
+        :param mode: permission mode(600 for example or u+x)
+        :type mode: str
+        :return: True, if action succeed, otherwise False
+        :rtype: bool
+        """
+        return self.host.run_command(
+            ['chmod', mode, path]
+        )[0] == 0

tests/test_filesystem.py

@@ -35,6 +35,9 @@ class TestFilesystem(object):
         'rm -rf /dir/to/remove': (0, '', ''),
         'cat %s' % "/tmp/file": (0, 'data', ''),
         'chmod +x /tmp/hello.sh': (0, '', ''),
+        'mkdir /dir/to/remove': (0, '', ''),
+        'chown root:root /dir/to/remove': (0, '', ''),
+        'chmod 600 /dir/to/remove': (0, '', '')
     }
     files = {}
 
@@ -84,3 +87,12 @@ def test_create_sctript(self):
         path = '/tmp/hello.sh'
         self.get_host().fs.create_script(data, path)
         assert self.files[path].data == data
+
+    def test_mkdir_positive(self):
+        assert self.get_host().fs.mkdir('/dir/to/remove')
+
+    def test_chown_positive(self):
+        assert self.get_host().fs.chown('/dir/to/remove', 'root', 'root')
+
+    def test_chmod_positive(self):
+        assert self.get_host().fs.chmod('/dir/to/remove', '600')