blob: 6c0737416beeb9d2fc64471e063e902b6ad49e75 [file] [log] [blame]
Patrick Williamsc0f7c042017-02-23 20:41:17 -06001#!/usr/bin/env python3
Patrick Williamsc124f4f2015-09-15 14:41:29 -05002#
3# Copyright (C) 2012 Richard Purdie
4#
Brad Bishopc342db32019-05-15 21:57:59 -04005# SPDX-License-Identifier: GPL-2.0-only
Patrick Williamsc124f4f2015-09-15 14:41:29 -05006#
Patrick Williamsc124f4f2015-09-15 14:41:29 -05007
8import os
9import sys, logging
10sys.path.insert(0, os.path.join(os.path.dirname(os.path.dirname(__file__)), 'lib'))
11
12import unittest
13try:
14 import bb
Brad Bishop19323692019-04-05 15:28:33 -040015 import hashserv
Brad Bishop1a4b7ee2018-12-16 17:11:34 -080016 import layerindexlib
Patrick Williamsc124f4f2015-09-15 14:41:29 -050017except RuntimeError as exc:
18 sys.exit(str(exc))
19
Patrick Williamsc0f7c042017-02-23 20:41:17 -060020tests = ["bb.tests.codeparser",
Andrew Geissler635e0e42020-08-21 15:58:33 -050021 "bb.tests.color",
Brad Bishop1a4b7ee2018-12-16 17:11:34 -080022 "bb.tests.cooker",
Patrick Williamsc0f7c042017-02-23 20:41:17 -060023 "bb.tests.cow",
24 "bb.tests.data",
Brad Bishopd7bf8c12018-02-25 22:55:05 -050025 "bb.tests.event",
Patrick Williamsc0f7c042017-02-23 20:41:17 -060026 "bb.tests.fetch",
27 "bb.tests.parse",
Brad Bishop19323692019-04-05 15:28:33 -040028 "bb.tests.persist_data",
Brad Bishop96ff1982019-08-19 13:50:42 -040029 "bb.tests.runqueue",
Andrew Geisslerc3d88e42020-10-02 09:45:00 -050030 "bb.tests.siggen",
Brad Bishop1a4b7ee2018-12-16 17:11:34 -080031 "bb.tests.utils",
Brad Bishop19323692019-04-05 15:28:33 -040032 "hashserv.tests",
Brad Bishop1a4b7ee2018-12-16 17:11:34 -080033 "layerindexlib.tests.layerindexobj",
34 "layerindexlib.tests.restapi",
35 "layerindexlib.tests.cooker"]
Patrick Williamsc124f4f2015-09-15 14:41:29 -050036
37for t in tests:
38 t = '.'.join(t.split('.')[:3])
39 __import__(t)
40
Patrick Williamsc124f4f2015-09-15 14:41:29 -050041
Patrick Williamsc0f7c042017-02-23 20:41:17 -060042# Set-up logging
43class StdoutStreamHandler(logging.StreamHandler):
44 """Special handler so that unittest is able to capture stdout"""
45 def __init__(self):
46 # Override __init__() because we don't want to set self.stream here
47 logging.Handler.__init__(self)
48
49 @property
50 def stream(self):
51 # We want to dynamically write wherever sys.stdout is pointing to
52 return sys.stdout
53
54
55handler = StdoutStreamHandler()
56bb.logger.addHandler(handler)
57bb.logger.setLevel(logging.DEBUG)
58
59
60ENV_HELP = """\
61Environment variables:
62 BB_SKIP_NETTESTS set to 'yes' in order to skip tests using network
63 connection
64 BB_TMPDIR_NOCLEAN set to 'yes' to preserve test tmp directories
65"""
66
67class main(unittest.main):
68 def _print_help(self, *args, **kwargs):
69 super(main, self)._print_help(*args, **kwargs)
70 print(ENV_HELP)
71
72
73if __name__ == '__main__':
74 main(defaultTest=tests, buffer=True)