[_722] test.

Author: d-w-moore

irods/test/modules/test_parallel_put_sigint_response.py

@@ -0,0 +1,79 @@
+import signal,tempfile
+
+import os
+import re
+import subprocess
+import sys
+import time
+
+import irods
+import irods.helpers
+import irods.client_configuration as config
+from irods.test import modules as test_modules
+from irods.test.modules.test_auto_close_of_data_objects__issue_456 import (
+    auto_close_data_objects,
+)
+
+OBJECT_SIZE = 2*1024**3
+OBJECT_NAME = 'data_get_issue__722'
+LOCAL_TEMPFILE_NAME = 'data_object_for_issue_722.dat'
+
+_clock_resolution = max(.01, *(time.clock_getres(getattr(time,symbol))
+                               for symbol in dir(time) if symbol.startswith('CLOCK_')))
+def wait_till_true( function ):
+    while not (truth_value := function()):
+        time.sleep(_clock_resolution)
+    return truth_value
+
+
+def test(
+         # TODO - accept unittest testcase 'test_instance' param , replace assert's with test_instance.assert
+    ):
+    """Creates a child process executing a long get() and ensures the process can be
+    terminated using SIGINT or SIGTERM.
+    """
+    program = os.path.join(test_modules.__path__[0], os.path.basename(__file__))
+
+    # Call into this same module as a command.  This will initiate another Python process that
+    # performs a lengthy data object "get" operation (see the main body of the script, below.)
+    process = subprocess.Popen([sys.executable, program],
+                               stderr=subprocess.PIPE,
+                               stdout=subprocess.PIPE, text = True)
+
+    localfile = process.stdout.readline().strip()
+    assert wait_till_true(lambda:os.path.exists(localfile) and os.stat(localfile).st_size > OBJECT_SIZE//2)
+
+    sig = signal.SIGINT
+    process.send_signal(sig)
+    class TestFailed(Exception):pass
+    try:
+        assert process.wait(timeout = 15) == -sig
+        assert re.search('KeyboardInterrupt',process.stderr.read())
+    except subprocess.TimeoutExpired as timeout_exc:
+        raise TestFailed("Non-daemon thread(s) probably prevented subprocess's main thread from exiting") from timeout_exc
+    return True
+
+
+if __name__ == "__main__":
+    session = irods.helpers.make_session()
+    hc = irods.helpers.home_collection(session)
+    TESTFILE_FILL = b'_'*(1024*1024)
+    object_path = f'{hc}/{OBJECT_NAME}'
+    with session.data_objects.open(object_path,'w') as f:
+        for y in range(OBJECT_SIZE//len(TESTFILE_FILL)):
+            f.write(TESTFILE_FILL)
+    local_path = None
+    try:
+        with tempfile.NamedTemporaryFile(prefix='local_file_issue_722.dat', delete = True) as t:
+            local_path = t.name
+
+        # Tell the parent process the name of the local file being "get"ted (got) from iRODS
+        print(local_path)
+        sys.stdout.flush()
+
+        session.data_objects.get(object_path, local_path)
+    finally:
+        if local_path is not None and os.path.exists(local_path):
+            os.unlink(local_path)
+        if session.data_objects.exists(object_path):
+            session.data_objects.unlink(object_path, force=True)