Time-Line/execute-specs.py at master · sugarlabs/Time-Line · GitHub

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
#!/usr/bin/env python
#
# Copyright (C) 2009, 2010, 2011  Rickard Lindberg, Roger Lindberg
#
# This file is part of Timeline.
#
# Timeline is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# Timeline is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with Timeline.  If not, see <http://www.gnu.org/licenses/>.

import sys
import os.path
import unittest
import doctest

def execute_all_specs():
    setup_paths()
    install_gettext_in_builtin_namespace()
    suite = create_suite()
    all_pass = execute_suite(suite)
    return all_pass

def setup_paths():
    root_dir = os.path.abspath(os.path.dirname(__file__))
    sys.path.insert(0, os.path.join(root_dir, "libs", "dev", "mock-0.7.2"))
    sys.path.insert(0, os.path.join(root_dir, "libs", "dependencies", "icalendar-3.2"))
    sys.path.insert(0, os.path.join(root_dir, "libs", "dependencies", "pytz-2012j"))

def install_gettext_in_builtin_namespace():
    def _(message):
        return "#%s#" % message
    import __builtin__
    __builtin__.__dict__["_"] = _

def create_suite():
    suite = unittest.TestSuite()
    add_specs(suite)
    add_doctests(suite)
    return suite

def execute_suite(suite):
    res = unittest.TextTestRunner(verbosity=1).run(suite)
    return res.wasSuccessful()

def add_specs(suite):
    for file in os.listdir(os.path.join(os.path.dirname(__file__), "specs")):
        if file.endswith(".py") and file != "__init__.py":
            module_name = os.path.basename(file)[:-3]
            abs_module_name = "specs.%s" % module_name
            load_test_cases_from_module_name(suite, abs_module_name)

def add_doctests(suite):
    load_doc_test_from_module_name(suite, "timelinelib.xml.parser")
    load_doc_test_from_module_name(suite, "timelinelib.utils")

def load_test_cases_from_module_name(suite, module_name):
    __import__(module_name)
    module = sys.modules[module_name]
    module_suite = unittest.defaultTestLoader.loadTestsFromModule(module)
    filtered = filter_suite(module_suite)
    suite.addTest(filtered)

def filter_suite(test):
    new_suite = unittest.TestSuite()
    if isinstance(test, unittest.TestCase):
        if include_test(test):
            new_suite.addTest(test)
    else:
        for subtest in test:
            new_suite.addTest(filter_suite(subtest))
    return new_suite

def include_test(test):
    if hasattr(test, "GUI") and "--skip-gui" in sys.argv:
        return False
    if hasattr(test, "IO") and "--skip-io" in sys.argv:
        return False
    return True

def load_doc_test_from_module_name(suite, module_name):
    __import__(module_name)
    module = sys.modules[module_name]
    module_suite = doctest.DocTestSuite(module)
    suite.addTest(module_suite)

if __name__ == '__main__':
    all_pass = execute_all_specs()
    if all_pass:
        sys.exit(0)
    else:
        sys.exit(1)