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
113
114
|
#!/usr/bin/env python
# -*- coding: utf-8 -*
'''
Unit test for the library
'''
import unittest
import test
import sys
import tempfile
sys.path.append('..')
import MAT
class TestRemovelib(test.MATTest):
'''
test the remove_all() method
'''
def test_remove(self):
'''make sure that the lib remove all compromizing meta'''
for _, dirty in self.file_list:
current_file = MAT.mat.create_class_file(dirty, False, add2archive=True)
current_file.remove_all()
current_file = MAT.mat.create_class_file(dirty, False, add2archive=True)
self.assertTrue(current_file.is_clean())
def test_remove_empty(self):
'''Test removal with clean files'''
for clean, _ in self.file_list:
current_file = MAT.mat.create_class_file(clean, False, add2archive=True)
current_file.remove_all()
current_file = MAT.mat.create_class_file(clean, False, add2archive=True)
self.assertTrue(current_file.is_clean())
class TestListlib(test.MATTest):
'''
test the get_meta() method
'''
def test_list(self):
'''check if get_meta returns metadata'''
for _, dirty in self.file_list:
current_file = MAT.mat.create_class_file(dirty, False, add2archive=True)
self.assertIsNotNone(current_file.get_meta())
def testlist_list_empty(self):
'''check that a listing of a clean file returns an empty dict'''
for clean, _ in self.file_list:
current_file = MAT.mat.create_class_file(clean, False, add2archive=True)
self.assertEqual(current_file.get_meta(), dict())
class TestisCleanlib(test.MATTest):
'''
test the is_clean() method
'''
def test_dirty(self):
'''test is_clean on dirty files'''
for _, dirty in self.file_list:
current_file = MAT.mat.create_class_file(dirty, False, add2archive=True)
self.assertFalse(current_file.is_clean())
def test_clean(self):
'''test is_clean on clean files'''
for clean, _ in self.file_list:
current_file = MAT.mat.create_class_file(clean, False, add2archive=True)
self.assertTrue(current_file.is_clean())
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'''
self.assertFalse(MAT.mat.create_class_file('not_writtable', False, add2archive=True))
def test_not_exist(self):
''' test MAT's behaviour on non-existent file'''
self.assertFalse(MAT.mat.create_class_file('ilikecookies', False, add2archive=True))
def test_empty(self):
''' test MAT's behaviour on empty file'''
self.assertFalse(MAT.mat.create_class_file('empty_file', False, add2archive=True))
class TestSecureRemove(unittest.TestCase):
''' Test the secure_remove function
'''
def test_remove_existing(self):
''' test the secure removal of an existing file
'''
_, file_to_remove = tempfile.mkstemp()
self.assertTrue(MAT.mat.secure_remove(file_to_remove))
def test_remove_fail(self):
''' test the secure removal of an non-removable file
'''
self.assertRaises(MAT.exceptions.UnableToRemoveFile, MAT.mat.secure_remove, '/NOTREMOVABLE')
def get_tests():
''' Returns every libtests'''
suite = unittest.TestSuite()
suite.addTest(unittest.makeSuite(TestRemovelib))
suite.addTest(unittest.makeSuite(TestListlib))
suite.addTest(unittest.makeSuite(TestisCleanlib))
suite.addTest(unittest.makeSuite(TestFileAttributes))
suite.addTest(unittest.makeSuite(TestSecureRemove))
return suite
if __name__ == '__main__':
unittest.TextTestRunner(verbosity=test.VERBOSITY).run(get_tests())
|
