diff --git a/tests/test_manhole.py b/tests/test_manhole.py index ea735c3..e921b8d 100644 --- a/tests/test_manhole.py +++ b/tests/test_manhole.py @@ -26,6 +26,17 @@ SOCKET_PATH = '/tmp/manhole-socket' +def handle_sigterm(signo, frame): + # Simulate real termination + print("Terminated") + sys.exit(128 + signo) + + +# Handling sigterm ensure that atexit functions are called, and we do not leave +# leftover /tmp/manhole-pid sockets. +signal.signal(signal.SIGTERM, handle_sigterm) + + def is_module_available(mod): try: return imp.find_module(mod)