fix tests

Author: efiop

src/isolate/server/server.py

@@ -12,7 +12,7 @@
 from functools import partial
 from queue import Empty as QueueEmpty
 from queue import Queue
-from typing import Any, Callable, Dict, Iterator, List, Tuple, cast
+from typing import Any, Callable, Iterator, cast
 from urllib.request import Request, urlopen
 
 import grpc
@@ -385,8 +385,7 @@ def Run(
         context: ServicerContext,
     ) -> Iterator[definitions.PartialRunResult]:
         try:
-            for message in self.run_function(request):
-                yield message
+            yield from self.run_function(request)
         except GRPCException as exc:
             return self.abort_with_msg(
                 f"{exc}",

tests/test_server.py

@@ -684,8 +684,8 @@ def test_server_proper_error_delegation(
 
     assert exc_info.value.code() == grpc.StatusCode.INVALID_ARGUMENT
     assert (
-        exc_info.value.details()
-        == "Error while serializing the execution result (object of type <class 'frame'>)."
+        "Error while serializing the execution result (object of type <class 'frame'>)."
+        in exc_info.value.details()
     )
     assert not user_logs
 
@@ -695,8 +695,8 @@ def test_server_proper_error_delegation(
 
     assert exc_info.value.code() == grpc.StatusCode.INVALID_ARGUMENT
     assert (
-        exc_info.value.details()
-        == "Error while serializing the execution result (object of type <class 'Exception'>)."
+        "Error while serializing the execution result (object of type <class 'Exception'>)."
+        in exc_info.value.details()
     )
     assert "relevant information" in "\n".join(log.message for log in user_logs)
 
@@ -712,12 +712,12 @@ def test_server_submit(
         function=prepare_request(imitate_server),
         callback=http_server.host,
     )
-    response = stub.Submit(request)
+    stub.Submit(request)
 
-    partial_results: list[definitions.PartialRunResult] = []
+    partial_results: List[definitions.PartialRunResult] = []
     while not any(pr.is_complete for pr in partial_results):
         try:
-            message = http_server.messages.get(timeout=10)
+            message = http_server.messages.get(timeout=120)
         except queue.Empty:
             raise ValueError("No message received from the server within 5 seconds")