Add some file-attribute related tests

2 files changed, 21 insertions, 0 deletions

diff --git a/test/clitest.py b/test/clitest.py
index ec3c19c..3b09c62 100644
--- a/test/clitest.py
+++ b/test/clitest.py
@@ -10,6 +10,7 @@ import unittest
 import subprocess
 import sys
 import tarfile
+import stat
 
 import test
 MAT_PATH = 'mat'
@@ -113,6 +114,14 @@ class TestFileAttributes(unittest.TestCase):
         stdout, _ = proc.communicate()
         self.assertEqual(str(stdout).strip('\n'), '[-] Unable to process empty_file')
 
+    def test_not_readable(self):
+        """ test MAT's behaviour on non-writable file"""
+        open('non_readable', 'a').close()
+        os.chmod('non_readable', 0 & stat.S_IWRITE)
+        proc = subprocess.Popen([MAT_PATH, 'non_readable'], stdout=subprocess.PIPE)
+        stdout, _ = proc.communicate()
+        os.remove('non_readable')
+
 
 class TestUnsupported(test.MATTest):
     """ test MAT's behaviour on unsupported files """
diff --git a/test/libtest.py b/test/libtest.py
index 03ae4a1..fa65528 100644
--- a/test/libtest.py
+++ b/test/libtest.py
@@ -7,6 +7,7 @@
 
 import os
 import sys
+import stat
 import shutil
 import tarfile
 import tempfile
@@ -91,6 +92,17 @@ class TestFileAttributes(unittest.TestCase):
         self.assertFalse(libmat.mat.create_class_file('empty_file', False, add2archive=True))
         os.remove('empty_file')
 
+    def test_not_writtable(self):
+        """ test MAT's behaviour on non-writable file"""
+        self.assertFalse(libmat.mat.create_class_file('not_writtable', False, add2archive=True))
+
+    def test_not_readable(self):
+        """ test MAT's behaviour on non-readable file"""
+        open('non_readable', 'a').close()
+        os.chmod('non_readable', 0 | stat.S_IWRITE)
+        self.assertFalse(libmat.mat.create_class_file('non_readable', False, add2archive=True))
+        os.remove('non_readable')
+
 
 class TestSecureRemove(unittest.TestCase):
     """ Test the secure_remove function