blob: e9e8f82118020d7a512b0ece2c74931d6585044a (
plain)
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
|
'''
Class for the testing suite :
- get the list of all test files
- create a copy of them on start
- remove the copy on end
'''
import shutil
import os
import glob
import sys
import tempfile
import unittest
sys.path.append('..')
from lib import mat
FILE_LIST = zip(glob.glob('clean*'), glob.glob('dirty*'))
class MATTest(unittest.TestCase):
def setUp(self):
'''
Create working copy of the clean and the dirty file in the TMP dir
'''
self.file_list = []
self.tmpdir = tempfile.mkdtemp()
for clean, dirty in FILE_LIST:
shutil.copy2(clean, self.tmpdir + os.sep + clean)
shutil.copy2(dirty, self.tmpdir + os.sep + dirty)
self.file_list.append((self.tmpdir + os.sep + clean,
self.tmpdir + os.sep + dirty))
def tearDown(self):
'''
Remove the tmp folder
'''
for clean, dirty in self.file_list:
mat.secure_remove(clean)
mat.secure_remove(dirty)
shutil.rmtree(self.tmpdir)
|
