summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--MAT/office.py10
-rw-r--r--test/clitest.py4
-rw-r--r--test/libtest.py18
3 files changed, 15 insertions, 17 deletions
diff --git a/MAT/office.py b/MAT/office.py
index 223fe84..232a3b8 100644
--- a/MAT/office.py
+++ b/MAT/office.py
@@ -83,7 +83,7 @@ class OpenDocumentStripper(archive.GenericArchiveStripper):
83 if os.path.isfile(name): 83 if os.path.isfile(name):
84 try: 84 try:
85 cfile = mat.create_class_file(name, False, 85 cfile = mat.create_class_file(name, False,
86 self.add2archive) 86 add2archive=self.add2archive)
87 cfile.remove_all() 87 cfile.remove_all()
88 logging.debug('Processing %s from %s' % (item, 88 logging.debug('Processing %s from %s' % (item,
89 self.filename)) 89 self.filename))
@@ -107,9 +107,8 @@ class OpenDocumentStripper(archive.GenericArchiveStripper):
107 try: 107 try:
108 zipin.getinfo('meta.xml') 108 zipin.getinfo('meta.xml')
109 except KeyError: # no meta.xml in the file 109 except KeyError: # no meta.xml in the file
110 kwargs = {'backup':self.backup, 'add2archive':self.add2archive}
111 czf = archive.ZipStripper(self.filename, self.parser, 110 czf = archive.ZipStripper(self.filename, self.parser,
112 'application/zip', **kwargs) 111 'application/zip', False, add2archive=self.add2archive)
113 if czf.is_clean(): 112 if czf.is_clean():
114 zipin.close() 113 zipin.close()
115 return True 114 return True
@@ -226,7 +225,7 @@ class OpenXmlStripper(archive.GenericArchiveStripper):
226 if os.path.isfile(name): # don't care about folders 225 if os.path.isfile(name): # don't care about folders
227 try: 226 try:
228 cfile = mat.create_class_file(name, False, 227 cfile = mat.create_class_file(name, False,
229 self.add2archive) 228 add2archive=self.add2archive)
230 cfile.remove_all() 229 cfile.remove_all()
231 logging.debug('Processing %s from %s' % (item, 230 logging.debug('Processing %s from %s' % (item,
232 self.filename)) 231 self.filename))
@@ -251,9 +250,8 @@ class OpenXmlStripper(archive.GenericArchiveStripper):
251 if item.startswith('docProps/'): 250 if item.startswith('docProps/'):
252 return False 251 return False
253 zipin.close() 252 zipin.close()
254 kwargs = {'backup':self.backup, 'add2archive':self.add2archive}
255 czf = archive.ZipStripper(self.filename, self.parser, 253 czf = archive.ZipStripper(self.filename, self.parser,
256 'application/zip', **kwargs) 254 'application/zip', False, add2archive=self.add2archive)
257 return czf.is_clean() 255 return czf.is_clean()
258 256
259 def get_meta(self): 257 def get_meta(self):
diff --git a/test/clitest.py b/test/clitest.py
index 0b041b9..bc0a7fd 100644
--- a/test/clitest.py
+++ b/test/clitest.py
@@ -22,14 +22,14 @@ class TestRemovecli(test.MATTest):
22 '''make sure that the cli remove all compromizing meta''' 22 '''make sure that the cli remove all compromizing meta'''
23 for _, dirty in self.file_list: 23 for _, dirty in self.file_list:
24 subprocess.call(['../mat', dirty]) 24 subprocess.call(['../mat', dirty])
25 current_file = mat.create_class_file(dirty, False, True) 25 current_file = mat.create_class_file(dirty, False, add2archive=True, low_pdf_quality=True)
26 self.assertTrue(current_file.is_clean()) 26 self.assertTrue(current_file.is_clean())
27 27
28 def test_remove_empty(self): 28 def test_remove_empty(self):
29 '''Test removal with clean files''' 29 '''Test removal with clean files'''
30 for clean, _ in self.file_list: 30 for clean, _ in self.file_list:
31 subprocess.call(['../mat', clean]) 31 subprocess.call(['../mat', clean])
32 current_file = mat.create_class_file(clean, False, True) 32 current_file = mat.create_class_file(clean, False, add2archive=True, low_pdf_quality=True)
33 self.assertTrue(current_file.is_clean()) 33 self.assertTrue(current_file.is_clean())
34 34
35 35
diff --git a/test/libtest.py b/test/libtest.py
index 2f0ee8f..7b31e3e 100644
--- a/test/libtest.py
+++ b/test/libtest.py
@@ -19,15 +19,15 @@ class TestRemovelib(test.MATTest):
19 def test_remove(self): 19 def test_remove(self):
20 '''make sure that the lib remove all compromizing meta''' 20 '''make sure that the lib remove all compromizing meta'''
21 for _, dirty in self.file_list: 21 for _, dirty in self.file_list:
22 current_file = mat.create_class_file(dirty, False, True) 22 current_file = mat.create_class_file(dirty, False, add2archive=True, low_pdf_quality=True)
23 current_file.remove_all() 23 current_file.remove_all()
24 current_file2 = mat.create_class_file(dirty, False, True) 24 current_file2 = mat.create_class_file(dirty, False, add2archive=True, low_pdf_quality=True)
25 self.assertTrue(current_file2.is_clean()) 25 self.assertTrue(current_file2.is_clean())
26 26
27 def test_remove_empty(self): 27 def test_remove_empty(self):
28 '''Test removal with clean files''' 28 '''Test removal with clean files'''
29 for clean, _ in self.file_list: 29 for clean, _ in self.file_list:
30 current_file = mat.create_class_file(clean, False, True) 30 current_file = mat.create_class_file(clean, False, add2archive=True, low_pdf_quality=True)
31 current_file.remove_all() 31 current_file.remove_all()
32 self.assertTrue(current_file.is_clean()) 32 self.assertTrue(current_file.is_clean())
33 33
@@ -39,14 +39,14 @@ class TestListlib(test.MATTest):
39 def test_list(self): 39 def test_list(self):
40 '''check if get_meta returns all the expected meta''' 40 '''check if get_meta returns all the expected meta'''
41 for _, dirty in self.file_list: 41 for _, dirty in self.file_list:
42 current_file = mat.create_class_file(dirty, False, True) 42 current_file = mat.create_class_file(dirty, False, add2archive=True, low_pdf_quality=True)
43 #FIXME assertisNotNone() : python 2.7 43 #FIXME assertisNotNone() : python 2.7
44 self.assertTrue(current_file.get_meta()) 44 self.assertTrue(current_file.get_meta())
45 45
46 def testlist_list_empty(self): 46 def testlist_list_empty(self):
47 '''check that a listing of a clean file return an empty dict''' 47 '''check that a listing of a clean file return an empty dict'''
48 for clean, _ in self.file_list: 48 for clean, _ in self.file_list:
49 current_file = mat.create_class_file(clean, False, True) 49 current_file = mat.create_class_file(clean, False, add2archive=True, low_pdf_quality=True)
50 self.assertEqual(current_file.get_meta(), dict()) 50 self.assertEqual(current_file.get_meta(), dict())
51 51
52 52
@@ -57,13 +57,13 @@ class TestisCleanlib(test.MATTest):
57 def test_dirty(self): 57 def test_dirty(self):
58 '''test is_clean on clean files''' 58 '''test is_clean on clean files'''
59 for _, dirty in self.file_list: 59 for _, dirty in self.file_list:
60 current_file = mat.create_class_file(dirty, False, True) 60 current_file = mat.create_class_file(dirty, False, add2archive=True, low_pdf_quality=True)
61 self.assertFalse(current_file.is_clean()) 61 self.assertFalse(current_file.is_clean())
62 62
63 def test_clean(self): 63 def test_clean(self):
64 '''test is_clean on dirty files''' 64 '''test is_clean on dirty files'''
65 for clean, _ in self.file_list: 65 for clean, _ in self.file_list:
66 current_file = mat.create_class_file(clean, False, True) 66 current_file = mat.create_class_file(clean, False, add2archive=True, low_pdf_quality=True)
67 self.assertTrue(current_file.is_clean()) 67 self.assertTrue(current_file.is_clean())
68 68
69 69
@@ -72,10 +72,10 @@ class TestFileAttributes(unittest.TestCase):
72 test various stuffs about files (readable, writable, exist, ...) 72 test various stuffs about files (readable, writable, exist, ...)
73 ''' 73 '''
74 def test_not_writtable(self): 74 def test_not_writtable(self):
75 self.assertFalse(mat.create_class_file('not_writtable', False, True)) 75 self.assertFalse(mat.create_class_file('not_writtable', False, add2archive=True, low_pdf_quality=True))
76 76
77 def test_not_exist(self): 77 def test_not_exist(self):
78 self.assertFalse(mat.create_class_file('ilikecookies', False, True)) 78 self.assertFalse(mat.create_class_file('ilikecookies', False, add2archive=True, low_pdf_quality=True))
79 79
80 80
81def get_tests(): 81def get_tests():