Extend reencryption timeout, and make reencryption / cbd request timeout parameterized.

This is a stop gap measure until we fully address timeouts post 7.0.0.
pull/3153/head
derekpierre 2023-06-21 15:36:05 -04:00
parent 0eb10e2d37
commit 66d50c363e
2 changed files with 7 additions and 5 deletions

View File

@ -47,7 +47,9 @@ class ThresholdDecryptionClient(ThresholdAccessControlClient):
node_or_sprout.mature()
response = (
self._learner.network_middleware.get_encrypted_decryption_share(
node_or_sprout, bytes(encrypted_request)
ursula=node_or_sprout,
decryption_request_bytes=bytes(encrypted_request),
timeout=timeout,
)
)
if response.status_code == 200:

View File

@ -253,23 +253,23 @@ class RestMiddleware:
)
return response
def reencrypt(self, ursula: 'Ursula', reencryption_request_bytes: bytes):
def reencrypt(self, ursula: "Ursula", reencryption_request_bytes: bytes, timeout=8):
response = self.client.post(
node_or_sprout=ursula,
path=f"reencrypt",
data=reencryption_request_bytes,
timeout=2
timeout=timeout,
)
return response
def get_encrypted_decryption_share(
self, ursula: "Ursula", decryption_request_bytes: bytes
self, ursula: "Ursula", decryption_request_bytes: bytes, timeout=8
):
response = self.client.post(
node_or_sprout=ursula,
path=f"decrypt",
data=decryption_request_bytes,
timeout=8
timeout=timeout,
)
return response