Merge pull request #492 from dimastbk/add-test

add test for DEADLINE_EXCEEDED
camunda-community-hub · Sep 26, 2024 · 7172743 · 7172743
2 parents bed8bf6 + 93c89ed
commit 7172743
Showing 1 changed file with 6 additions and 0 deletions.
diff --git 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,