diff --git a/tests/unit/grpc_internals/zeebe_adapter_base_test.py b/tests/unit/grpc_internals/zeebe_adapter_base_test.py index 90497ebe..ed30bda8 100644 --- a/tests/unit/grpc_internals/zeebe_adapter_base_test.py +++ b/tests/unit/grpc_internals/zeebe_adapter_base_test.py @@ -5,6 +5,7 @@ from pyzeebe.errors import ( ZeebeBackPressureError, + ZeebeDeadlineExceeded, ZeebeGatewayUnavailableError, ZeebeInternalError, ) @@ -35,6 +36,11 @@ async def test_raises_back_pressure_error_on_resource_exhausted(self, zeebe_adap with pytest.raises(ZeebeBackPressureError): await zeebe_adapter._handle_grpc_error(error) + async def test_raises_deadline_exceeded_on_deadline_exceeded(self, zeebe_adapter: ZeebeAdapterBase): + error = grpc.aio.AioRpcError(grpc.StatusCode.DEADLINE_EXCEEDED, None, None) + with pytest.raises(ZeebeDeadlineExceeded): + await zeebe_adapter._handle_grpc_error(error) + async def test_raises_gateway_unavailable_on_unavailable_status( self, zeebe_adapter: ZeebeAdapterBase,