File: test.py

package info (click to toggle)
mat 0.6.1-4
  • links: PTS, VCS
  • area: main
  • in suites: stretch
  • size: 3,888 kB
  • ctags: 263
  • sloc: python: 1,629; makefile: 7; sh: 6
file content (116 lines) | stat: -rw-r--r-- 3,152 bytes parent folder | download
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 = [pair for pair in zip(clean, dirty) if not pair[0].endswith('pdf')]

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 paths to run the testsuite on the _local_
    version of MAT. See `run_all_tests` for more information about
    what paths 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)