mirror of
https://github.com/github/codeql.git
synced 2025-12-16 16:53:25 +01:00
195b70aca6
This can also serve for a place to add tests for constructs like threading.Thread, mulitprocess.Process, concurrent.futures.ThreadPoolExecutor, and concurrent.futures.ProcessPoolExecutor.
92 lines
2.9 KiB
Python
92 lines
2.9 KiB
Python
import sys
|
|
|
|
def check_output(outtext, f):
|
|
if outtext == "OK\n":
|
|
pass
|
|
else:
|
|
raise RuntimeError("Function failed", outtext, f.__name__)
|
|
|
|
|
|
def check_test_function(f):
|
|
from io import StringIO
|
|
import sys
|
|
|
|
capturer = StringIO()
|
|
old_stdout = sys.stdout
|
|
sys.stdout = capturer
|
|
f()
|
|
sys.stdout = old_stdout
|
|
check_output(capturer.getvalue(), f)
|
|
|
|
|
|
def check_async_test_function(f):
|
|
from io import StringIO
|
|
import sys
|
|
import asyncio
|
|
|
|
capturer = StringIO()
|
|
old_stdout = sys.stdout
|
|
sys.stdout = capturer
|
|
asyncio.run(f())
|
|
sys.stdout = old_stdout
|
|
check_output(capturer.getvalue(), f)
|
|
|
|
|
|
def check_tests_valid(testFile):
|
|
import importlib
|
|
|
|
tests = importlib.import_module(testFile)
|
|
for i in dir(tests):
|
|
# print("Considering", i)
|
|
if i.startswith("test_"):
|
|
item = getattr(tests, i)
|
|
if callable(item):
|
|
print("Checking", testFile, item.__name__)
|
|
check_test_function(item)
|
|
|
|
elif i.startswith("atest_"):
|
|
item = getattr(tests, i)
|
|
if callable(item):
|
|
print("Checking", testFile, item.__name__)
|
|
check_async_test_function(item)
|
|
|
|
|
|
def check_tests_valid_after_version(testFile, version):
|
|
|
|
if sys.version_info[:2] >= version:
|
|
print("INFO: Will run tests in", testFile, "since we're running Python", version, "or newer")
|
|
check_tests_valid(testFile)
|
|
else:
|
|
print("WARN: Will not run tests in", testFile, "since we're running Python", sys.version_info[:2], "and need", version, "or newer")
|
|
|
|
if __name__ == "__main__":
|
|
check_tests_valid("coverage.classes")
|
|
check_tests_valid("coverage.test")
|
|
check_tests_valid("coverage.argumentPassing")
|
|
check_tests_valid("coverage.datamodel")
|
|
check_tests_valid("coverage.test_builtins")
|
|
check_tests_valid("coverage.loops")
|
|
check_tests_valid("coverage.functional")
|
|
check_tests_valid("coverage-py2.classes")
|
|
check_tests_valid("coverage-py3.classes")
|
|
check_tests_valid("variable-capture.in")
|
|
check_tests_valid("variable-capture.nonlocal")
|
|
check_tests_valid("variable-capture.global")
|
|
check_tests_valid("variable-capture.dict")
|
|
check_tests_valid("variable-capture.test_collections")
|
|
check_tests_valid("variable-capture.by_value")
|
|
check_tests_valid("variable-capture.test_library_calls")
|
|
check_tests_valid("variable-capture.test_fields")
|
|
check_tests_valid("module-initialization.multiphase")
|
|
check_tests_valid("fieldflow.test")
|
|
check_tests_valid("fieldflow.test_dict")
|
|
check_tests_valid_after_version("match.test", (3, 10))
|
|
check_tests_valid("exceptions.test")
|
|
check_tests_valid_after_version("exceptions.test_group", (3, 11))
|
|
|
|
# The below fails when trying to import modules
|
|
# check_tests_valid("module-initialization.test")
|
|
# check_tests_valid("module-initialization.testOnce")
|
|
|
|
print("\n🎉 All tests passed 🎉")
|