Avoid timeout overflow in reverse proxies (#176)

Signed-off-by: Michel Hidalgo <mhidalgo@theaiinstitute.com>

Author: mhidalgo-rai

spot_wrapper/testing/grpc.py

@@ -680,12 +680,12 @@ class ProxiedUnaryRpcHandler(ProxiedRpcHandler):
 
     def __call__(self, request: typing.Any, context: grpc.ServicerContext) -> typing.Any:
         future = self._server.submit(request)
-        return future.result(timeout=context.time_remaining())
+        return future.result(timeout=min(context.time_remaining(), threading.TIMEOUT_MAX))
 
 
 class ProxiedStreamRpcHandler(ProxiedRpcHandler):
     """A gRPC any-stream handler that proxies requests."""
 
     def __call__(self, request: typing.Any, context: grpc.ServicerContext) -> typing.Iterator:
         future = self._server.submit(request)
-        yield from future.result(timeout=context.time_remaining())
+        yield from future.result(timeout=min(context.time_remaining(), threading.TIMEOUT_MAX))