Files
openide/python/helpers/pycharm/_jb_unittest_runner.py
Ilya.Kazakevich 9614f73c67 CWM-2243: Close suites on the Python side.
Python runners aren't aware of suites: they report ``fully.qualified.test.names``.

These names are splitted to the suites on the Python side, and just before process ends we close all of them.

GitOrigin-RevId: 4e1ad5b308cd393ac8877e1f6e1f616b254a130c
2023-02-01 17:43:40 +00:00

42 lines
1.8 KiB
Python

# coding=utf-8
import os
import sys
from unittest import main
from _jb_runner_tools import jb_start_tests, jb_doc_args, JB_DISABLE_BUFFERING, \
PROJECT_DIR, jb_finish_tests
from teamcity import unittestpy
if __name__ == '__main__':
path, targets, additional_args = jb_start_tests()
args = ["python -m unittest"]
if path:
assert os.path.exists(path), "{0}: No such file or directory".format(path)
if sys.version_info > (3, 0) and os.path.isfile(path):
# In Py3 it is possible to run script directly which is much more stable than discovery machinery
# For example it supports hyphens in file names PY-23549
additional_args = [path] + additional_args
else:
discovery_args = ["discover", "-s"]
# Unittest in py2 does not support running script directly (and folders in py2 and py3),
# but it can use "discover" to find all tests in some folder (optionally filtering by script)
if os.path.isfile(path):
discovery_args += [os.path.dirname(path), "-p", os.path.basename(path)]
else:
discovery_args.append(path)
discovery_args += ["-t",
PROJECT_DIR] # To force unit calculate path relative to this folder
additional_args = discovery_args + additional_args
elif targets:
additional_args += targets
args += additional_args
jb_doc_args("unittests", args)
# Working dir should be on path, that is how unittest work when launched from command line
sys.path.insert(0, PROJECT_DIR)
try:
sys.exit(main(argv=args, module=None, testRunner=unittestpy.TeamcityTestRunner,
buffer=not JB_DISABLE_BUFFERING))
finally:
jb_finish_tests()