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
|
#!/usr/bin/python
'''
Unit test for the CLI interface
'''
import unittest
import subprocess
import sys
sys.path.append('..')
from lib 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, 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, 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(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(stdout, "[+] File %s" % 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(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(stdout.strip('\n'), '[+] %s is not clean' % dirty)
class TestFileAttributes(unittest.TestCase):
'''
test various stuffs about files (readable, writable, exist, ...)
'''
def test_not_readable(self):
proc = subprocess.Popen(['../mat', 'not_readable'],
stdout=subprocess.PIPE)
stdout, _ = proc.communicate()
self.assertEqual(stdout.strip('\n'), 'Unable to pocess %s' % 'not_readable')
def test_not_writtable(self):
proc = subprocess.Popen(['../mat', 'not_writtable'],
stdout=subprocess.PIPE)
stdout, _ = proc.communicate()
self.assertEqual(stdout.strip('\n'), 'Unable to pocess %s' % 'not_writtable')
def test_not_exist(self):
proc = subprocess.Popen(['../mat', 'ilikecookies'],
stdout=subprocess.PIPE)
stdout, _ = proc.communicate()
self.assertEqual(stdout.strip('\n'), 'Unable to pocess %s' % 'ilikecookies')
def main():
suite = unittest.TestSuite()
suite.addTest(unittest.makeSuite(TestRemovecli))
suite.addTest(unittest.makeSuite(TestListcli))
suite.addTest(unittest.makeSuite(TestisCleancli))
test_result = unittest.TextTestRunner(verbosity=test.VERBOSITY).run(suite)
return len(test_result.failures)
if __name__ == '__main__':
sys.exit(main())
|
