Added some tests for issue 387. They are run in a separate process on

purpose.

Author: hategan

tests/test_issue_387_1.py

@@ -0,0 +1,11 @@
+import os
+import subprocess
+import sys
+
+import pytest
+
+
+@pytest.mark.timeout(5)
+def test_issue_387_1() -> None:
+    subprocess.run([sys.executable, os.path.abspath(__file__)[:-2] + '.run'],
+                   shell=True, check=True, capture_output=True)

tests/test_issue_387_1.run

@@ -0,0 +1,40 @@
+import logging
+import os
+from multiprocessing import Process, set_start_method
+from threading import Thread
+
+import psij
+
+
+def func():
+    # Get logs from PSI/J
+    logger = logging.getLogger()
+    logger.setLevel("DEBUG")
+    logger.addHandler(logging.StreamHandler())
+
+    exe = psij.JobExecutor.get_instance("local")
+    job = psij.Job(psij.JobSpec("test", "echo", arguments=["foo"]))
+    exe.submit(job)
+    print(job, flush=True)
+
+    # This hangs on the second round
+    job.wait()
+    print(job, flush=True)
+
+
+def fn():
+    print(os.getpid())
+
+
+if __name__ == "__main__":
+    Thread(target=fn).start()
+    set_start_method("fork")
+    p = Process(target=func)
+    p.start()
+    p.join()
+
+    print("Done, and again...")
+
+    p = Process(target=func)
+    p.start()
+    p.join()

tests/test_issue_387_2.py

@@ -0,0 +1,11 @@
+import os
+import subprocess
+import sys
+
+import pytest
+
+
+@pytest.mark.timeout(5)
+def test_issue_387_2() -> None:
+    subprocess.run([sys.executable, os.path.abspath(__file__)[:-2] + '.run'],
+                   shell=True, check=True, capture_output=True)

tests/test_issue_387_2.run

@@ -0,0 +1,23 @@
+import logging
+from threading import Thread
+import psij
+
+
+def func():
+    # Get logs from PSI/J
+    logger = logging.getLogger()
+    logger.setLevel("DEBUG")
+    logger.addHandler(logging.StreamHandler())
+
+    exe = psij.JobExecutor.get_instance("local")
+    job = psij.Job(psij.JobSpec("test", "echo", arguments=["foo"]))
+    exe.submit(job)
+
+    # This hangs
+    job.wait()
+
+if __name__ == "__main__":
+    p = Thread(target=func)
+    p.start()
+    p.join()
+

tests/test_issue_387_3.py

@@ -0,0 +1,11 @@
+import os
+import subprocess
+import sys
+
+import pytest
+
+
+@pytest.mark.timeout(5)
+def test_issue_387_3() -> None:
+    subprocess.run([sys.executable, os.path.abspath(__file__)[:-2] + '.run'],
+                   shell=True, check=True, capture_output=True)

tests/test_issue_387_3.run

@@ -0,0 +1,23 @@
+import logging
+from threading import Thread
+import psij
+
+
+def func():
+    # Get logs from PSI/J
+    logger = logging.getLogger()
+    logger.setLevel("DEBUG")
+    logger.addHandler(logging.StreamHandler())
+
+    exe = psij.JobExecutor.get_instance("local")
+    job = psij.Job(psij.JobSpec("test", "echo", arguments=["foo"]))
+    exe.submit(job)
+
+    # This hangs
+    job.wait()
+
+if __name__ == "__main__":
+    p = Thread(target=func)
+    p.start()
+    p.join()
+