extend test

Author: jan-janssen

tests/test_cache_fileexecutor_serial.py

@@ -105,7 +105,7 @@ def test_executor_error_file(self):
         self.assertTrue("error.out" in working_directory_file_lst)
         os.remove(os.path.join(cwd, "error.out"))
 
-    def test_executor_function(self):
+    def test_executor_function_cancel_true(self):
         fs1 = Future()
         q = Queue()
         q.put(
@@ -126,6 +126,51 @@ def test_executor_function(self):
                 "execute_function": execute_in_subprocess,
                 "resource_dict": {"cores": 1, "cwd": None, "cache_directory": cache_dir},
                 "terminate_function": terminate_subprocess,
+                "cancel_futures_on_shutdown": True,
+            },
+        )
+        process.start()
+        self.assertFalse(fs1.done())
+        self.assertEqual(fs1.result(), 3)
+        self.assertTrue(fs1.done())
+        q.put({"shutdown": True, "wait": True})
+        process.join()
+
+    def test_executor_function_cancel_false(self):
+        fs1 = Future()
+        q = Queue()
+        q.put(
+            {
+                "fn": my_funct,
+                "args": (),
+                "kwargs": {"a": 1, "b": 2},
+                "future": fs1,
+                "resource_dict": {},
+            }
+        )
+        cache_dir = os.path.abspath("executorlib_cache")
+        os.makedirs(cache_dir, exist_ok=True)
+        process = Thread(
+            target=execute_tasks_h5,
+            kwargs={
+                "future_queue": q,
+                "execute_function": execute_in_subprocess,
+                "resource_dict": {"cores": 1, "cwd": None, "cache_directory": cache_dir},
+                "terminate_function": terminate_subprocess,
+                "cancel_futures_on_shutdown": False,
+            },
+        )
+        process.start()
+        self.assertTrue(fs1.cancelled())
+        sleep(2)
+        process = Thread(
+            target=execute_tasks_h5,
+            kwargs={
+                "future_queue": q,
+                "execute_function": execute_in_subprocess,
+                "resource_dict": {"cores": 1, "cwd": None, "cache_directory": cache_dir},
+                "terminate_function": terminate_subprocess,
+                "cancel_futures_on_shutdown": False,
             },
         )
         process.start()