bitbake-selftest 1.8 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374
  1. #!/usr/bin/env python3
  2. #
  3. # Copyright (C) 2012 Richard Purdie
  4. #
  5. # SPDX-License-Identifier: GPL-2.0-only
  6. #
  7. import os
  8. import sys, logging
  9. sys.path.insert(0, os.path.join(os.path.dirname(os.path.dirname(__file__)), 'lib'))
  10. import unittest
  11. try:
  12. import bb
  13. import hashserv
  14. import layerindexlib
  15. except RuntimeError as exc:
  16. sys.exit(str(exc))
  17. tests = ["bb.tests.codeparser",
  18. "bb.tests.color",
  19. "bb.tests.cooker",
  20. "bb.tests.cow",
  21. "bb.tests.data",
  22. "bb.tests.event",
  23. "bb.tests.fetch",
  24. "bb.tests.parse",
  25. "bb.tests.persist_data",
  26. "bb.tests.runqueue",
  27. "bb.tests.siggen",
  28. "bb.tests.utils",
  29. "hashserv.tests",
  30. "layerindexlib.tests.layerindexobj",
  31. "layerindexlib.tests.restapi",
  32. "layerindexlib.tests.cooker"]
  33. for t in tests:
  34. t = '.'.join(t.split('.')[:3])
  35. __import__(t)
  36. # Set-up logging
  37. class StdoutStreamHandler(logging.StreamHandler):
  38. """Special handler so that unittest is able to capture stdout"""
  39. def __init__(self):
  40. # Override __init__() because we don't want to set self.stream here
  41. logging.Handler.__init__(self)
  42. @property
  43. def stream(self):
  44. # We want to dynamically write wherever sys.stdout is pointing to
  45. return sys.stdout
  46. handler = StdoutStreamHandler()
  47. bb.logger.addHandler(handler)
  48. bb.logger.setLevel(logging.DEBUG)
  49. ENV_HELP = """\
  50. Environment variables:
  51. BB_SKIP_NETTESTS set to 'yes' in order to skip tests using network
  52. connection
  53. BB_TMPDIR_NOCLEAN set to 'yes' to preserve test tmp directories
  54. """
  55. class main(unittest.main):
  56. def _print_help(self, *args, **kwargs):
  57. super(main, self)._print_help(*args, **kwargs)
  58. print(ENV_HELP)
  59. if __name__ == '__main__':
  60. main(defaultTest=tests, buffer=True)