GeezerVison is showing github.com — pythonVSCode/python_files/testing_tools/unittest_discovery.py at main · Busra67/pythonVSCode

Expand file tree

unittest_discovery.py

Copy path

More file actions

Latest commit

History

64 lines (53 loc) · 1.71 KB

unittest_discovery.py

Copy path

File metadata and controls

64 lines (53 loc) · 1.71 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

import contextlib

import inspect

import os

import sys

import traceback

import unittest

start_dir = sys.argv[1]

pattern = sys.argv[2]

top_level_dir = sys.argv[3] if len(sys.argv) >= 4 else None

sys.path.insert(0, os.getcwd())

def get_sourceline(obj):

try:

s, n = inspect.getsourcelines(obj)

except Exception:

try:

# this handles `tornado` case we need a better

# way to get to the wrapped function.

# XXX This is a temporary solution

s, n = inspect.getsourcelines(obj.orig_method)

except Exception:

return "*"

for i, v in enumerate(s):

if v.strip().startswith(("def", "async def")):

return str(n + i)

return "*"

def generate_test_cases(suite):

for test in suite:

if isinstance(test, unittest.TestCase):

yield test

else:

for test_case in generate_test_cases(test):

yield test_case

try:

loader = unittest.TestLoader()

suite = loader.discover(start_dir, pattern=pattern, top_level_dir=top_level_dir)

print("start") # Don't remove this line

loader_errors = []

for s in generate_test_cases(suite):

tm = getattr(s, s._testMethodName)

testId = s.id()

if testId.startswith("unittest.loader._FailedTest"):

loader_errors.append(s._exception)

else:

print(testId.replace(".", ":") + ":" + get_sourceline(tm))

except Exception:

print("=== exception start ===")

traceback.print_exc()

print("=== exception end ===")

for error in loader_errors:

with contextlib.suppress(Exception):

print("=== exception start ===")

print(error.msg)

print("=== exception end ===")