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
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
|
#!/usr/bin/env python
# -*- coding: utf-8 -*
'''
Unit test for the library
'''
import unittest
import test
import sys
import tempfile
import tarfile
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')
class TestArchiveProcessing(test.MATTest):
''' Test archives cleaning
'''
def test_remove_bz2(self):
tar = tarfile.open("test.tar.bz2", "w:bz2")
for _,dirty in self.file_list:
tar.add(dirty)
tar.close()
current_file = MAT.mat.create_class_file("test.tar.bz2", False, add2archive=False)
current_file.remove_all()
current_file = MAT.mat.create_class_file("test.tar.bz2", False, add2archive=False)
self.assertTrue(current_file.is_clean())
def test_remove_tar(self):
tar = tarfile.open("test.tar", "w")
for _,dirty in self.file_list:
tar.add(dirty)
tar.close()
current_file = MAT.mat.create_class_file("test.tar", False, add2archive=False)
current_file.remove_all()
current_file = MAT.mat.create_class_file("test.tar", False, add2archive=False)
self.assertTrue(current_file.is_clean())
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))
suite.addTest(unittest.makeSuite(TestArchiveProcessing))
return suite
if __name__ == '__main__':
unittest.TextTestRunner(verbosity=test.VERBOSITY).run(get_tests())
|
