1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
|
#!/usr/bin/env python
# -*- coding: utf-8 -*
'''
Unit test for the CLI interface
'''
import unittest
import subprocess
import sys
sys.path.append('..')
from MAT import mat
import test
class TestRemovecli(test.MATTest):
'''
test if cli correctly remove metadatas
'''
def test_remove(self):
'''make sure that the cli remove all compromizing meta'''
for _, dirty in self.file_list:
subprocess.call(['../mat', dirty])
current_file = mat.create_class_file(dirty, False, add2archive=True, low_pdf_quality=True)
self.assertTrue(current_file.is_clean())
def test_remove_empty(self):
'''Test removal with clean files'''
for clean, _ in self.file_list:
subprocess.call(['../mat', clean])
current_file = mat.create_class_file(clean, False, add2archive=True, low_pdf_quality=True)
self.assertTrue(current_file.is_clean())
class TestListcli(test.MATTest):
'''
test if cli correctly display metadatas
'''
def test_list_clean(self):
'''check if get_meta returns meta'''
for clean, _ in self.file_list:
proc = subprocess.Popen(['../mat', '-d', clean],
stdout=subprocess.PIPE)
stdout, _ = proc.communicate()
self.assertEqual(str(stdout).strip('\n'), "[+] File %s \
:\nNo harmful metadata found" % clean)
def test_list_dirty(self):
'''check if get_meta returns all the expected meta'''
for _, dirty in self.file_list:
proc = subprocess.Popen(['../mat', '-d', dirty],
stdout=subprocess.PIPE)
stdout, _ = proc.communicate()
self.assertNotEqual(str(stdout), "[+] File %s :\n No\
harmul metadata found" % dirty)
class TestisCleancli(test.MATTest):
'''
check if cli correctly check if a file is clean or not
'''
def test_clean(self):
'''test is_clean on clean files'''
for clean, _ in self.file_list:
proc = subprocess.Popen(['../mat', '-c', clean],
stdout=subprocess.PIPE)
stdout, _ = proc.communicate()
self.assertEqual(str(stdout).strip('\n'), '[+] %s is clean' % clean)
def test_dirty(self):
'''test is_clean on dirty files'''
for _, dirty in self.file_list:
proc = subprocess.Popen(['../mat', '-c', dirty],
stdout=subprocess.PIPE)
stdout, _ = proc.communicate()
self.assertEqual(str(stdout).strip('\n'), '[+] %s is not clean' % dirty)
class TestFileAttributes(unittest.TestCase):
'''
test various stuffs about files (readable, writable, exist, ...)
'''
def test_not_writtable(self):
''' test MAT's behaviour on non-writable file'''
proc = subprocess.Popen(['../mat', 'not_writtable'],
stdout=subprocess.PIPE)
stdout, _ = proc.communicate()
self.assertEqual(str(stdout).strip('\n'), '[-] %s is not writable' % 'not_writtable')
def test_not_exist(self):
''' test MAT's behaviour on non-existent file'''
proc = subprocess.Popen(['../mat', 'ilikecookies'],
stdout=subprocess.PIPE)
stdout, _ = proc.communicate()
self.assertEqual(str(stdout).strip('\n'), 'Unable to process %s' % 'ilikecookies')
def test_empty(self):
''' test MAT's behaviour on empty file'''
proc = subprocess.Popen(['../mat', 'empty_file'],
stdout=subprocess.PIPE)
stdout, _ = proc.communicate()
self.assertEqual(str(stdout).strip('\n'), 'Unable to process %s' % 'ilikecookies')
def get_tests():
''' Return every clitests'''
suite = unittest.TestSuite()
suite.addTest(unittest.makeSuite(TestRemovecli))
suite.addTest(unittest.makeSuite(TestListcli))
suite.addTest(unittest.makeSuite(TestisCleancli))
return suite
|
