Add tests running Quit

Author: Dorian Birraux

wolframclient/tests/evaluation/test_coroutine.py

@@ -2,6 +2,7 @@
 
 import logging
 
+from wolframclient.exception import WolframKernelException
 from wolframclient.deserializers import WXFConsumer, binary_deserialize
 from wolframclient.evaluation import (
     WolframCloudAsyncSession,
@@ -111,6 +112,25 @@ async def test_eval_parallel(self):
         res = await asyncio.gather(*tasks)
         self.assertEqual(res, list(range(1, 11)))
 
+    @run_in_loop
+    async def test_quit_restart(self):
+        try:
+            async_session = WolframLanguageAsyncSession(
+                kernel_path, kernel_loglevel=logging.INFO
+            )
+            pid1 = await async_session.evaluate("$ProcessID")
+            self.assertEqual(pid1, async_session.kernel_controller.pid)
+            with self.assertRaises(WolframKernelException):
+                await async_session.evaluate("Quit[]")
+            await async_session.terminate()
+            pid2 = await async_session.evaluate("$ProcessID")
+            self.assertEqual(pid2, async_session.kernel_controller.pid)
+            self.assertNotEqual(pid1, pid2)
+        finally:
+            if async_session:
+                await async_session.terminate()
+            self.assertTrue(async_session.stopped)
+
     def test_kwargs_parameters(self):
         TestKernelBase.class_kwargs_parameters(self, WolframLanguageAsyncSession)
 
@@ -207,7 +227,7 @@ async def test_pool_from_mixed_kernel_cloud_path(self):
             kernel_path,
         )
         async with WolframEvaluatorPool(
-            sessions, kernel_loglevel=logging.INFO, STARTUP_TIMEOUT=5, TERMINATE_TIMEOUT=3
+            sessions, kernel_loglevel=logging.INFO, STARTUP_TIMEOUT=20, TERMINATE_TIMEOUT=3
         ) as pool:
             await self._pool_evaluation_check(pool)
         for session in sessions:

wolframclient/tests/evaluation/test_kernel.py

@@ -178,6 +178,21 @@ def test_auto_start_session(self):
             session.terminate()
             self.assertTrue(session.stopped)
 
+    def test_kernel_dies_restart(self):
+        try:
+            session = WolframLanguageSession(kernel_path)
+            pid1 = session.evaluate("$ProcessID")
+            self.assertEqual(pid1, session.kernel_controller.pid)
+            with self.assertRaises(WolframKernelException):
+                session.evaluate("Quit[]")
+            session.terminate()
+            pid2 = session.evaluate("$ProcessID")
+            self.assertEqual(pid2, session.kernel_controller.pid)
+            self.assertNotEqual(pid1, pid2)
+        finally:
+            session.terminate()
+            self.assertTrue(session.stopped)
+
     def test_pure_function_inputform(self):
         f = self.kernel_session.function("#+1&")
         self.assertEqual(f(3), 4)
@@ -298,7 +313,7 @@ def test_throw(self):
 class TestSessionTimeout(TestCaseSettings):
     def test_evaluate_async_basic_inputform(self):
         future = self.kernel_session.evaluate_future("1+1")
-        self.assertEqual(future.result(timeout=2), 2)
+        self.assertEqual(future.result(timeout=5), 2)
 
     def test_evaluate_async_basic_wl(self):
         future = self.kernel_session.evaluate_future(wl.Plus(1, 2))