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
|
#!/usr/bin/env python
# -*- coding: utf-8 -*
"""
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
import subprocess
VERBOSITY = 15
clean = glob.glob('clean*')
clean.sort()
dirty = glob.glob('dirty*')
dirty.sort()
FILE_LIST = zip(clean, dirty)
try: # PDF render processing
import cairo
import gi
gi.require_version('Poppler', '0.18')
from gi.repository import Poppler
import pdfrw
except ImportError:
FILE_LIST.remove(('clean é.pdf', 'dirty é.pdf'))
try: # python-mutagen : audio file format
import mutagen
except ImportError:
FILE_LIST.remove(('clean é.ogg', 'dirty é.ogg'))
FILE_LIST.remove(('clean é.mp3', 'dirty é.mp3'))
FILE_LIST.remove(('clean é.flac', 'dirty é.flac'))
try: # exiftool
subprocess.check_output(['exiftool', '-ver'])
except OSError:
FILE_LIST.remove(('clean é.tif', 'dirty é.tif'))
class MATTest(unittest.TestCase):
"""
Parent class of all test-functions
"""
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_file, dirty_file in FILE_LIST:
clean_dir = os.path.join(self.tmpdir, clean_file)
dirty_dir = os.path.join(self.tmpdir, dirty_file)
shutil.copy2(clean_file, clean_dir)
shutil.copy2(dirty_file, dirty_dir)
self.file_list.append((clean_dir, dirty_dir))
def tearDown(self):
"""
Remove the tmp folder
"""
for root, dirs, files in os.walk(self.tmpdir):
for d in dirs + files:
os.chmod(os.path.join(root, d), 0o777)
shutil.rmtree(self.tmpdir)
def run_all_tests():
"""
This method will run all tests, both for cli and lib.
The imports of clitest and libtest are done here because
we're modifying the PATH (technically, it's two path:
the one used to spawn the `mat` process, and the one for Python import)
in the main function.
"""
import clitest
import libtest
suite = unittest.TestSuite()
suite.addTests(clitest.get_tests())
suite.addTests(libtest.get_tests())
return unittest.TextTestRunner(verbosity=VERBOSITY).run(suite).wasSuccessful()
def set_local():
""" Monkey patch pathes to run the testsuite on the _local_
version of MAT. See `run_all_tests` for more information about
what pathes we're changing and why.
"""
os.environ['PATH'] = '..:' + os.environ['PATH']
sys.path.append('..')
if __name__ == '__main__':
import argparse
parser = argparse.ArgumentParser(description='MAT testsuite')
parser.add_argument('-s', '--system', action='store_true',
help='Test the system-wide version of mat')
if parser.parse_args().system is False:
set_local()
sys.exit(run_all_tests() is False)
|
