← Back to branch summary

~ubuntuone-control-tower/ubuntuone-storage-protocol/stable-1-6

« back to all changes in this revision

Viewing changes to contrib/test

  • Committer: Tarmac
  • Author(s): Rodney Dawes
  • Date: 2010-11-16 15:10:48 UTC
  • mfrom: (121.1.4 use-packages)
  • Revision ID: tarmac-20101116151048-b0e20j7lorb4yhe1
Switch to using packaged mocker and ubuntuone-dev-tools
Use pyflakes with u1lint and also run pep8
Fix a lot of pylint/pyflakes/pep8 errors

Show diffs side-by-side

added added

removed removed

Lines of Context:
contrib/test
1
 
#!/usr/bin/env python
2
 
 
3
 
import os
4
 
import re
5
 
import signal
6
 
import sys
7
 
import string
8
 
import subprocess
9
 
import unittest
10
 
 
11
 
sys.path.insert(0, os.path.abspath("."))
12
 
 
13
 
class TestRunner(object):
14
 
 
15
 
    def _load_unittest(self, relpath):
16
 
        """Load unit tests from a Python module with the given relative path."""
17
 
        assert relpath.endswith(".py"), (
18
 
            "%s does not appear to be a Python module" % relpath)
19
 
        modpath = relpath.replace(os.path.sep, ".")[:-3]
20
 
        module = __import__(modpath, None, None, [""])
21
 
 
22
 
        # If the module has a 'suite' or 'test_suite' function, use that
23
 
        # to load the tests.
24
 
        if hasattr(module, "suite"):
25
 
            return module.suite()
26
 
        elif hasattr(module, "test_suite"):
27
 
            return module.test_suite()
28
 
        else:
29
 
            return unittest.defaultTestLoader.loadTestsFromModule(module)
30
 
 
31
 
    def _collect_tests(self, testpath, test_pattern):
32
 
        """Return the set of unittests."""
33
 
        suite = unittest.TestSuite()
34
 
        if test_pattern:
35
 
            pattern = re.compile('.*%s.*' % test_pattern)
36
 
        else:
37
 
            pattern = None 
38
 
 
39
 
        if testpath:
40
 
            module_suite = self._load_unittest(testpath)
41
 
            if pattern:
42
 
                for inner_suite in module_suite._tests:
43
 
                    for test in inner_suite._tests:
44
 
                        if pattern.match(test.id()):
45
 
                            suite.addTest(test)
46
 
            else:
47
 
                suite.addTests(module_suite)
48
 
            return suite
49
 
 
50
 
        # We don't use the dirs variable, so ignore the warning
51
 
        # pylint: disable-msg=W0612
52
 
        for root, dirs, files in os.walk("tests"):
53
 
            for file in files:
54
 
                path = os.path.join(root, file)
55
 
                if file.endswith(".py") and file.startswith("test_"):
56
 
                    module_suite = self._load_unittest(path)
57
 
                    if pattern:
58
 
                        for inner_suite in module_suite._tests:
59
 
                            for test in inner_suite._tests:
60
 
                                if pattern.match(test.id()):
61
 
                                    suite.addTest(test)
62
 
                    else:
63
 
                        suite.addTests(module_suite)
64
 
        return suite
65
 
 
66
 
    def run(self, testpath, test_pattern=None):
67
 
        """run the tests. """
68
 
        from twisted.internet import reactor
69
 
        from twisted.trial.reporter import TreeReporter
70
 
        from twisted.trial.runner import TrialRunner
71
 
 
72
 
        workingDirectory = os.path.join(os.getcwd(), "_trial_temp", "tmp")
73
 
        runner = TrialRunner(reporterFactory=TreeReporter, realTimeErrors=True,
74
 
                            workingDirectory=workingDirectory)
75
 
 
76
 
        # setup a custom XDG_CACHE_HOME and create the logs directory
77
 
        xdg_cache = os.path.join(os.getcwd(), "_trial_temp", "xdg_cache")
78
 
        os.environ["XDG_CACHE_HOME"] = xdg_cache
79
 
        # setup the ROOT_DIR env var
80
 
        os.environ['ROOT_DIR'] = os.getcwd()
81
 
        if not os.path.exists(xdg_cache):
82
 
            os.makedirs(xdg_cache)
83
 
        success = 0
84
 
        suite = self._collect_tests(testpath, test_pattern)
85
 
        result = runner.run(suite)
86
 
        success = result.wasSuccessful()
87
 
        if not success:
88
 
            sys.exit(1)
89
 
        else:
90
 
            sys.exit(0)
91
 
 
92
 
 
93
 
if __name__ == '__main__':
94
 
    from optparse import OptionParser
95
 
    usage = '%prog [options] path'
96
 
    parser = OptionParser(usage=usage)
97
 
    parser.add_option("-t", "--test", dest="test",
98
 
                  help = "run specific tests, e.g: className.methodName")
99
 
 
100
 
    (options, args) = parser.parse_args()
101
 
    if args:
102
 
        testpath = args[0]
103
 
        if not os.path.exists(testpath):
104
 
            print "the path to test does not exists!"
105
 
            sys.exit()
106
 
    else:
107
 
        testpath = None
108
 
    TestRunner().run(testpath, options.test)