Files
deskflow/scripts/tests.py
Nick Bolton b02077550c Expand BSD sockets poll tests and remove legacy-poll code (#7414)
* Move integ test to corredt place and re-enable tests

* Depend on `poll.h` (removes support for non-POSIX systems)

* Only build tests for each arch

* Move back to unit tests (poll is mocked)

* Better error message for valgrind not found

* Simplify dependency injection for BSD sockets poll test

* Improve test readability for BSD net poll

* Split out 2-in-1 test for `isAnyAddr`

* Stub out sleep function

* Improve coverage for pollSocket

* Use gmock ON_CALL instead of manual mock

* Remove unused function signature

* Use conventional deps struct instead of std functional

* Add test for socket data FD set to -1

* Add assertation for adding unblock pipe

* Use older style array alloc

* Less precision around `getNetworkDataForThread` value

* Use `ssize_t` for `ignore`

* Remove unused var

* Update ChangeLog
2024-07-29 18:04:17 +01:00

68 lines
1.7 KiB
Python
Executable File

#!/usr/bin/env python3
import lib.env as env
env.ensure_in_venv(__file__)
import argparse, os, sys
import lib.cmd_utils as cmd_utils
import lib.colors as colors
valgrind_bin = "valgrind"
def main():
parser = argparse.ArgumentParser()
parser.add_argument("--unit-tests", action="store_true")
parser.add_argument("--integ-tests", action="store_true")
parser.add_argument(
"--filter-file",
type=str,
help="Takes the base filename without extension and uses it as a filter",
)
parser.add_argument(
"--ignore-return-code",
action="store_true",
help="Ignore the return code of the test command",
)
parser.add_argument(
"--valgrind",
action="store_true",
help="Run the test command with valgrind",
)
args = parser.parse_args()
binary = get_binary_path(args)
if args.filter_file:
file_base = os.path.basename(args.filter_file)
without_ext = os.path.splitext(file_base)[0]
command = [binary, f"--gtest_filter={without_ext}*"]
else:
command = [binary]
if args.valgrind:
if not cmd_utils.has_command(valgrind_bin):
print(f"{colors.ERROR_TEXT} {valgrind_bin} not found")
sys.exit(1)
command = [valgrind_bin] + command
result = cmd_utils.run(command, print_cmd=True, check=False)
if not args.ignore_return_code:
sys.exit(result.returncode)
def get_binary_path(args):
base_dir = "./build/bin"
if args.unit_tests:
return f"{base_dir}/unittests"
elif args.integ_tests:
return f"{base_dir}/integtests"
else:
raise RuntimeError("No test type specified")
if __name__ == "__main__":
main()