gdnative_python/tests/threading/main.py

41 lines
1.0 KiB
Python
Raw Normal View History

import os
import pytest
from godot import exposed
from godot.bindings import Node, OS
__current_node = None
def set_current_node(node):
global __current_node
assert __current_node is None
__current_node = node
def get_current_node():
return __current_node
@exposed
class Main(Node):
def _ready(self):
set_current_node(self)
# Retrieve command line arguments passed through --pytest=...
prefix = "--pytest="
pytest_args = []
for gdarg in OS.get_cmdline_args():
arg = str(gdarg)
if arg.startswith(prefix):
pytest_args += arg[len(prefix) :].split(",")
if all(arg.startswith("-") for arg in pytest_args):
# Filter to avoid scanning `plugins` and `lib` directories
pytest_args += [x for x in os.listdir() if x.startswith("test_")]
# Run tests here
print(f"running `pytest {' '.join(pytest_args)}`")
if pytest.main(pytest_args):
OS.set_exit_code(1)
# Exit godot
self.get_tree().quit()