Statistics
| Branch: | Tag: | Revision:

root / kamaki / cli / test.py @ 1e1d6f27

History | View | Annotate | Download (3.7 kB)

1
# Copyright 2013 GRNET S.A. All rights reserved.
2
#
3
# Redistribution and use in source and binary forms, with or
4
# without modification, are permitted provided that the following
5
# conditions are met:
6
#
7
#   1. Redistributions of source code must retain the above
8
#      copyright notice, this list of conditions and the following
9
#      disclaimer.
10
#
11
#   2. Redistributions in binary form must reproduce the above
12
#      copyright notice, this list of conditions and the following
13
#      disclaimer in the documentation and/or other materials
14
#      provided with the distribution.
15
#
16
# THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS
17
# OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
18
# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
19
# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR
20
# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
21
# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
22
# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
23
# USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
24
# AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25
# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
26
# ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
27
# POSSIBILITY OF SUCH DAMAGE.
28
#
29
# The views and conclusions contained in the software and
30
# documentation are those of the authors and should not be
31
# interpreted as representing official policies, either expressed
32
# or implied, of GRNET S.A.
33

    
34
from unittest import makeSuite, TestSuite, TextTestRunner, TestCase
35
from inspect import getmembers, isclass
36

    
37
from kamaki.cli.command_tree.test import Command, CommandTree
38
from kamaki.cli.argument.test import (
39
    Argument, ConfigArgument, RuntimeConfigArgument, FlagArgument,
40
    ValueArgument, IntArgument, DateArgument, VersionArgument,
41
    KeyValueArgument, ProgressBarArgument, ArgumentParseManager)
42

    
43

    
44
#  TestCase auxiliary methods
45

    
46
def runTestCase(cls, test_name, args=[], failure_collector=[]):
47
    """
48
    :param cls: (TestCase) a set of Tests
49

50
    :param test_name: (str)
51

52
    :param args: (list) these are prefixed with test_ and used as params when
53
        instantiating cls
54

55
    :param failure_collector: (list) collects info of test failures
56

57
    :returns: (int) total # of run tests
58
    """
59
    suite = TestSuite()
60
    if args:
61
        suite.addTest(cls('_'.join(['test'] + args)))
62
    else:
63
        suite.addTest(makeSuite(cls))
64
    print('* Test * %s *' % test_name)
65
    r = TextTestRunner(verbosity=2).run(suite)
66
    failure_collector += r.failures
67
    return r.testsRun
68

    
69

    
70
def get_test_classes(module=__import__(__name__), name=''):
71
    module_stack = [module]
72
    while module_stack:
73
        module = module_stack[-1]
74
        module_stack = module_stack[:-1]
75
        for objname, obj in getmembers(module):
76
            if (objname == name or not name):
77
                if isclass(obj) and objname != 'TestCase' and (
78
                        issubclass(obj, TestCase)):
79
                    yield (obj, objname)
80

    
81

    
82
def main(argv):
83
    found = False
84
    failure_collector = list()
85
    num_of_tests = 0
86
    for cls, name in get_test_classes(name=argv[1] if len(argv) > 1 else ''):
87
        found = True
88
        num_of_tests += runTestCase(cls, name, argv[2:], failure_collector)
89
    if not found:
90
        print('Test "%s" not found' % ' '.join(argv[1:]))
91
    else:
92
        for i, failure in enumerate(failure_collector):
93
            print('Failure %s: ' % (i + 1))
94
            for field in failure:
95
                print('\t%s' % field)
96
        print('\nTotal tests run: %s' % num_of_tests)
97
        print('Total failures: %s' % len(failure_collector))
98

    
99

    
100
if __name__ == '__main__':
101
    from sys import argv
102
    main(argv)