1 # Copyright 2013 GRNET S.A. All rights reserved.
3 # Redistribution and use in source and binary forms, with or
4 # without modification, are permitted provided that the following
7 # 1. Redistributions of source code must retain the above
8 # copyright notice, this list of conditions and the following
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.
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.
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.
34 from unittest import makeSuite, TestSuite, TextTestRunner, TestCase
35 from inspect import getmembers, isclass
37 from kamaki.cli.command_tree.test import Command, CommandTree
38 from kamaki.cli.argument.test import Argument
41 # TestCase auxiliary methods
43 def runTestCase(cls, test_name, args=[], failure_collector=[]):
45 :param cls: (TestCase) a set of Tests
47 :param test_name: (str)
49 :param args: (list) these are prefixed with test_ and used as params when
52 :param failure_collector: (list) collects info of test failures
54 :returns: (int) total # of run tests
58 suite.addTest(cls('_'.join(['test'] + args)))
60 suite.addTest(makeSuite(cls))
61 print('* Test * %s *' % test_name)
62 r = TextTestRunner(verbosity=2).run(suite)
63 failure_collector += r.failures
67 def get_test_classes(module=__import__(__name__), name=''):
68 module_stack = [module]
70 module = module_stack[-1]
71 module_stack = module_stack[:-1]
72 for objname, obj in getmembers(module):
73 if (objname == name or not name):
74 if isclass(obj) and objname != 'TestCase' and (
75 issubclass(obj, TestCase)):
81 failure_collector = list()
83 for cls, name in get_test_classes(name=argv[1] if len(argv) > 1 else ''):
85 num_of_tests += runTestCase(cls, name, argv[2:], failure_collector)
87 print('Test "%s" not found' % ' '.join(argv[1:]))
89 for i, failure in enumerate(failure_collector):
90 print('Failure %s: ' % (i + 1))
93 print('\nTotal tests run: %s' % num_of_tests)
94 print('Total failures: %s' % len(failure_collector))
97 if __name__ == '__main__':